Update app.py

app.py

@@ -1,6 +1,8 @@
 from flask import Flask, request
 import requests
 import json
+import os
+import uuid
 from db import fetch_json_from_github  # Import the GitHub JSON fetcher
 from qwen import get_qwen_response  # Import the Qwen response function
 
@@ -22,6 +24,23 @@ except Exception as e:
 AI_SERVICE_URL = "https://aoamrnuwara.pythonanywhere.com/send-message"
 AI_SERVICE_TOKEN = "123400"
 
+# CDN Upload Configuration
+UPLOAD_API_URL = "https://api.braininc.net/api/me/upload/studios"
+AUTH_TOKEN = "72ec00483379076f580eb8126f29da802a5140c3"
+
+def upload_image(file_path):
+    """Upload image to get CDN URL"""
+    headers = {"authorization": f"token {AUTH_TOKEN}"}
+    try:
+        with open(file_path, "rb") as f:
+            files = {"file": (os.path.basename(file_path), f, "image/png")}
+            response = requests.post(UPLOAD_API_URL, headers=headers, files=files)
+            if response.status_code == 201:
+                return response.json().get("cdn_url", "")
+    except Exception as e:
+        print(f"Upload error: {str(e)}")
+    return ""
+
 @app.route("/webhook", methods=["GET"])
 def verify_webhook():
     mode = request.args.get("hub.mode")
@@ -46,24 +65,57 @@ def handle_message():
             continue
 
         for messaging_event in entry["messaging"]:
-            if "message" in messaging_event and "text" in messaging_event["message"]:
+            if "message" in messaging_event:
                 sender_id = messaging_event["sender"]["id"]
-                message_text = messaging_event["message"]["text"]
-
-                # Get AI response from Qwen
-                ai_response = get_qwen_response(message_text)
-
-                # Send response to messaging service
-                headers = {
-                    "Authorization": f"Bearer {AI_SERVICE_TOKEN}",
-                    "Content-Type": "application/json"
-                }
-                payload = {
-                    "recipient_id": sender_id,
-                    "message_text": ai_response,
-                    "page_access_token": page_token  # Include page-specific token
-                }
-                requests.post(AI_SERVICE_URL, json=payload, headers=headers)
+                message = messaging_event["message"]
+                text_content = message.get("text", "")
+                attachments = message.get("attachments", [])
+                image_cdn_urls = []
+
+                # Process image attachments
+                for attachment in attachments:
+                    if attachment["type"] == "image":
+                        image_url = attachment["payload"]["url"]
+                        try:
+                            # Download image from Messenger
+                            img_response = requests.get(image_url)
+                            img_response.raise_for_status()
+                            temp_path = f"temp_{uuid.uuid4()}.jpg"
+                            
+                            with open(temp_path, "wb") as f:
+                                f.write(img_response.content)
+                            
+                            # Upload to CDN
+                            cdn_url = upload_image(temp_path)
+                            if cdn_url:
+                                image_cdn_urls.append(cdn_url)
+                        except Exception as e:
+                            print(f"Image processing error: {str(e)}")
+                        finally:
+                            if os.path.exists(temp_path):
+                                os.remove(temp_path)
+
+                # Build message content for Qwen
+                user_content = []
+                if text_content:
+                    user_content.append({"type": "text", "text": text_content})
+                for cdn_url in image_cdn_urls:
+                    user_content.append({"type": "image", "image": cdn_url})
+
+                if user_content:
+                    ai_response = get_qwen_response(user_content)
+                    
+                    # Send response back to user
+                    headers = {
+                        "Authorization": f"Bearer {AI_SERVICE_TOKEN}",
+                        "Content-Type": "application/json"
+                    }
+                    payload = {
+                        "recipient_id": sender_id,
+                        "message_text": ai_response,
+                        "page_access_token": page_token  # Include page-specific token
+                    }
+                    requests.post(AI_SERVICE_URL, json=payload, headers=headers)
 
     return "OK", 200