Update app.py

app.py

@@ -117,32 +117,40 @@ 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")
+        generate_button.style(height=40)
+        
+        # Function to check points and enable/disable generate button
+        def check_points_and_update(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:
+                return points, gr.Button("Generate", visible=points >= 5)
+        
         # Arrange user_chat_id_input, points_output, and get_points_button in one row
         with gr.Row():
             user_chat_id_input
             points_output
             get_points_button
 
-        # 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")
+        # Arrange generate_button in a separate row
+        with gr.Row():
+            generate_button
         
-        result_output = gr.Image(label="Generated Image", type="pil")
-        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
+        # Set up interactions
+        get_points_button.click(
+            fn=lambda user_chat_id: check_points_and_update(user_chat_id),
+            inputs=[user_chat_id_input],
+            outputs=[points_output, generate_button]
+        )
 
-        # 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]
         )
-
-        get_points_button.click(
-            fn=lambda user_chat_id: get_user_points(user_chat_id),
-            inputs=[user_chat_id_input],
-            outputs=points_output
-        )
         
         gr.HTML("""
         <style>