Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 from flask import Flask, request, jsonify, render_template
 from flask_cors import CORS
 from brain import MairaBrain
@@ -6,76 +5,27 @@ from brain import MairaBrain
 app = Flask(__name__)
 CORS(app)
 
-# --- Configuration for the 5 Neural Cores ---
-REPO_SMALL = "CyberCoder225/maira-model"
-FILE_SMALL = "SmolLM2-360M-Instruct.Q4_K_M.gguf"
+# Initialize models here... (keep your current initialization code)
 
-REPO_PRIME = "bartowski/Llama-3.2-1B-Instruct-GGUF"
-FILE_PRIME = "Llama-3.2-1B-Instruct-Q4_K_M.gguf"
-
-REPO_LOGIC = "Qwen/Qwen2.5-1.5B-Instruct-GGUF"
-FILE_LOGIC = "qwen2.5-1.5b-instruct-q4_k_m.gguf"
-
-REPO_CHAT = "h2oai/h2o-danube3-500m-chat-GGUF"
-FILE_CHAT = "h2o-danube3-500m-chat-Q4_K_M.gguf"
-
-REPO_ART = "bartowski/granite-3.0-2b-instruct-GGUF"
-FILE_ART = "granite-3.0-2b-instruct-Q4_K_M.gguf"
-
-# --- Initialize All Brains ---
-print("🌌 Initializing Neural Cores...")
-maira_lite = MairaBrain(REPO_SMALL, FILE_SMALL)
-maira_prime = MairaBrain(REPO_PRIME, FILE_PRIME)
-maira_logic = MairaBrain(REPO_LOGIC, FILE_LOGIC)
-maira_chat = MairaBrain(REPO_CHAT, FILE_CHAT)
-maira_art = MairaBrain(REPO_ART, FILE_ART)
-print("✅ All Cores Online. Creator: CyberCoder225")
-
-@app.route('/', methods=['GET'])
-def home():
-    # This serves your index.html from the /templates folder
-    try:
-        return render_template('index.html')
-    except:
-        # Fallback if you haven't created the /templates folder yet
-        return jsonify({
-            "status": "online", 
-            "message": "HTML not found. Use /chat API or add index.html to /templates folder.",
-            "owner": "CyberCoder225"
-        })
+@app.route('/')
+def index():
+    # Looks for /templates/index.html
+    return render_template('index.html')
 
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
         data = request.json
-        user_id = data.get("user_id", "CyberCoder225")
-        user_input = data.get("message", "")
-        model_type = data.get("model_type", "small")
-
-        # Routing Logic
-        if model_type == "medium":
-            answer = maira_prime.get_response(user_id, user_input)
-            model_used = "maira-prime"
-        elif model_type == "qwen":
-            answer = maira_logic.get_response(user_id, user_input)
-            model_used = "maira-logic"
-        elif model_type == "danube":
-            answer = maira_chat.get_response(user_id, user_input)
-            model_used = "maira-chat"
-        elif model_type == "granite":
-            answer = maira_art.get_response(user_id, user_input)
-            model_used = "maira-art"
-        else:
-            answer = maira_lite.get_response(user_id, user_input)
-            model_used = "maira-lite"
-                                
-        return jsonify({
-            "maira": answer,
-            "metadata": {"model": model_used, "creator": "CyberCoder225"}
-        })
+        msg = data.get("message")
+        m_type = data.get("model_type", "small")
+        
+        # Select brain based on user's manual choice
+        if m_type == "qwen": answer = maira_logic.get_response("boss", msg)
+        elif m_type == "granite": answer = maira_art.get_response("boss", msg)
+        elif m_type == "medium": answer = maira_prime.get_response("boss", msg)
+        elif m_type == "danube": answer = maira_chat.get_response("boss", msg)
+        else: answer = maira_lite.get_response("boss", msg)
+
+        return jsonify({"maira": answer})
     except Exception as e:
-        print(f"❌ Error: {str(e)}")
-        return jsonify({"maira": f"System error, boss: {str(e)}"}), 500
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+        return jsonify({"error": str(e)}), 500