Update brain.py

brain.py

@@ -6,64 +6,47 @@ from ctransformers import AutoModelForCausalLM
 app = Flask(__name__)
 CORS(app)
 
-print("🛡️ RiShre Security: Initializing Fast Studio Core...")
+print("🛡️ RiShre Security: Initializing Stable Core...")
 
 model = None
-
 try:
-    print("📥 Loading Qwen-Coder via CTransformers (No-Build Mode)...")
-
-    # Hum Mistral ki jagah Qwen-2.5-Coder use karenge kyunki wo 10 files ke liye best hai [cite: 2026-03-01]
+    print("📥 Loading Mistral-7B (No-Build Mode)...")
+    # Mistral v0.3 is 100% compatible with ctransformers
     model = AutoModelForCausalLM.from_pretrained(
-        "Qwen/Qwen2.5-Coder-7B-Instruct-GGUF",
-        model_file="qwen2.5-coder-7b-instruct-q4_k_m.gguf",
-        model_type="gpt2",   # Qwen ke liye gpt2 best chalta hai [cite: 2026-02-19]
-        gpu_layers=0,        # CPU only
-        context_length=8192, # 10 files handle karne ke liye [cite: 2026-02-19]
+        "bartowski/Mistral-7B-Instruct-v0.3-GGUF",
+        model_file="Mistral-7B-Instruct-v0.3-Q4_K_M.gguf",
+        model_type="mistral",
+        context_length=8192, # For handling multiple files [cite: 2026-02-19]
         threads=4
     )
-
     print("🚀 RiShre AI: Core Online & Ready.")
-
 except Exception as e:
     print(f"❌ Ignition Failed: {e}")
-    model = None
 
 @app.route("/api/chat", methods=["POST"])
 def chat():
-    if model is None:
-        return jsonify({"error": "Core Offline"}), 500
-
+    if model is None: return jsonify({"error": "Core Offline"}), 500
     try:
         data = request.get_json()
         user_msg = data.get("message", "")
-        # Frontend se files ka context bhi aayega yahan [cite: 2026-03-01]
         context_files = data.get("files", "")
 
         system_prompt = (
-            "You are RiShre AI, an elite coder created by Badge94 on 17 March 2026. "
-            "Identify the file extension like .tsx, .html, .py for every code block. "
-            "Think about cross-file dependencies."
+            "You are RiShre AI, created by Badge94 on 17 March 2026. "
+            "Identify the file origin like .ts, .tsx, .html, .css for all code blocks. "
+            "Help the user build their applications."
         )
 
-        # Chat format for Qwen
-        prompt = f"System: {system_prompt}\nContext: {context_files}\nUser: {user_msg}\nAssistant:"
-        
-        response = model(
-            prompt,
-            max_new_tokens=1024,
-            temperature=0.3,
-            top_p=0.9
-        )
+        prompt = f"<s>[INST] {system_prompt}\n\nContext Files:\n{context_files}\n\nTask: {user_msg} [/INST]"
         
+        # Generating response
+        response = model(prompt, max_new_tokens=1024, temperature=0.7)
         return jsonify({"text": response})
-
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
 @app.route("/")
-def health():
-    return "RiShre AI: Status 100% Protected"
+def health(): return "RiShre AI: 100% Protected"
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)