Update app.py

app.py

@@ -11,6 +11,7 @@ client_image = Client("mukaist/DALLE-4K")
 
 # Retrieve secret token from environment variables
 webhook_server = os.getenv('webhook_server')
+db_path = os.getenv('db_path')  # Ensure you have this set up
 
 # Define resolutions
 resolutions = {
@@ -104,6 +105,14 @@ 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("""
@@ -118,16 +127,7 @@ def create_gradio_interface():
         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)
+        generate_button = gr.Button("Generate", visible=False)
         
         # Arrange user_chat_id_input, points_output, and get_points_button in one row
         with gr.Row():
@@ -141,7 +141,7 @@ def create_gradio_interface():
         
         # Set up interactions
         get_points_button.click(
-            fn=lambda user_chat_id: check_points_and_update(user_chat_id),
+            fn=update_points_and_check,
             inputs=[user_chat_id_input],
             outputs=[points_output, generate_button]
         )
@@ -152,6 +152,19 @@ def create_gradio_interface():
             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,
+            inputs=[user_chat_id_input],
+            outputs=[points_output, generate_button]
+        )
+        
         gr.HTML("""
         <style>
             footer.svelte-1rjryqp {