Update app.py

app.py

@@ -1,14 +1,20 @@
 from flask import Flask, request, Response, jsonify, stream_with_context
 from flask_cors import CORS
-import os, json, httpx
+import os, json, httpx, base64
 import config
 
-
 app = Flask(__name__)
 CORS(app)
 
 OPENROUTER_API_KEY = os.getenv("API_KEYS")
 
+# File upload configuration
+ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp'}
+MAX_FILE_SIZE = 10 * 1024 * 1024  # 10MB
+
+def allowed_file(filename):
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+
 @app.get("/health")
 def health():
     return jsonify({"status": "ready"}), 200
@@ -22,33 +28,40 @@ def _headers():
     }
 
 def _payload(q: str):
-    data=q.split("%")
-    content=""
-    system=""
-    if data[0]=="Text":
-        content=data[1]
+    data = q.split("%")
+    content = ""
+    system = ""
+    
+    if data[0] == "Text":
+        content = data[1]
     else:
-        content={
-          "type": "image_url",
-          "image_url": {
-            "url": data[1]
-          }
+        content = {
+            "type": "image_url",
+            "image_url": {
+                "url": data[1]
+            }
         }
-    info=config.config
-    system=info[0][data[3]]
+    
+    info = config.config
+    system = info[0][data[3]]
     
     return {
         "model": "meta-llama/llama-4-maverick-17b-128e-instruct",
         "stream": True,
-        "messages": [{"role":"system","content":system},{"role": "user", "content": content}],
+        "messages": [
+            {"role": "system", "content": system},
+            {"role": "user", "content": content}
+        ],
         "temperature": 0.7,
-        "top_p":0.9,
+        "top_p": 0.9,
         "max_tokens": 999,
     }
 
 def stream_openrouter(q: str):
     if not OPENROUTER_API_KEY:
-        yield "data: Error: Missing API key\n\n"
+        yield "data: Error: Missing API key
+
+"
         return
         
     try:
@@ -71,16 +84,24 @@ def stream_openrouter(q: str):
                                 content = json_data['choices'][0].get('delta', {}).get('content')
                                 print(content)
                                 if content:
-                                    yield f"data: {json.dumps(content)}\n\n"
+                                    yield f"data: {json.dumps(content)}
+
+"
                         except json.JSONDecodeError:
                             continue
                 
-                yield "data: [DONE]\n\n"
+                yield "data: [DONE]
+
+"
                 
     except httpx.HTTPError as e:
-        yield f"data: Error: {str(e)}\n\n"
+        yield f"data: Error: {str(e)}
+
+"
     except Exception as e:
-        yield f"data: Unexpected error: {str(e)}\n\n"
+        yield f"data: Unexpected error: {str(e)}
+
+"
 
 @app.post("/ask")
 def ask():
@@ -88,12 +109,11 @@ def ask():
     q = payload.get("question", "").strip()
     lang = payload.get("selectedLanguage", "").strip()
     mode = payload.get("selectedMode", "").strip()
-    prompt=""
-    q2lang=""
+    
     if not q:
         q = "Say hello in one short sentence."
 
-    prompt="Text"+"%"+q+"%"+mode+"%"+lang
+    prompt = "Text" + "%" + q + "%" + mode + "%" + lang
     print(prompt)
         
     return Response(
@@ -107,26 +127,54 @@ def ask():
 
 @app.post("/askimage")
 def askimage():
-    payload = request.get_json(silent=True) or {}
-    q = payload.get("question", "")
-    lang = payload.get("selectedLanguage", "").strip()
-    mode = payload.get("selectedMode", "").strip()
-    prompt=""
-    q2lang=""
-    if not q:
-        q = "Say hello in one short sentence."
-    prompt="Image"+"%"+q+"%"+mode+"%"+lang
-    print(prompt)
-
+    # Get form data instead of JSON
+    image_file = request.files.get('image')
+    lang = request.form.get('selectedLanguage', '').strip()
+    mode = request.form.get('selectedMode', '').strip()
+    
+    if not image_file:
+        return jsonify({"error": "No image provided"}), 400
+    
+    # Validate file type
+    if not allowed_file(image_file.filename):
+        return jsonify({"error": "Invalid file type. Allowed: png, jpg, jpeg, gif, webp"}), 400
+    
+    try:
+        # Read image file
+        image_bytes = image_file.read()
         
-    return Response(
-        stream_with_context(stream_openrouter(prompt)),
-        mimetype="text/event-stream",
-        headers={
-            "Cache-Control": "no-cache",
-            "X-Accel-Buffering": "no"
-        }
-    )
+        # Check file size
+        if len(image_bytes) > MAX_FILE_SIZE:
+            return jsonify({"error": "File too large. Max size: 10MB"}), 400
+        
+        # Convert to base64
+        base64_image = base64.b64encode(image_bytes).decode('utf-8')
+        
+        # Determine image format
+        image_format = image_file.content_type
+        if not image_format:
+            # Fallback based on extension
+            ext = image_file.filename.rsplit('.', 1)[1].lower()
+            image_format = f"image/{ext}"
+        
+        # Create data URL
+        image_data_url = f"data:{image_format};base64,{base64_image}"
+        
+        # Build prompt in existing format
+        prompt = "Image" + "%" + image_data_url + "%" + mode + "%" + lang
+        print("Image upload received:", image_file.filename)
+        
+        return Response(
+            stream_with_context(stream_openrouter(prompt)),
+            mimetype="text/event-stream",
+            headers={
+                "Cache-Control": "no-cache",
+                "X-Accel-Buffering": "no"
+            }
+        )
+    except Exception as e:
+        print(f"Error processing image: {str(e)}")
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860, debug=False)