Update app.py

app.py

@@ -19,16 +19,20 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 TG_BOT_TOKEN = os.environ.get("TELEGRAM_BOT_TOKEN") 
 TG_CHAT_ID = os.environ.get("TELEGRAM_CHAT_ID")     
 
-# MODELS (Router friendly models)
+# MODELS LIST (Fallback System)
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"
+# Teen alag providers ke vision models
+VISION_MODELS = [
+    "meta-llama/Llama-3.2-11B-Vision-Instruct",
+    "Qwen/Qwen2-VL-7B-Instruct",
+    "microsoft/Phi-3.5-vision-instruct"
+]
 
-# GLOBAL ROUTER URL (Jo HF ne manga hai)
 ROUTER_URL = "https://router.huggingface.co/v1/chat/completions"
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Official Router Mode) is Active! 🔥"})
+    return jsonify({"status": "TS AI Brain (Multi-Vision + Telegram) is Online! 🚀"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -37,10 +41,10 @@ def chat():
     image_url = None
 
     if is_form_data or request.files:
-        user_message = request.form.get("message", "Explain this image.")
+        user_message = request.form.get("message", "Analyze this image.")
         image_file = request.files.get("image")
         if image_file:
-            # 1. TELEGRAM UPLOAD (STORAGE)
+            # 1. TELEGRAM UPLOAD (Isme koi panga nahi hai, ye chal raha hai)
             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}
@@ -51,7 +55,7 @@ def chat():
                     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']}"
             except Exception as e:
-                return jsonify({"error": "Telegram upload fail", "details": str(e)}), 500
+                return jsonify({"error": "Telegram Fail", "details": str(e)}), 500
 
     else:
         data = request.json
@@ -60,52 +64,53 @@ def chat():
     # 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 personal AI ho. Hinglish mein baat karo."}]
 
     headers = {
         "Authorization": f"Bearer {HF_TOKEN}",
         "Content-Type": "application/json",
-        "x-wait-for-model": "true" # Ye HF ko bolta hai ki model load hone tak ruko!
+        "x-wait-for-model": "true"
     }
 
     try:
+        reply = ""
         if image_url:
-            # --- VISION CALL VIA ROUTER ---
-            payload = {
-                "model": VISION_MODEL,
-                "messages": [
-                    {
-                        "role": "user", 
-                        "content": [
-                            {"type": "text", "text": user_message},
-                            {"type": "image_url", "image_url": {"url": image_url}}
-                        ]
-                    }
-                ],
-                "max_tokens": 1000
-            }
-            resp = requests.post(ROUTER_URL, headers=headers, json=payload)
-            result = resp.json()
+            # --- 2. MULTI-MODEL VISION FALLBACK ---
+            last_error = ""
+            for model_name in VISION_MODELS:
+                payload = {
+                    "model": model_name,
+                    "messages": [{"role": "user", "content": [{"type": "text", "text": user_message}, {"type": "image_url", "image_url": {"url": image_url}}]}],
+                    "max_tokens": 800
+                }
+                resp = requests.post(ROUTER_URL, headers=headers, json=payload)
+                result = resp.json()
+                
+                if "choices" in result:
+                    reply = result["choices"]["message"]["content"]
+                    break # Agar ek model ne jawab de diya, toh loop rok do
+                else:
+                    last_error = result
+                    print(f"Model {model_name} failed, trying next...")
             
-            if "choices" in result:
-                reply = result["choices"]["message"]["content"]
-            else:
-                return jsonify({"error": "Router Vision Fail", "api_response": result}), 500
+            if not reply:
+                return jsonify({"error": "Saare Vision Models down hain!", "last_api_error": last_error}), 500
             
-            history.append({"role": "user", "content": f"[Screenshot] {user_message}"})
+            history.append({"role": "user", "content": f"[Boss sent an image] {user_message}"})
         else:
-            # --- TEXT CALL VIA ROUTER ---
+            # --- 3. TEXT ONLY CALL ---
             history.append({"role": "user", "content": user_message})
             payload = {"model": TEXT_MODEL, "messages": history, "max_tokens": 1000}
             resp = requests.post(ROUTER_URL, headers=headers, json=payload)
             reply = resp.json()["choices"][0]["message"]["content"]
 
+        # Save and return
         history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
         return jsonify({"reply": reply})
 
     except Exception as e:
-        return jsonify({"error": "System Crash", "details": str(e)}), 500
+        return jsonify({"error": "Critical Crash", "details": str(e)}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)