Update app.py

app.py

@@ -6,7 +6,7 @@ import firebase_admin
 from firebase_admin import credentials, firestore
 
 app = Flask(__name__)
-CORS(app) 
+CORS(app)
 
 # --- FIREBASE SETUP ---
 try:
@@ -15,21 +15,21 @@ try:
         firebase_admin.initialize_app(cred)
     db = firestore.client()
 except Exception as e:
-    print(f"Firebase Error: {e}")
+    print("Firebase error:", e)
 
 # --- SECRETS ---
 HF_TOKEN = os.environ.get("HF_TOKEN")
-GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
-TG_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN") 
-TG_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID")     
+OPENROUTER_API_KEY = os.environ.get("OPENROUTER_API_KEY")
+TG_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN")
+TG_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID")
 
-# MODELS
+# --- MODELS ---
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "meta-llama/llama-4-scout-17b-16e-instruct"
+VISION_MODEL = "qwen/qwen2.5-vl-7b-instruct"
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Crash-Proof Mode) is Active! 🚀"})
+    return jsonify({"status": "TS AI (OpenRouter Vision + HF Brain) 🚀"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -38,75 +38,114 @@ def chat():
         user_message = ""
         image_url = None
 
-        # 1. READ REQUEST
+        # ===== INPUT HANDLE =====
         if is_form_data:
-            user_message = request.form.get("message", "Is photo mein kya error hai?")
+            user_message = request.form.get("message", "Explain this image")
             if 'image' in request.files:
                 image_file = request.files['image']
-                
-                # 2. TELEGRAM UPLOAD WITH STRICT CHECK
+
+                # --- TELEGRAM UPLOAD ---
                 tg_url = f"https://api.telegram.org/bot{TG_BOT_TOKEN}/sendPhoto"
                 files = {'photo': (image_file.filename, image_file.read(), image_file.mimetype)}
                 data = {'chat_id': TG_CHAT_ID}
+
                 tg_resp = requests.post(tg_url, data=data, files=files).json()
-                
+
                 if not tg_resp.get('ok'):
-                    # Agar Telegram ne photo block ki, toh yahi error aayega
-                    return jsonify({"error": "Telegram upload fail", "details": tg_resp}), 500
-                
+                    return jsonify({"error": "Telegram upload failed", "details": tg_resp}), 500
+
                 file_id = tg_resp['result']['photo'][-1]['file_id']
-                file_info = requests.get(f"https://api.telegram.org/bot{TG_BOT_TOKEN}/getFile?file_id={file_id}").json()
+                file_info = requests.get(
+                    f"https://api.telegram.org/bot{TG_BOT_TOKEN}/getFile?file_id={file_id}"
+                ).json()
+
                 image_url = f"https://api.telegram.org/file/bot{TG_BOT_TOKEN}/{file_info['result']['file_path']}"
+
         else:
             data = request.get_json(silent=True) or {}
             user_message = data.get("message", "")
 
-        # 3. HISTORY LOAD
+        # ===== LOAD HISTORY =====
         doc_ref = db.collection(u'chats').document(u'TS_Boss_Session')
         doc = doc_ref.get()
-        history = doc.to_dict().get('messages', []) if doc.exists else [{"role": "system", "content": "Tum TS Boss ke AI ho. Hinglish mein baat karo."}]
+        history = doc.to_dict().get('messages', []) if doc.exists else [
+            {"role": "system", "content": "Tum TS Boss ke AI ho. Hinglish mein baat karo."}
+        ]
 
         reply = ""
 
-        # 4. VISION CALL (Agar Image Hai)
+        # ===================================================
+        # 👁️ VISION (OPENROUTER)
+        # ===================================================
         if image_url:
-            groq_url = "https://api.groq.com/openai/v1/chat/completions"
-            headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
+            vision_url = "https://openrouter.ai/api/v1/chat/completions"
+
+            headers = {
+                "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+                "HTTP-Referer": "https://your-app.com",
+                "X-Title": "TS AI",
+                "Content-Type": "application/json"
+            }
+
             payload = {
                 "model": VISION_MODEL,
-                "messages": [{"role": "user", "content": [{"type": "text", "text": user_message}, {"type": "image_url", "image_url": {"url": image_url}}]}],
-                "max_tokens": 1024
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [
+                            {"type": "text", "text": user_message},
+                            {"type": "image_url", "image_url": {"url": image_url}}
+                        ]
+                    }
+                ],
+                "max_tokens": 800
             }
-            resp = requests.post(groq_url, headers=headers, json=payload).json()
-            
-            # STRICT RESPONSE PARSING (Ab crash nahi hoga)
+
+            resp = requests.post(vision_url, headers=headers, json=payload).json()
+
+            # SAFE PARSING
             if isinstance(resp, dict) and "choices" in resp:
-                reply = resp["choices"]["message"]["content"]
+                reply = resp["choices"][0]["message"]["content"]
             else:
-                return jsonify({"error": "Groq API Error", "raw_response": resp}), 500
-        
-        # 5. TEXT CALL (Sirf Text Hai Toh)
+                return jsonify({"error": "OpenRouter Vision failed", "raw": resp}), 500
+
+            history.append({"role": "user", "content": f"[Image] {user_message}"})
+
+        # ===================================================
+        # 💬 TEXT (HUGGING FACE)
+        # ===================================================
         else:
             history.append({"role": "user", "content": user_message})
+
             hf_url = "https://router.huggingface.co/v1/chat/completions"
-            headers = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
-            payload = {"model": TEXT_MODEL, "messages": history}
+
+            headers = {
+                "Authorization": f"Bearer {HF_TOKEN}",
+                "Content-Type": "application/json"
+            }
+
+            payload = {
+                "model": TEXT_MODEL,
+                "messages": history,
+                "max_tokens": 1000
+            }
+
             resp = requests.post(hf_url, headers=headers, json=payload).json()
-            
-            # STRICT RESPONSE PARSING (Yehi tha asli culprit pichli baar)
+
             if isinstance(resp, dict) and "choices" in resp:
                 reply = resp["choices"][0]["message"]["content"]
             else:
-                return jsonify({"error": "Hugging Face Text API Error", "raw_response": resp}), 500
+                return jsonify({"error": "HF failed", "raw": resp}), 500
 
-        # 6. SAVE AND RETURN
+        # ===== SAVE =====
         history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
+
         return jsonify({"reply": reply})
 
     except Exception as e:
-        return jsonify({"error": "Python Code Crash", "details": str(e)}), 500
+        return jsonify({"error": "System Crash", "details": str(e)}), 500
+
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
-    
+    app.run(host='0.0.0.0', port=7860)