Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import requests
+import time
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import firebase_admin
@@ -8,28 +9,23 @@ from firebase_admin import credentials, firestore
 app = Flask(__name__)
 CORS(app) 
 
-
+# --- FIREBASE SETUP ---
 cred = credentials.Certificate("firebase_key.json")
 firebase_admin.initialize_app(cred)
 db = firestore.client()
 
-
+# --- SECRETS ---
 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
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"
-VISION_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct"
-
-
-ROUTER_URL = "https://router.huggingface.co/v1/chat/completions"
-
-VISION_API_URL = f"https://api-inference.huggingface.co/models/{VISION_MODEL}/v1/chat/completions"
+VISION_MODEL = "Qwen/Qwen2-VL-7B-Instruct" # Sabse fast vision model
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Direct Vision Access) is Active! 🔥"})
+    return jsonify({"status": "TS AI Brain (Final Debug Mode) is Active! 🔥"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -38,10 +34,10 @@ def chat():
     image_url = None
 
     if is_form_data or request.files:
-        user_message = request.form.get("message", "Is photo ko dekho.")
+        user_message = request.form.get("message", "Is photo ko read karo.")
         image_file = request.files.get("image")
         if image_file:
-            
+            # 1. 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}
@@ -52,43 +48,56 @@ 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 Fail", "details": str(e)}), 500
+                return jsonify({"error": "Telegram upload fail", "details": str(e)}), 500
     else:
         data = request.json
         user_message = data.get("message", "")
 
-    # 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."}]
 
     headers = {"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"}
 
     try:
         if image_url:
-            
+            # 2. VISION API CALL (Direct Model Path)
             payload = {
                 "model": VISION_MODEL,
                 "messages": [{"role": "user", "content": [{"type": "text", "text": user_message}, {"type": "image_url", "image_url": {"url": image_url}}]}],
-                "max_tokens": 500
+                "max_tokens": 1000
             }
+            # Direct Inference URL for Vision
+            V_URL = f"https://api-inference.huggingface.co/models/{VISION_MODEL}/v1/chat/completions"
+            resp = requests.post(V_URL, headers=headers, json=payload)
             
-            response = requests.post(VISION_API_URL, headers=headers, json=payload)
-            reply = response.json()["choices"]["message"]["content"]
-            history.append({"role": "user", "content": f"[Photo Sent] {user_message}"})
-        else:
+            # Agar model load ho raha ho (503), toh thoda wait karke dobara try karein
+            if resp.status_code == 503:
+                time.sleep(5)
+                resp = requests.post(V_URL, headers=headers, json=payload)
+
+            result = resp.json()
+            if "choices" in result:
+                reply = result["choices"][0]["message"]["content"].strip()
+            else:
+                return jsonify({"error": "Vision Model ne sahi jawab nahi diya", "api_response": result}), 500
             
+            history.append({"role": "user", "content": f"[Screenshot Sent] {user_message}"})
+        else:
+            # 3. TEXT API CALL
             history.append({"role": "user", "content": user_message})
             payload = {"model": TEXT_MODEL, "messages": history, "max_tokens": 1000}
-            response = requests.post(ROUTER_URL, headers=headers, json=payload)
-            reply = response.json()["choices"][0]["message"]["content"]
+            T_URL = "https://router.huggingface.co/v1/chat/completions"
+            resp = requests.post(T_URL, headers=headers, json=payload)
+            reply = resp.json()["choices"]["message"]["content"].strip()
 
         history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
         return jsonify({"reply": reply})
 
     except Exception as e:
-        return jsonify({"error": "API Error", "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)