Update app.py

app.py

@@ -121,7 +121,7 @@ def create_gradio_interface():
         # Create other components
         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")
+        generate_button = gr.Button("Generate", visible=False)  # Start with the button hidden
         
         result_output = gr.Image(label="Generated Image", type="pil")
         message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
@@ -137,20 +137,27 @@ def create_gradio_interface():
             resolution_dropdown
             generate_button
         
-        def update_points_and_generate_button(user_chat_id):
+        def handle_get_points(user_chat_id):
             points = get_user_points(user_chat_id)
             if points >= 5:
                 return points, gr.Button.update(visible=True)
             else:
                 return points, gr.Button.update(visible=False)
-        
-        def handle_get_points(user_chat_id):
-            points, button_state = update_points_and_generate_button(user_chat_id)
-            return points, button_state
 
-        # Set up interaction
+        def handle_generate_image(prompt, resolution_key, user_chat_id):
+            points = get_user_points(user_chat_id)
+            if points >= 5:
+                result = gradio_interface(prompt, resolution_key, user_chat_id)
+                if result[0]:  # If image generation is successful
+                    return result[0], "The image was generated successfully."
+                else:
+                    return None, "There was an error processing your photo. Please try again later."
+            else:
+                return None, "Insufficient points. Please get more points before generating an image."
+
+        # Set up interactions
         generate_button.click(
-            fn=lambda prompt, resolution_key, user_chat_id: gradio_interface(prompt, resolution_key, user_chat_id),
+            fn=handle_generate_image,
             inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
             outputs=[result_output, message_output]
         )