Update app.py

app.py

@@ -4,13 +4,20 @@ import os
 import logging
 from urllib.parse import parse_qs, urlparse
 
+# Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
+
+# Define resolutions
 resolutions = {
     "896x1152": (896, 1152),
     "1024x1024": (1024, 1024),
     "1216x832": (1216, 832)
 }
+
+# Define the default style
 DEFAULT_STYLE = "3840 x 2160"
+
+# Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
@@ -18,6 +25,7 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     resolution = resolutions.get(resolution_key, (1024, 1024))
     width, height = resolution
     full_prompt = f"{prompt}, Canon EOS R5, 4K, Photo-Realistic, appearing photorealistic with super fine details, high resolution, natural look, hyper realistic photography, cinematic lighting, --ar 64:37, --v 6.0, --style raw, --stylize 750"
+    
     try:
         result = client_image.predict(
             prompt=full_prompt,
@@ -50,19 +58,8 @@ def gradio_interface(prompt, resolution_key):
         return None, "There was an error processing your photo. Please try again later."
 
 def create_gradio_interface(username):
-        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
-        resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
-        generate_button = gr.Button("Generate")
-        
-        result_output = gr.Image(label="Generated Image", type="pil")
-        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
-        
-        generate_button.click(fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
-                              inputs=[prompt_input, resolution_dropdown],
-                              outputs=[result_output, message_output])
-
-
     with gr.Blocks() as interface:
+        # Personalized HTML content
         gr.HTML(f"""
         <script src="https://telegram.org/js/telegram-web-app.js"></script>
         <script>
@@ -73,7 +70,22 @@ def create_gradio_interface(username):
         </script>
         <h3>Welcome, {username}! Generate images based on prompts.</h3>
         """)
+
+        # Create input components
+        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
+        resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
+        generate_button = gr.Button("Generate")
         
+        # Create output components
+        result_output = gr.Image(label="Generated Image", type="pil")
+        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
+        
+        # Set up interaction
+        generate_button.click(
+            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
+            inputs=[prompt_input, resolution_dropdown],
+            outputs=[result_output, message_output]
+        )
         
         gr.HTML("""
         <style>
@@ -82,7 +94,7 @@ def create_gradio_interface(username):
             }
         </style>
         """)
-
+    
     return interface
 
 def launch_gradio(username):
@@ -101,5 +113,6 @@ def get_username_from_url():
     logger.info(f"Extracted username: {username}")
     return username
 
+# Launch Gradio with the extracted or default username
 username = get_username_from_url()
 launch_gradio(username)