Update app.py

app.py

@@ -1,19 +1,33 @@
 from flask import Flask, request, jsonify
-from flask_cors import CORS  # <--- Add this!
+from flask_cors import CORS
 from brain import MairaBrain
 
 app = Flask(__name__)
-CORS(app)  # <--- This allows your HTML to talk to the API!
+CORS(app)
 
-# Config
-REPO_ID = "CyberCoder225/maira-model"
-FILENAME = "SmolLM2-360M-Instruct.Q4_K_M.gguf"
+# --- Configuration for both models ---
+# Model 1: Small/Fast
+REPO_SMALL = "CyberCoder225/maira-model"
+FILE_SMALL = "SmolLM2-360M-Instruct.Q4_K_M.gguf"
 
-maira = MairaBrain(REPO_ID, FILENAME)
+# Model 2: Medium/Smart (Llama-3.2-1B)
+REPO_MEDIUM = "bartowski/Llama-3.2-1B-Instruct-GGUF"
+FILE_MEDIUM = "Llama-3.2-1B-Instruct-Q4_K_M.gguf"
+
+# Initialize both brains
+print("🌌 Initializing Maira Small...")
+maira_small = MairaBrain(REPO_SMALL, FILE_SMALL)
+
+print("🌟 Initializing Maira Medium...")
+maira_medium = MairaBrain(REPO_MEDIUM, FILE_MEDIUM)
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({"status": "online", "version": "2.1 (Modular)"})
+    return jsonify({
+        "status": "online", 
+        "version": "5.0 (Multi-Neural)",
+        "models": ["small", "medium"]
+    })
 
 @app.route('/chat', methods=['POST'])
 def chat():
@@ -21,11 +35,21 @@ def chat():
         data = request.json
         user_id = data.get("user_id", "guest")
         user_input = data.get("message", "")
+        # Get the model choice from the request, default to 'small'
+        model_type = data.get("model_type", "small")
 
-        answer = maira.get_response(user_id, user_input)
+        # Choose the brain based on user preference
+        if model_type == "medium":
+            answer = maira_medium.get_response(user_id, user_input)
+            model_used = "maira-medium"
+        else:
+            answer = maira_small.get_response(user_id, user_input)
+            model_used = "maira-small"
         
-        # We match the key 'maira' to your brain's output
-        return jsonify({"maira": answer})
+        return jsonify({
+            "maira": answer,
+            "metadata": {"model": model_used}
+        })
     except Exception as e:
         return jsonify({"error": str(e)}), 500