Update app.py

app.py

@@ -124,6 +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):
+    webhook_url = f"{webhook_server}/notify_image"
+    payload = {'user_chat_id': user_chat_id, 'file_path': file_path}
+    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}")
+        
 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."
@@ -133,6 +143,7 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
         file_path = result[0][0].get('image')
         if file_path and os.path.exists(file_path):
             update_user_points(user_chat_id, -5)
+            notify_webhook_with_file(user_chat_id, file_path)  # Notify webhook with file path
             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."