Update app.py

app.py

@@ -1,11 +1,8 @@
-from flask import Flask, request, jsonify
 import gradio as gr
 from gradio_client import Client
 import os
 import logging
 
-app = Flask(__name__)
-
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
 
@@ -47,33 +44,22 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         logger.error(f"Error generating image: {e}")
         return None
 
-def gradio_interface(prompt, resolution_key):
+def gradio_interface(username, prompt, resolution_key):
     result = generate_image(prompt, resolution_key)
     
     if result and result[0]:
         file_path = result[0][0].get('image')
         if file_path and os.path.exists(file_path):
-            return file_path, "The image was generated successfully."
+            return file_path, f"The image was generated successfully for {username}."
         else:
             return None, "The image file is not available. Please try again later."
     else:
         return None, "There was an error processing your photo. Please try again later."
 
-def create_gradio_interface(username):
+def create_gradio_interface():
     with gr.Blocks() as interface:
-        # Personalized HTML content
-        gr.HTML(f"""
-        <script src="https://telegram.org/js/telegram-web-app.js"></script>
-        <script>
-            document.addEventListener('DOMContentLoaded', function() {{
-                window.Telegram.WebApp.init();
-                window.Telegram.WebApp.setTitle("Welcome, {username}!");
-            }});
-        </script>
-        <h3>Welcome, {username}! Generate images based on prompts.</h3>
-        """)
-
         # Create input components
+        username_input = gr.Textbox(label="Username", placeholder="Enter your username here...")
         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")
@@ -84,8 +70,8 @@ def create_gradio_interface(username):
         
         # Set up interaction
         generate_button.click(
-            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
-            inputs=[prompt_input, resolution_dropdown],
+            fn=lambda username, prompt, resolution_key: gradio_interface(username, prompt, resolution_key),
+            inputs=[username_input, prompt_input, resolution_dropdown],
             outputs=[result_output, message_output]
         )
         
@@ -99,14 +85,9 @@ def create_gradio_interface(username):
     
     return interface
 
-@app.route('/')
-def serve_gradio():
-    username = request.args.get('username', 'Guest')
-    logger.info(f"Extracted username: {username}")
-
-    # Launch Gradio interface
-    interface = create_gradio_interface(username)
-    return gr.Interface.load(interface).launch(inline=True)
+def launch_gradio():
+    interface = create_gradio_interface()
+    interface.launch()
 
-if __name__ == '__main__':
-    app.run(debug=True)
+# Launch the Gradio interface
+launch_gradio()