Update app.py

app.py

@@ -1,20 +1,17 @@
 from flask import Flask, request, Response, jsonify, stream_with_context
 from flask_cors import CORS
-import os, json, httpx, base64
+import os, json, httpx
 import config
+import base64
+from io import BytesIO
+from PIL import Image
+
 
 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
@@ -28,40 +25,33 @@ 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
-
-"
+        yield "data: Error: Missing API key\n\n"
         return
         
     try:
@@ -84,24 +74,16 @@ def stream_openrouter(q: str):
                                 content = json_data['choices'][0].get('delta', {}).get('content')
                                 print(content)
                                 if content:
-                                    yield f"data: {json.dumps(content)}
-
-"
+                                    yield f"data: {json.dumps(content)}\n\n"
                         except json.JSONDecodeError:
                             continue
                 
-                yield "data: [DONE]
-
-"
+                yield "data: [DONE]\n\n"
                 
     except httpx.HTTPError as e:
-        yield f"data: Error: {str(e)}
-
-"
+        yield f"data: Error: {str(e)}\n\n"
     except Exception as e:
-        yield f"data: Unexpected error: {str(e)}
-
-"
+        yield f"data: Unexpected error: {str(e)}\n\n"
 
 @app.post("/ask")
 def ask():
@@ -109,11 +91,12 @@ 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(
@@ -135,34 +118,22 @@ def askimage():
     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
+        # Read image file and convert to base64
         image_bytes = image_file.read()
-        
-        # 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}"
+        image_format = image_file.content_type.split('/')[-1]
+        if image_format == 'jpeg':
+            image_format = 'jpg'
         
         # Create data URL
-        image_data_url = f"data:{image_format};base64,{base64_image}"
+        image_data_url = f"data:{image_file.content_type};base64,{base64_image}"
         
-        # Build prompt in existing format
+        # Build prompt in your existing format
         prompt = "Image" + "%" + image_data_url + "%" + mode + "%" + lang
-        print("Image upload received:", image_file.filename)
+        print(prompt)
         
         return Response(
             stream_with_context(stream_openrouter(prompt)),
@@ -173,7 +144,6 @@ def askimage():
             }
         )
     except Exception as e:
-        print(f"Error processing image: {str(e)}")
         return jsonify({"error": str(e)}), 500
 
 if __name__ == "__main__":