Update app.py

app.py

@@ -11,7 +11,6 @@ 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 = {
@@ -80,10 +79,10 @@ def get_user_points(user_chat_id):
     try:
         response = requests.get(webhook_url, params=params)
         response.raise_for_status()
-        return response.json().get('points', "Error retrieving points")
+        return response.json().get('points', 0)  # Default to 0 if points not found
     except requests.RequestException as e:
         logger.error(f"Error fetching user points: {e}")
-        return "Failed to retrieve user points"
+        return 0
 
 def extract_user_chat_id_from_url(url):
     parsed_url = urlparse(url)
@@ -121,7 +120,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", visible=False)  # Start with the button hidden
+        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)
@@ -140,9 +139,9 @@ def create_gradio_interface():
         def handle_get_points(user_chat_id):
             points = get_user_points(user_chat_id)
             if points >= 5:
-                return points, gr.Button.update(visible=True)
+                return points, gr.update(visible=True)
             else:
-                return points, gr.Button.update(visible=False)
+                return points, gr.update(visible=False)
 
         def handle_generate_image(prompt, resolution_key, user_chat_id):
             points = get_user_points(user_chat_id)