Update app.py

app.py

@@ -121,7 +121,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 == "Guest":
+    if points == "User not found":
         # If user is not found, update points_output with the registration message
         return None, "Please register with @myexpsupportBot"
     
@@ -175,7 +175,7 @@ def create_gradio_interface():
         
         # Set up interactions
         generate_button.click(
-            fn=gradio_interface,
+            fn=handle_generate_image,
             inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
             outputs=[result_output, message_output]
         )