Update app.py

app.py

@@ -124,9 +124,9 @@ 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_url):
+def notify_webhook_with_file(user_chat_id, full_file_url):
     webhook_url = f"{webhook_server}/notify_image"
-    payload = {'user_chat_id': user_chat_id, 'file_path': file_url}
+    payload = {'user_chat_id': user_chat_id, 'file_path': full_file_url}
     try:
         response = requests.post(webhook_url, json=payload)
         response.raise_for_status()
@@ -154,15 +154,18 @@ def gradio_interface(prompt, resolution_key, 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."
+        file_path = result[0][0].get('image')
+        if file_path:
+            # Combine base URL with file path to create full URL
+            base_url = "https://mukaist-dalle-4k.hf.space/file="
+            full_file_url = base_url + file_path
+            
+            if os.path.exists(file_path):
+                update_user_points(user_chat_id, -5)
+                notify_webhook_with_file(user_chat_id, full_file_url)  # Notify webhook with full file URL
+                return full_file_url, "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 handle_generate_image(prompt, resolution_key, user_chat_id):
     points = get_user_points(user_chat_id)