Update app.py

app.py

@@ -1,4 +1,5 @@
-from flask import Flask, request, jsonify
+import os
+from flask import Flask, request, jsonify, render_template
 from flask_cors import CORS
 from brain import MairaBrain
 
@@ -6,24 +7,18 @@ app = Flask(__name__)
 CORS(app)
 
 # --- Configuration for the 5 Neural Cores ---
-
-# 1. Maira Lite (Original Small)
 REPO_SMALL = "CyberCoder225/maira-model"
 FILE_SMALL = "SmolLM2-360M-Instruct.Q4_K_M.gguf"
 
-# 2. Maira Prime (Llama 3.2 1B)
 REPO_PRIME = "bartowski/Llama-3.2-1B-Instruct-GGUF"
 FILE_PRIME = "Llama-3.2-1B-Instruct-Q4_K_M.gguf"
 
-# 3. Maira Logic (Qwen 2.5 1.5B)
 REPO_LOGIC = "Qwen/Qwen2.5-1.5B-Instruct-GGUF"
 FILE_LOGIC = "qwen2.5-1.5b-instruct-q4_k_m.gguf"
 
-# 4. Maira Chat (Danube 3 500M) - FIXED FILENAME
 REPO_CHAT = "h2oai/h2o-danube3-500m-chat-GGUF"
 FILE_CHAT = "h2o-danube3-500m-chat-Q4_K_M.gguf"
 
-# 5. Maira Art (Granite 3.0 2B) - SWAPPED TO OPEN MODEL
 REPO_ART = "bartowski/granite-3.0-2b-instruct-GGUF"
 FILE_ART = "granite-3.0-2b-instruct-Q4_K_M.gguf"
 
@@ -38,22 +33,26 @@ print("✅ All Cores Online. Creator: CyberCoder225")
 
 @app.route('/', methods=['GET'])
 def home():
-    return jsonify({
-        "status": "online", 
-        "version": "6.0 (Quintessence)",
-        "owner": "CyberCoder225",
-        "active_models": ["small", "medium", "qwen", "danube", "granite"]
-    })
+    # 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('/chat', methods=['POST'])
 def chat():
     try:
         data = request.json
-        user_id = data.get("user_id", "guest")
+        user_id = data.get("user_id", "CyberCoder225")
         user_input = data.get("message", "")
         model_type = data.get("model_type", "small")
 
-        # Routing Logic for 5 Models
+        # Routing Logic
         if model_type == "medium":
             answer = maira_prime.get_response(user_id, user_input)
             model_used = "maira-prime"
@@ -63,19 +62,20 @@ def chat():
         elif model_type == "danube":
             answer = maira_chat.get_response(user_id, user_input)
             model_used = "maira-chat"
-        elif model_type == "granite": # Updated tag name
+        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"}
         })
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        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)