Update app.py

app.py

@@ -9,11 +9,14 @@ from upload import upload_image_to_cdn  # Import the new upload function
 
 app = Flask(__name__)
 
+# In-memory chat history storage (for demonstration - use database in production)
+chat_histories = {}
+
 # Fetch page credentials from GitHub on startup
 try:
     github_response = fetch_json_from_github()
     if github_response["success"]:
-        PAGE_CREDENTIALS = {page['page_id']: page['page_token'] for page in github_response["data"]["pages"]}
+        PAGE_CREDENTIALS = {page['page_id']: page for page in github_response["data"]["pages"]}
         print("Successfully loaded page credentials from GitHub.")
     else:
         print(f"Error loading page credentials: {github_response['message']}")
@@ -25,10 +28,6 @@ except Exception as e:
 AI_SERVICE_URL = "https://aoamrnuwara.pythonanywhere.com/send-message"
 AI_SERVICE_TOKEN = "123400"
 
-# Qwen CDN Upload Configuration
-BASE_URL = "https://chat.qwen.ai"
-AUTH_TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6ImRjOGQyYzY4LWZjNmEtNDEwYy05NWZjLWQ5MDBmNTM4ZTMwMiIsImV4cCI6MTc0NjM1OTExMH0.ms8Agbit2ObnTMJyeW_dUbk-tV_ON_dc-RjDKCDLAwA"
-
 @app.route("/webhook", methods=["GET"])
 def verify_webhook():
     mode = request.args.get("hub.mode")
@@ -45,8 +44,10 @@ def handle_message():
     body = request.get_json()
 
     for entry in body["entry"]:
-        page_id = entry["id"]  # Extract page_id from the incoming message
-        page_token = PAGE_CREDENTIALS.get(page_id)
+        page_id = entry["id"]
+        page_config = PAGE_CREDENTIALS.get(page_id, {})
+        page_token = page_config.get('page_token')
+        system_prompt = page_config.get('system', 'Default system prompt')
 
         if not page_token:
             print(f"Page ID {page_id} not found in credentials. Skipping...")
@@ -56,6 +57,16 @@ def handle_message():
             if "message" in messaging_event:
                 sender_id = messaging_event["sender"]["id"]
                 message = messaging_event["message"]
+
+                # Get or create chat history for this user
+                user_history = chat_histories.setdefault(sender_id, {
+                    'history': [],
+                    'system_prompt': system_prompt
+                })
+
+                # Update system prompt if changed
+                user_history['system_prompt'] = system_prompt
+
                 text_content = message.get("text", "")
                 attachments = message.get("attachments", [])
                 image_cdn_urls = []
@@ -69,10 +80,10 @@ def handle_message():
                             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 Qwen CDN
                             cdn_url = upload_image_to_cdn(BASE_URL, AUTH_TOKEN, temp_path)
                             if cdn_url:
@@ -83,6 +94,17 @@ def handle_message():
                             if os.path.exists(temp_path):
                                 os.remove(temp_path)
 
+                # Add user message to history if it's text
+                if text_content:
+                    user_history['history'].append({
+                        "role": "user",
+                        "content": text_content
+                    })
+
+                # Keep only last 5 messages (3 user + 2 assistant to maintain context)
+                while len(user_history['history']) > 5:
+                    user_history['history'].pop(0)
+
                 # Build message content for Qwen
                 user_content = []
                 if text_content:
@@ -91,8 +113,17 @@ def handle_message():
                     user_content.append({"type": "image", "image": cdn_url})
 
                 if user_content:
-                    ai_response = get_qwen_response(user_content)
-                    
+                    ai_response = get_qwen_response(
+                        system_prompt=user_history['system_prompt'],
+                        chat_history=user_history['history']
+                    )
+
+                    # Add AI response to history
+                    user_history['history'].append({
+                        "role": "assistant",
+                        "content": ai_response
+                    })
+
                     # Send response back to user
                     headers = {
                         "Authorization": f"Bearer {AI_SERVICE_TOKEN}",
@@ -101,7 +132,7 @@ def handle_message():
                     payload = {
                         "recipient_id": sender_id,
                         "message_text": ai_response,
-                        "page_access_token": page_token  # Include page-specific token
+                        "page_access_token": page_token
                     }
                     requests.post(AI_SERVICE_URL, json=payload, headers=headers)