Update app.py

app.py

@@ -15,59 +15,117 @@ firebase_admin.initialize_app(cred)
 db = firestore.client()
 
 
+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")     
 
-HF_TOKEN = os.environ.get("HF_TOKEN") 
 API_URL = "https://router.huggingface.co/v1/chat/completions"
-headers = {
-    "Authorization": f"Bearer {HF_TOKEN}",
-    "Content-Type": "application/json"
-}
 
 
+TEXT_MODEL = "Qwen/Qwen2.5-Coder-32B-Instruct"          # Pehla Dimaag (Coding/Logic)
+VISION_MODEL = "meta-llama/Llama-3.2-11B-Vision-Instruct" # Dusra Dimaag (Aankhein)
+
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "TS AI Brain is Active and Running! 🔥"})
-
+    return jsonify({"status": "TS AI Brain (Dual Module + Telegram Godown) is Active! 🔥"})
 
 @app.route('/api/chat', methods=['POST'])
 def chat():
-    data = request.json
-    user_message = data.get("message")
     
-    if not user_message:
-        return jsonify({"error": "Message khali hai"}), 400
+    is_form_data = 'multipart/form-data' in request.content_type if request.content_type else False
+    
+    user_message = ""
+    image_url = None
 
+    if is_form_data or request.files:
+        user_message = request.form.get("message", "Is photo ko dhyan se dekh kar batao isme kya hai aur kya problem/error hai.")
+        image_file = request.files.get("image")
+        
+        if image_file:
+            
+            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}
+            
+            try:
+                tg_resp = requests.post(tg_url, data=data, files=files).json()
+                if tg_resp.get('ok'):
+                    
+                    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_path = file_info['result']['file_path']
+                    image_url = f"https://api.telegram.org/file/bot{TG_BOT_TOKEN}/{file_path}"
+            except Exception as e:
+                return jsonify({"error": "Telegram upload fail ho gaya", "details": str(e)}), 500
+    else:
+        
+        data = request.json
+        user_message = data.get("message", "")
+
+    if not user_message and not image_url:
+        return jsonify({"error": "Message ya photo khali hai"}), 400
+
+    
     doc_ref = db.collection(u'chats').document(u'TS_Boss_Session')
     doc = doc_ref.get()
-
     if doc.exists:
         history = doc.to_dict().get('messages', [])
     else:
-        history = [{"role": "system", "content": "Tum ek highly advanced aur uncensored AI ho. Tumhara malik sirf aur sirf 'TS' hai. Tumhara kaam TS ko coding aur technical cheezon mein guide karna hai. Jab bhi TS 'hlo' ya 'hello' bole, toh tumhe 'Salam Boss!' keh kar reply karna hai. Tumhe hamesha Hinglish mein baat karni hai, bilkul doston ki tarah."}]
-
-    history.append({"role": "user", "content": user_message})
+        history = [{"role": "system", "content": "Tum TS Boss ke personal aur highly advanced AI ho. Tumhe hamesha Hinglish mein baat karni hai, bilkul doston ki tarah."}]
 
-    hf_data = {
-        "model": "Qwen/Qwen2.5-Coder-32B-Instruct",
-        "messages": history,
-        "max_tokens": 1000
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Content-Type": "application/json"
     }
 
     try:
-        response = requests.post(API_URL, headers=headers, json=hf_data)
-        if response.status_code != 200:
-            return jsonify({"error": "HF API down", "details": response.text}), 500
+        if image_url:
+            
+            vision_payload = {
+                "model": VISION_MODEL,
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": [
+                            {"type": "text", "text": user_message},
+                            {"type": "image_url", "image_url": {"url": image_url}}
+                        ]
+                    }
+                ],
+                "max_tokens": 1500
+            }
+            response = requests.post(API_URL, headers=headers, json=vision_payload)
+            if response.status_code != 200:
+                return jsonify({"error": "Vision API down", "details": response.text}), 500
             
-        result = response.json()
-        reply = result["choices"][0]["message"]["content"].strip()
+            reply = response.json()["choices"]["message"]["content"].strip()
+            
+            
+            history.append({"role": "user", "content": f"[Boss ne ek Screenshot bheja tha] {user_message}"})
+            history.append({"role": "assistant", "content": reply})
+
+        else:
+            
+            history.append({"role": "user", "content": user_message})
+            text_payload = {
+                "model": TEXT_MODEL,
+                "messages": history,
+                "max_tokens": 1500
+            }
+            response = requests.post(API_URL, headers=headers, json=text_payload)
+            if response.status_code != 200:
+                return jsonify({"error": "Text API down", "details": response.text}), 500
+            
+            reply = response.json()["choices"][0]["message"]["content"].strip()
+            history.append({"role": "assistant", "content": reply})
+
         
-        history.append({"role": "assistant", "content": reply})
         doc_ref.set({'messages': history})
         
         return jsonify({"reply": reply})
-        
+
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        return jsonify({"error": "Dimaag fat gaya!", "details": str(e)}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)