Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 import requests
-import time
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 import firebase_admin
@@ -16,23 +15,17 @@ db = firestore.client()
 
 # --- 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")     
 
-# MODELS LIST (Fallback System)
+# MODELS
 TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-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"
-]
-
-ROUTER_URL = "https://router.huggingface.co/v1/chat/completions"
+VISION_MODEL = "llama-3.2-11b-vision-preview" # Groq ka superfast vision model
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain (Multi-Vision + Telegram) is Online! 🚀"})
+    return jsonify({"status": "TS AI Brain (Groq-Vision Hybrid) is Live! 🔥"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
@@ -41,10 +34,10 @@ def chat():
     image_url = None
 
     if is_form_data or request.files:
-        user_message = request.form.get("message", "Analyze this image.")
+        user_message = request.form.get("message", "Explain this image.")
         image_file = request.files.get("image")
         if image_file:
-            # 1. TELEGRAM UPLOAD (Isme koi panga nahi hai, ye chal raha hai)
+            # 1. TELEGRAM STORAGE (Working perfectly)
             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}
@@ -55,62 +48,43 @@ 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 Fail"}), 500
 
     else:
-        data = request.json
-        user_message = data.get("message", "")
+        user_message = request.json.get("message", "")
 
     # 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 personal AI ho. Hinglish mein baat karo."}]
-
-    headers = {
-        "Authorization": f"Bearer {HF_TOKEN}",
-        "Content-Type": "application/json",
-        "x-wait-for-model": "true"
-    }
+    history = doc.to_dict().get('messages', []) if doc.exists else [{"role": "system", "content": "Tum TS Boss ke AI ho. Hinglish mein baat karo."}]
 
     try:
-        reply = ""
         if image_url:
-            # --- 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 not reply:
-                return jsonify({"error": "Saare Vision Models down hain!", "last_api_error": last_error}), 500
-            
-            history.append({"role": "user", "content": f"[Boss sent an image] {user_message}"})
+            # --- 2. VISION CALL VIA GROQ (No more Router errors!) ---
+            groq_url = "https://api.groq.com/openai/v1/chat/completions"
+            headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "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}}]}],
+            }
+            resp = requests.post(groq_url, headers=headers, json=payload)
+            reply = resp.json()["choices"]["message"]["content"]
+            history.append({"role": "user", "content": f"[Screenshot] {user_message}"})
         else:
-            # --- 3. TEXT ONLY CALL ---
+            # --- 3. TEXT CALL VIA HUGGING FACE (32B) ---
             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)
+            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}
+            resp = requests.post(hf_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": "Critical Crash", "details": str(e)}), 500
+        return jsonify({"error": "Vision API Error", "details": str(e)}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)