Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import logging
 import sqlite3
 import requests
+from urllib.parse import urlparse, parse_qs
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -78,12 +79,18 @@ def get_user_points(username):
     try:
         response = requests.get(webhook_url, params=params)
         response.raise_for_status()
-        return response.json()
+        return response.json().get('points', "Error retrieving points")
     except requests.RequestException as e:
         logger.error(f"Error fetching user points: {e}")
-        return {"error": "Failed to retrieve user points"}
+        return "Failed to retrieve user points"
 
-def gradio_interface(prompt, resolution_key, username):
+def extract_username_from_url(url):
+    parsed_url = urlparse(url)
+    params = parse_qs(parsed_url.query)
+    return params.get('username', ["Guest"])[0]
+
+def gradio_interface(prompt, resolution_key, url):
+    username = extract_username_from_url(url)
     result = generate_image(prompt, resolution_key)
     
     if result and result[0]:
@@ -97,18 +104,21 @@ def gradio_interface(prompt, resolution_key, username):
     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"""
+        gr.HTML("""
         <script src="https://telegram.org/js/telegram-web-app.js"></script>
         <script>
-            document.addEventListener('DOMContentLoaded', function() {{
+            document.addEventListener('DOMContentLoaded', function() {
                 window.Telegram.WebApp.init();
-                window.Telegram.WebApp.setTitle("Welcome, {username}!");
-            }});
+                const urlParams = new URLSearchParams(window.location.search);
+                const username = urlParams.get('username') || "Guest";
+                window.Telegram.WebApp.setTitle(`Welcome, ${username}!`);
+                document.getElementById('username').value = username;  // Set hidden textbox with username
+            });
         </script>
-        <h3>Welcome, {username}! Generate images based on prompts.</h3>
+        <h3>Welcome! Generate images based on prompts.</h3>
         """)
 
         # Create input components
@@ -124,15 +134,19 @@ def create_gradio_interface(username):
         get_points_button = gr.Button("Get Points")
         points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
         
+        # Hidden textbox to hold the username
+        username_input = gr.Textbox(visible=False, default="Guest", elem_id="username")
+        
         # Set up interaction
         generate_button.click(
-            fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key, username),
-            inputs=[prompt_input, resolution_dropdown],
+            fn=lambda prompt, resolution_key, username: gradio_interface(prompt, resolution_key, username),
+            inputs=[prompt_input, resolution_dropdown, username_input],
             outputs=[result_output, message_output]
         )
 
         get_points_button.click(
-            fn=lambda: get_user_points(username),
+            fn=lambda username: get_user_points(username),
+            inputs=[username_input],
             outputs=points_output
         )
         
@@ -146,10 +160,10 @@ def create_gradio_interface(username):
     
     return interface
 
-def launch_gradio(username):
-    interface = create_gradio_interface(username)
+def launch_gradio():
+    interface = create_gradio_interface()
     interface.launch()
 
 if __name__ == '__main__':
-    # Launch Gradio with a default username
-    launch_gradio("Guest")
+    # Launch Gradio
+    launch_gradio()