Update app.py

app.py

@@ -4,20 +4,13 @@ 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__)
 
@@ -64,9 +57,6 @@ def create_gradio_interface(username):
             document.addEventListener('DOMContentLoaded', function() {{
                 window.Telegram.WebApp.init();
                 window.Telegram.WebApp.setTitle("Welcome, {username}!");
-                document.getElementById('my-button').addEventListener('click', function() {{
-                    window.Telegram.WebApp.close();
-                }});
             }});
         </script>
         <h3>Welcome, {username}! Generate images based on prompts.</h3>
@@ -94,7 +84,6 @@ def create_gradio_interface(username):
     return interface
 
 def launch_gradio(username):
-    # Launch the Gradio interface
     interface = create_gradio_interface(username)
     interface.launch()
 
@@ -104,12 +93,11 @@ def extract_username_from_url(url):
     return params.get('username', ["Guest"])[0]
 
 def get_username_from_url():
-    # Placeholder URL, replace with actual URL or method to get URL in deployment
-    url = 'https://measmonysuon-imagen.hf.space/?username=Guest'
+    # Placeholder URL; replace with actual URL or method to get URL in deployment
+    url = 'https://measmonysuon-imagen.hf.space/?username=MoMo'
     username = extract_username_from_url(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)