Update app.py

app.py

@@ -105,14 +105,6 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
         return None, "There was an error processing your photo. Please try again later."
 
 def create_gradio_interface():
-    def update_points_and_check(user_chat_id):
-        points = get_user_points(user_chat_id)
-        if points == "Failed to retrieve user points":
-            return points, gr.Button("Generate", visible=False)
-        else:
-            points = int(points)
-            return points, gr.Button("Generate", visible=points >= 5)
-
     with gr.Blocks() as interface:
         # Personalized HTML content
         gr.HTML("""
@@ -126,41 +118,39 @@ def create_gradio_interface():
         # Create the button to get user points
         get_points_button = gr.Button("Get Points")
         
-        # Create a component for the button state
-        generate_button = gr.Button("Generate", visible=False)
+        # 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", elem_id="generate-button", visible=True)
         
-        # Arrange user_chat_id_input, points_output, and get_points_button in one row
+        result_output = gr.Image(label="Generated Image", type="pil")
+        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
+
+        # Arrange components in rows
         with gr.Row():
             user_chat_id_input
             points_output
             get_points_button
-
-        # Arrange generate_button in a separate row
-        with gr.Row():
-            generate_button
         
-        # Set up interactions
-        get_points_button.click(
-            fn=update_points_and_check,
-            inputs=[user_chat_id_input],
-            outputs=[points_output, generate_button]
-        )
+        # Ensure 'Generate' button is disabled initially
+        generate_button.update(visible=False)
 
+        def update_generate_button_state(user_chat_id):
+            points = get_user_points(user_chat_id)
+            if points >= 5:
+                return gr.Button.update(visible=True)
+            else:
+                return gr.Button.update(visible=False)
+
+        # Set up interaction
         generate_button.click(
             fn=lambda prompt, resolution_key, user_chat_id: gradio_interface(prompt, resolution_key, user_chat_id),
             inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
             outputs=[result_output, message_output]
         )
-        
-        # Add other components for prompt input, resolution selection, etc.
-        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
-        resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
-        result_output = gr.Image(label="Generated Image", type="pil")
-        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
 
-        # Update the generate button's visibility based on the points check
         get_points_button.click(
-            fn=update_points_and_check,
+            fn=lambda user_chat_id: [get_user_points(user_chat_id), update_generate_button_state(user_chat_id)],
             inputs=[user_chat_id_input],
             outputs=[points_output, generate_button]
         )