Update app.py

app.py

@@ -25,6 +25,8 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+url = 'https://measmonysuon-imagen.hf.space/?user_chat_id=00000000'
+
 def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     resolution = resolutions.get(resolution_key, (1024, 1024))
     width, height = resolution
@@ -80,7 +82,7 @@ def get_user_points(user_chat_id):
 def retrieve_user_points(user_chat_id):
     """Retrieve user points and display them."""
     points = get_user_points(user_chat_id)
-    if points == "User not found":
+    if points == "Guest":
         return "Please register with @myexpsupportBot", ""
     return f"Your current points: {points}", points
     
@@ -117,7 +119,7 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
 def handle_generate_image(prompt, resolution_key, user_chat_id):
     points = get_user_points(user_chat_id)
     
-    if points == "User not found":
+    if points == "Guest":
         # If user is not found, update points_output with the registration message
         return None, "Please register with @myexpsupportBot"
     
@@ -178,12 +180,12 @@ def create_gradio_interface(user_chat_id):
         # Set up interactions for the points retrieval
         get_points_button.click(
             fn=lambda url: (
-                extract_user_chat_id_from_url(url),  # Extract user_chat_id
-                retrieve_user_points(extract_user_chat_id_from_url(url))[0]  # Retrieve points based on extracted user_chat_id
+                (user_chat_id := extract_user_chat_id_from_url(url)),
+                points := retrieve_user_points(user_chat_id)[0],  # Retrieve points based on extracted user_chat_id
+                (user_chat_id, points)
             ),
-            inputs=[user_chat_id_input],
-            outputs=[points_output]  # Update user_chat_id_input and points_outpu
-        
+            inputs=[gr.Textbox(default="https://measmonysuon-imagen.hf.space/?user_chat_id=00000000")],  # Use a default URL or let it be dynamic
+            outputs=[user_chat_id_input, points_output]  # Update both outputs
         )
         gr.HTML("""
         <style>