Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import time
 import base64
+import io
 import random
 import json
 import requests
@@ -11,6 +12,7 @@ from huggingface_hub import InferenceClient
 from zoneinfo import ZoneInfo
 import re
 from playwright.sync_api import sync_playwright
+from PIL import Image
 
 app = Flask(__name__)
 
@@ -320,6 +322,33 @@ def serpapi_search(query: str, location=None, num_results=15):
         print(f"[SEARCH] ❌ Error: {e}")
         return f"Unable to find results for: {query}"
 
+def adaptive_compress_base64_image(image_base64, max_size=1_000_000):
+    header = ""
+    if image_base64.startswith("data:"):
+        header, image_base64 = image_base64.split(",", 1)
+        header += ","
+
+    img = Image.open(io.BytesIO(base64.b64decode(image_base64))).convert("RGB")
+
+    max_dim = 1400
+    quality = 85
+
+    while True:
+        tmp = img.copy()
+        tmp.thumbnail((max_dim, max_dim))
+
+        buf = io.BytesIO()
+        tmp.save(buf, "JPEG", quality=quality, optimize=True)
+        b64 = base64.b64encode(buf.getvalue()).decode()
+
+        if len(b64) <= max_size or max_dim < 400:
+            return header + b64
+
+        if quality > 40:
+            quality -= 10
+        else:
+            max_dim = int(max_dim * 0.8)
+            quality = 85
 # =======================================
 # 💬 Streaming Chat (with API Key Fallback)
 # =======================================
@@ -735,6 +764,7 @@ def chat():
             base64.b64decode(test_b64, validate=True)
         except Exception:
             return Response("Invalid base64 image", mimetype="text/plain", status=400)
+        image_base64 = adaptive_compress_base64_image(image_base64)
         cohere_url = "https://api.cohere.ai/v2/chat"
         payload = {
             "model": "c4ai-aya-vision-32b",