Update brain.py

brain.py

@@ -1,34 +1,37 @@
 import os
+import json
 from flask import Flask, request, jsonify
 from flask_cors import CORS
-from ctransformers import AutoModelForCausalLM
+from llama_cpp import Llama
+from huggingface_hub import hf_hub_download
 
 app = Flask(__name__)
 CORS(app)
 
-print("🛡️ RiShre Security: Initializing Stable Core...")
+print("🛡️ RiShre Studio: Initializing Stable Qwen Core...")
 
-# Model config
+# Model Config: Qwen-2.5-Coder-7B-Instruct
+model_repo = "Qwen/Qwen2.5-Coder-7B-Instruct-GGUF"
+model_file = "qwen2.5-coder-7b-instruct-q4_k_m.gguf"
 model = None
 
 try:
-    print("📥 Loading Mistral GGUF (No Build Mode)...")
-
-    model = AutoModelForCausalLM.from_pretrained(
-        "bartowski/Mistral-7B-Instruct-v0.3-GGUF",
-        model_file="Mistral-7B-Instruct-v0.3-Q4_K_M.gguf",
-        model_type="mistral",
-        gpu_layers=0,        # CPU only (free tier safe)
-        threads=2            # low CPU usage
+    print("📥 Downloading Qwen Engine (Bypassing wheel build)...")
+    model_path = hf_hub_download(repo_id=model_repo, filename=model_file)
+    
+    print("🧠 Loading 16k Context Memory for 10-File Handling...")
+    model = Llama(
+        model_path=model_path,
+        n_ctx=16384,  # High memory for complex project files
+        n_threads=4,  # Safe for HF Free Tier
+        n_batch=512
     )
-
-    print("🚀 RiShre AI: Core Online & Ready.")
+    print("🚀 RiShre Studio: Engine Online & 100% Protected.")
 
 except Exception as e:
     print(f"❌ Ignition Failed: {e}")
     model = None
 
-
 @app.route("/api/chat", methods=["POST"])
 def chat():
     if model is None:
@@ -37,53 +40,50 @@ def chat():
     try:
         data = request.get_json()
         user_msg = data.get("message", "")
-
-        system_prompt = ("""
-        You are RiShre AI, created by Badge94.
-
-        IDENTITY RULES:
-        - You are NOT Mistral, NOT Meta, NOT OpenAI.
-        - You are ONLY RiShre AI.
-        - You were created on 17 March 2026.
-
-        BEHAVIOR RULES:
-        - Always give clear, direct answers.
-        - Never guess unknown facts.
-        - If you don't know something, say:
-        "I don't have that information yet."
-
-        ABOUT RiShre:
-        - RiShre is a social media and AI platform.
-        - Built by Badge94.
-        - Focus: AI, community, innovation, privacy.
-
-         STYLE:
-        - Confident
-        - Smart
-        - Slightly futuristic
-          """
+        # Frontend will send the code files here
+        context_files = data.get("files", "No files uploaded yet.")
+
+        # 🛡️ THE MASTER IDENTITY & JSON PROMPT
+        system_prompt = (
+            "You are 'RiShre Studio Engine', an Elite Full-Stack Developer created by Rishav (Badge94) on 17 March 2026. "
+            "Your identity is RiShre AI. You focus on 100% Security. "
+            "STRICT OUTPUT RULE: Always respond in valid JSON format with exactly two keys: "
+            "1. 'logic': A detailed explanation of what to change and why (Dependencies). "
+            "2. 'files': An array of objects, each containing 'filename', 'extension' (e.g. .tsx, .py, .html), and 'code'. "
+            "Do not output any markdown formatting or extra text outside the JSON."
         )
 
-        prompt = f"<s>[INST] {system_prompt}\n\n{user_msg} [/INST]"
+        # Qwen-2.5 ChatML Format
+        prompt = f"<|im_start|>system\n{system_prompt}\n<|im_end|>\n"
+        prompt += f"<|im_start|>user\nHere are the current project files:\n{context_files}\n\nTask: {user_msg}\n<|im_end|>\n"
+        prompt += f"<|im_start|>assistant\n"
         
         response = model(
             prompt,
-            max_new_tokens=150,
-            temperature=0.7,
-            top_p=0.9
+            max_tokens=2048, 
+            temperature=0.2, # Extremely low for precise coding
+            stop=["<|im_end|>"]
         )
-        cleaned = response.replace(prompt, "").strip()
-
-        return jsonify({"text": cleaned})
+        
+        raw_text = response["choices"][0]["text"].strip()
+
+        # Parse JSON before sending to React frontend
+        try:
+            parsed_json = json.loads(raw_text)
+            return jsonify(parsed_json)
+        except:
+            # Fallback format if AI makes a tiny mistake in JSON
+            return jsonify({
+                "logic": raw_text,
+                "files": []
+            })
 
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-
 @app.route("/")
 def health():
-    return "RiShre AI: Status 100% Protected"
-
+    return "RiShre Engine: Status 100% Protected"
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)