Create app.py

app.py

@@ -0,0 +1,74 @@
+import os
+import requests
+from flask import Flask, request, jsonify
+from flask_cors import CORS
+import firebase_admin
+from firebase_admin import credentials, firestore
+
+app = Flask(__name__)
+
+CORS(app) 
+
+
+cred = credentials.Certificate("firebase_key.json")
+firebase_admin.initialize_app(cred)
+db = firestore.client()
+
+
+
+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"
+}
+
+
+@app.route('/', methods=['GET'])
+def home():
+    return jsonify({"status": "TS AI Brain is Active and Running! 🔥"})
+
+
+@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
+
+    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})
+
+    hf_data = {
+        "model": "Qwen/Qwen2.5-Coder-32B-Instruct",
+        "messages": history,
+        "max_tokens": 1000
+    }
+
+    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
+            
+        result = response.json()
+        reply = result["choices"][0]["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": str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)
+