Update app.py

app.py

@@ -124,16 +124,16 @@ def notify_webhook(user_chat_id):
     except requests.RequestException as e:
         logger.error(f"Error sending webhook notification: {e}")
 
-def notify_webhook_with_file(user_chat_id, file_path):
+def notify_webhook_with_file(user_chat_id, file_url):
     webhook_url = f"{webhook_server}/notify_image"
-    payload = {'user_chat_id': user_chat_id, 'file_path': file_path}
+    payload = {'user_chat_id': user_chat_id, 'file_path': file_url}
     try:
         response = requests.post(webhook_url, json=payload)
         response.raise_for_status()
         logger.info(f"Webhook notification sent successfully for user {user_chat_id}. Response: {response.status_code}")
     except requests.RequestException as e:
-        logger.error(f"Error sending webhook notification with file path: {e}")
-        
+        logger.error(f"Error sending webhook notification with file URL: {e}"))
+'''        
 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 +147,23 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
             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."
+'''
+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."
 
+    result = generate_image(prompt, resolution_key)
+    if result and result[0]:
+        file_url = result[0][0].get('image')  # Ensure this is a URL
+        if file_url and file_url.startswith('http'):
+            update_user_points(user_chat_id, -5)
+            notify_webhook_with_file(user_chat_id, file_url)  # Notify webhook with file URL
+            return file_url, "The image was generated successfully."
+        return None, "The image URL is not available. Please try again later."
+    return None, "There was an error processing your photo. Please try again later."
+
+
+    
 def handle_generate_image(prompt, resolution_key, user_chat_id):
     points = get_user_points(user_chat_id)
     if isinstance(points, str):