Update app.py

app.py

@@ -3,7 +3,6 @@ from gradio_client import Client
 import os
 import logging
 import requests
-from urllib.parse import urlparse, parse_qs
 
 webhook_server = os.getenv('webhook_server')
 
@@ -48,39 +47,32 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
 
 def request_otp(user_chat_id):
     try:
-        # Make the request to the /request_otp endpoint
         response = requests.post(f"{webhook_server}/request_otp", json={"user_chat_id": user_chat_id})
-        response.raise_for_status()  # Ensure we raise an exception for HTTP errors
+        response.raise_for_status()
         
-        # Log the response details
         logger.debug(f"Response status code: {response.status_code}")
         logger.debug(f"Response content: {response.text}")
         
-        # Parse and return the JSON response
         response_data = response.json()
         
-        # Check if the response contains an error
         if 'error' in response_data:
             logger.error(f"Error response from /request_otp: {response_data['error']}")
             return {"success": False, "error": response_data.get("error", "Unknown error")}
         
-        # Return success response
         return {"success": response_data.get("success", False)}
     
     except requests.RequestException as e:
-        # Log request exceptions and return an error response
         logger.error(f"Request exception: {e}")
         return {"success": False, "error": "Request exception occurred"}
     
     except ValueError as e:
-        # Handle JSON decode errors
         logger.error(f"JSON decode error: {e}")
         return {"success": False, "error": "Invalid JSON response"}
 
 def verify_otp(user_chat_id, otp):
     try:
         response = requests.post(f"{webhook_server}/verify_otp", json={"user_chat_id": user_chat_id, "otp": otp})
-        response.raise_for_status()  # Raise an exception for HTTP errors
+        response.raise_for_status()
         return response.json().get("valid", False)
     except requests.RequestException as e:
         logger.error(f"Request exception: {e}")
@@ -90,11 +82,9 @@ def verify_otp(user_chat_id, otp):
         return False
 
 def retrieve_user_points(user_chat_id, otp):
-    # Verify OTP via Flask server
     if not verify_otp(user_chat_id, otp):
         return "Invalid or expired OTP. Please request a new OTP."
 
-    # Retrieve user points via Flask server
     response = requests.get(f"{webhook_server}/get_points", params={"user_chat_id": user_chat_id})
     if response.status_code == 200:
         data = response.json()
@@ -134,11 +124,6 @@ def notify_webhook(user_chat_id):
     except requests.RequestException as e:
         logger.error(f"Error sending webhook notification: {e}")
 
-def extract_user_chat_id_from_url(url):
-    parsed_url = urlparse(url)
-    params = parse_qs(parsed_url.query)
-    return params.get('user_chat_id', ["Guest"])[0]
-
 def gradio_interface(prompt, resolution_key, user_chat_id):
     if not user_chat_id.strip():
         return None, "User Chat ID cannot be blank. Please provide a valid User Chat ID."
@@ -147,7 +132,7 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
     if result and result[0]:
         file_path = result[0][0].get('image')
         if file_path and os.path.exists(file_path):
-            update_user_points(user_chat_id, -5)  # Deduct points for each image generated
+            update_user_points(user_chat_id, -5)
             return file_path, "The image was generated successfully."
         return None, "The image file is not available. Please try again later."
     return None, "There was an error processing your photo. Please try again later."
@@ -229,11 +214,8 @@ def create_gradio_interface():
     return interface
 
 def launch_gradio():
-    url = 'https://measmonysuon-imagen.hf.space/?user_chat_id=00000000'
-    user_chat_id = extract_user_chat_id_from_url(url)
-    
     interface = create_gradio_interface()
-    interface.launch()
+    interface.launch(show_error=True)  # Enable verbose error reporting
 
 if __name__ == '__main__':
     launch_gradio()