Update app.py

app.py

@@ -48,15 +48,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
+        
+        # Log the response details
         logger.debug(f"Response status code: {response.status_code}")
         logger.debug(f"Response content: {response.text}")
-        return response.json()
+        
+        # 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": str(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"}