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", elem_id="generate-button", visible=True)
+        generate_button = gr.Button("Generate")
         
         result_output = gr.Image(label="Generated Image", type="pil")
         message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
@@ -132,15 +132,21 @@ def create_gradio_interface():
             points_output
             get_points_button
         
-        # Ensure 'Generate' button is disabled initially
-        generate_button.update(visible=False)
-
-        def update_generate_button_state(user_chat_id):
+        with gr.Row():
+            prompt_input
+            resolution_dropdown
+            generate_button
+        
+        def update_points_and_generate_button(user_chat_id):
             points = get_user_points(user_chat_id)
             if points >= 5:
-                return gr.Button.update(visible=True)
+                return points, gr.Button.update(visible=True)
             else:
-                return gr.Button.update(visible=False)
+                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
         generate_button.click(
@@ -150,7 +156,7 @@ def create_gradio_interface():
         )
 
         get_points_button.click(
-            fn=lambda user_chat_id: [get_user_points(user_chat_id), update_generate_button_state(user_chat_id)],
+            fn=handle_get_points,
             inputs=[user_chat_id_input],
             outputs=[points_output, generate_button]
         )