Update app.py

app.py

@@ -50,7 +50,19 @@ def gradio_interface(prompt, resolution_key):
         return None, "There was an error processing your photo. Please try again later."
 
 def create_gradio_interface(username):
-    with gr.Blocks() as interface:
+        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:
         gr.HTML(f"""
         <script src="https://telegram.org/js/telegram-web-app.js"></script>
         <script>
@@ -62,16 +74,6 @@ def create_gradio_interface(username):
         <h3>Welcome, {username}! Generate images based on prompts.</h3>
         """)
         
-        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])
         
         gr.HTML("""
         <style>