Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from gradio_client import Client
 import os
 import logging
+from urllib.parse import parse_qs, urlparse
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -44,22 +45,33 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         logger.error(f"Error generating image: {e}")
         return None
 
-def gradio_interface(username, prompt, resolution_key):
+def gradio_interface(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, f"The image was generated successfully for {username}."
+            return file_path, "The image was generated successfully."
         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():
+def create_gradio_interface(username):
     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")
@@ -70,8 +82,8 @@ def create_gradio_interface():
         
         # Set up interaction
         generate_button.click(
-            fn=lambda username, prompt, resolution_key: gradio_interface(username, prompt, resolution_key),
-            inputs=[username_input, prompt_input, resolution_dropdown],
+            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
+            inputs=[prompt_input, resolution_dropdown],
             outputs=[result_output, message_output]
         )
         
@@ -85,9 +97,23 @@ def create_gradio_interface():
     
     return interface
 
-def launch_gradio():
-    interface = create_gradio_interface()
+def launch_gradio(username):
+    interface = create_gradio_interface(username)
     interface.launch()
 
-# Launch the Gradio interface
-launch_gradio()
+def extract_username_from_url(url):
+    parsed_url = urlparse(url)
+    params = parse_qs(parsed_url.query)
+    logger.debug(f"Parsed URL parameters: {params}")  # Debugging information
+    return params.get('username', ["Guest"])[0]
+
+def get_username_from_url(url):
+    # Extract username from the actual URL
+    username = extract_username_from_url(url)
+    logger.info(f"Extracted username: {username}")
+    return username
+
+# Replace this URL with your actual URL where the Gradio app is hosted
+url = 'https://measmonysuon-imagen.hf.space/?username=measmony'  # This should be dynamically obtained if needed
+username = get_username_from_url(url)
+launch_gradio(username)