Update app.py

app.py

@@ -7,119 +7,96 @@ from flask import Flask, request, jsonify, render_template_string
 import google.generativeai as genai
 
 # ==========================================
-# 👇 API KEYS SETUP 👇
+# 👇 DIAGNOSTIC KEY LOADING 👇
 # ==========================================
-keys_string = os.environ.get("API_KEYS", "")
-# Clean up keys (remove spaces/newlines)
-API_KEYS = [k.strip() for k in keys_string.replace(',', ' ').replace('\n', ' ').split() if k.strip()]
+keys_raw = os.environ.get("API_KEYS", "")
+# Split and clean
+API_KEYS = [k.strip() for k in keys_raw.replace(',', ' ').replace('\n', ' ').split() if k.strip()]
 
-# 👇 MODELS TO TRY (Priority Order) 👇
-MODELS = ["gemini-1.5-flash", "gemini-1.5-pro", "gemini-pro"]
-
-print(f"✅ Loaded {len(API_KEYS)} API Keys.")
+# Debug log to console
+print(f"DEBUG: Found {len(API_KEYS)} keys.")
 
 current_key_index = 0
 app = Flask(__name__)
 
 # --- 💾 DATABASE ---
 DB_FILE = "chat_db.json"
-
 def load_db():
     try:
         if os.path.exists(DB_FILE):
             with open(DB_FILE, 'r') as f: return json.load(f)
     except: pass
     return {}
-
 def save_db(db):
     try:
         with open(DB_FILE, 'w') as f: json.dump(db, f, indent=2)
-    except Exception as e:
-        print(f"Save Error: {e}")
-
+    except: pass
 user_db = load_db()
 
 # --- 🧠 SYSTEM INSTRUCTION ---
 SYSTEM_INSTRUCTION = """
 ROLE: You are "Student's AI", a coding tutor.
-PERMANENT RULES:
-1. **BLACK BOX:** Always format code inside Markdown blocks (```python ... ```).
-2. **INDENTATION:** Code must have perfect spacing.
-3. **LANGUAGE:** Tamil & English mix (Tanglish).
-4. **BRIEF:** Keep explanations short and crisp.
+RULES: Format code in Markdown (```python). Explain in Tanglish.
 """
 
-def get_next_key():
-    """Rotates to the next available API Key"""
-    global current_key_index
-    current_key_index = (current_key_index + 1) % len(API_KEYS)
-    return API_KEYS[current_key_index]
-
 def generate_with_retry(prompt, history_messages=[]):
-    """
-    SMART LOGIC:
-    - If 429 (Quota): Switch KEY immediately.
-    - If 404 (Not Found): Switch MODEL immediately.
-    """
     global current_key_index
     
-    if not API_KEYS: return "❌ Error: API Keys Missing in Secrets."
+    # 1. CHECK KEYS FIRST
+    if not API_KEYS:
+        return "🚨 **CRITICAL ERROR:** API Keys List is EMPTY. Please check Hugging Face Secrets. Ensure name is `API_KEYS`."
 
-    # History Formatting
-    recent_history = history_messages[-10:] 
+    # History Format
     formatted_history = []
-    for msg in recent_history:
+    for msg in history_messages[-6:]: # Keep context small to avoid overload
         role = "user" if msg["role"] == "user" else "model"
         formatted_history.append({"role": role, "parts": [msg["content"]]})
 
-    last_error = ""
+    # MODELS TO TRY (Standard & Legacy)
+    # gemini-1.5-flash: New & Fast
+    # gemini-pro: Old & Reliable (Best for free tier)
+    models_to_test = ["gemini-1.5-flash", "gemini-pro"]
     
-    # We allow (Number of Keys * Number of Models) attempts total
-    max_attempts = len(API_KEYS) * len(MODELS)
+    log_messages = [] # To show user if all fails
 
-    for attempt in range(max_attempts):
+    # RETRY LOGIC
+    for i in range(len(API_KEYS)):
         key = API_KEYS[current_key_index]
         
-        # Try current key with all models priority
-        for model_name in MODELS:
+        # Try both models for this key
+        for model_name in models_to_test:
             try:
                 genai.configure(api_key=key)
-                model = genai.GenerativeModel(
-                    model_name=model_name,
-                    system_instruction=SYSTEM_INSTRUCTION
-                )
+                model = genai.GenerativeModel(model_name=model_name, system_instruction=SYSTEM_INSTRUCTION)
                 
+                # Test call
                 chat = model.start_chat(history=formatted_history)
                 response = chat.send_message(prompt)
+                return response.text # SUCCESS!
                 
-                # If success, return text
-                return response.text
-
             except Exception as e:
-                error_str = str(e).lower()
+                err = str(e)
+                log_messages.append(f"Key #{current_key_index+1} | {model_name} -> {err}")
                 
-                # CASE 1: QUOTA EXCEEDED (429) -> SWITCH KEY
-                if "429" in error_str or "exhausted" in error_str:
-                    print(f"⚠️ Key #{current_key_index+1} Quota Full ({model_name}). Switching Key...")
-                    get_next_key() # Rotate Key
-                    time.sleep(1)  # Cool down
-                    break # Break inner model loop, try new key
+                # If Quota error, switch key immediately
+                if "429" in err or "quota" in err.lower():
+                    current_key_index = (current_key_index + 1) % len(API_KEYS)
+                    break # Break model loop, go to next key
                 
-                # CASE 2: MODEL NOT FOUND (404) -> TRY NEXT MODEL (SAME KEY)
-                elif "404" in error_str or "not found" in error_str:
-                    print(f"⚠️ {model_name} not found. Trying next model...")
-                    continue # Try next model in the list
-                
-                # CASE 3: OTHER ERRORS -> SWITCH KEY
-                else:
-                    print(f"⚠️ Error on {model_name}: {error_str}")
-                    last_error = error_str
-                    get_next_key()
-                    break 
+                # If 404, just try next model
+                if "404" in err:
+                    continue
+        
+        # Move to next key if models failed
+        current_key_index = (current_key_index + 1) % len(API_KEYS)
+        time.sleep(1)
 
-    return f"⚠️ **System Busy:** All keys/models failed. (Last Error: {last_error})"
+    # IF WE REACH HERE, EVERYTHING FAILED.
+    # PRINT THE LOGS TO THE USER SO WE KNOW WHY.
+    error_report = "<br>".join(log_messages)
+    return f"⚠️ **ALL FAILED (Debug Report):**<br><small>{error_report}</small>"
 
-# --- UI TEMPLATE (STABLE - NO CHANGES) ---
+# --- UI TEMPLATE (Unchanged) ---
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en">
@@ -127,219 +104,103 @@ HTML_TEMPLATE = """
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no">
     <title>Student's AI</title>
-    <link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css" rel="stylesheet">
-    <link href="https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;500&family=Inter:wght@300;400;600;800&display=swap" rel="stylesheet">
+    <link href="[https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css](https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css)" rel="stylesheet">
     <style>
         :root { --bg: #000; --sidebar: #0a0a0a; --header: rgba(0,0,0,0.9); --user-bg: #222; --text: #ececec; }
-        * { box-sizing: border-box; -webkit-tap-highlight-color: transparent; }
-        body, html { margin: 0; padding: 0; height: 100%; width: 100%; background: var(--bg); color: var(--text); font-family: 'Inter', sans-serif; overflow: hidden; }
-        pre { background: #161616; border: 1px solid #333; border-radius: 8px; padding: 15px; overflow-x: auto; font-family: 'JetBrains Mono', monospace; }
-        code { font-family: 'JetBrains Mono', monospace; font-size: 13px; color: #e6edf3; }
-        .k { color: #ff7b72; } .s { color: #a5d6ff; } .c { color: #8b949e; } .nf { color: #d2a8ff; } .m { color: #79c0ff; }
-        #app-container { display: none; flex-direction: column; height: 100%; }
-        header { height: 60px; padding: 0 20px; background: var(--header); border-bottom: 1px solid #222; display: flex; align-items: center; justify-content: space-between; flex-shrink: 0; z-index: 50; }
-        #sidebar { position: fixed; inset: 0; width: 280px; background: var(--sidebar); border-right: 1px solid #222; transform: translateX(-100%); transition: 0.3s; z-index: 100; padding: 20px; display: flex; flex-direction: column; }
-        #sidebar.open { transform: translateX(0); }
-        .overlay { position: fixed; inset: 0; background: rgba(0,0,0,0.7); z-index: 90; display: none; }
-        .overlay.active { display: block; }
-        #chat-box { flex: 1; overflow-y: auto; padding: 20px 20px 40px 20px; display: flex; flex-direction: column; gap: 20px; }
-        .msg { display: flex; flex-direction: column; width: 100%; animation: fade 0.3s forwards; }
-        @keyframes fade { from{opacity:0; transform:translateY(10px);} to{opacity:1; transform:translateY(0);} }
-        .user-msg { align-items: flex-end; }
-        .user-msg .bubble { background: var(--user-bg); padding: 12px 18px; border-radius: 18px 18px 4px 18px; max-width: 85%; }
-        .ai-msg { align-items: flex-start; }
-        .ai-text { width: 100%; line-height: 1.6; font-size: 16px; color: #eee; }
-        .ai-text p:first-child { margin-top: 0; }
-        .input-wrap { padding: 15px 20px; background: #000; border-top: 1px solid #222; flex-shrink: 0; z-index: 60; }
-        .input-bar { width: 100%; max-width: 800px; margin: 0 auto; background: #161616; border: 1px solid #333; border-radius: 30px; padding: 8px 12px 8px 20px; display: flex; align-items: center; gap: 10px; }
-        textarea { flex: 1; background: transparent; border: none; color: #fff; font-size: 16px; max-height: 100px; padding: 5px 0; resize: none; outline: none; font-family: 'Inter', sans-serif; }
-        .send-btn { width: 40px; height: 40px; min-width: 40px; background: #fff; color: #000; border-radius: 50%; border: none; cursor: pointer; display: flex; justify-content: center; align-items: center; transition: 0.2s; z-index: 10; }
-        .send-btn:active { transform: scale(0.9); background: #ddd; }
-        #login-overlay { position: fixed; inset: 0; z-index: 2000; background: #000; display: flex; flex-direction: column; align-items: center; justify-content: center; }
-        .login-box { width: 90%; max-width: 400px; padding: 40px; background: #111; border: 1px solid #333; border-radius: 20px; text-align: center; }
-        .login-input { width: 100%; padding: 15px; background: #000; border: 1px solid #333; color: #fff; border-radius: 10px; margin-bottom: 20px; text-align: center; }
-        .login-btn { width: 100%; padding: 15px; background: #fff; color: #000; border: none; border-radius: 10px; font-weight: 700; cursor: pointer; }
+        body, html { margin: 0; padding: 0; height: 100%; width: 100%; background: var(--bg); color: var(--text); font-family: sans-serif; overflow: hidden; }
+        pre { background: #161616; border: 1px solid #333; padding: 15px; overflow-x: auto; color: #e6edf3; }
+        #app-container { display: flex; flex-direction: column; height: 100%; }
+        #chat-box { flex: 1; overflow-y: auto; padding: 20px; display: flex; flex-direction: column; gap: 20px; }
+        .msg { padding: 10px 15px; border-radius: 10px; max-width: 85%; line-height: 1.5; }
+        .user-msg { align-self: flex-end; background: #222; }
+        .ai-msg { align-self: flex-start; }
+        .input-wrap { padding: 15px; background: #000; border-top: 1px solid #333; display: flex; gap: 10px; }
+        input { flex: 1; padding: 10px; border-radius: 20px; border: 1px solid #333; background: #111; color: #fff; }
+        button { padding: 10px 20px; border-radius: 20px; border: none; background: #fff; cursor: pointer; }
     </style>
 </head>
 <body>
-    <div id="login-overlay">
-        <div class="login-box">
-            <h2 style="color:#fff;">Student's AI</h2>
-            <input type="text" id="username-input" class="login-input" placeholder="Enter Name" autocomplete="off">
-            <button onclick="handleLogin()" class="login-btn">Start</button>
-        </div>
-    </div>
     <div id="app-container">
-        <div class="overlay" id="overlay" onclick="toggleSidebar()"></div>
-        <div id="sidebar">
-            <div style="padding-bottom:20px; border-bottom:1px solid #222; margin-bottom:20px; font-weight:bold; display:flex; gap:10px; align-items:center;">
-                <div id="avatar" style="width:35px; height:35px; background:#333; border-radius:50%; display:flex; align-items:center; justify-content:center;">U</div>
-                <span id="display-name">User</span>
-            </div>
-            <div style="background:#fff; color:#000; padding:12px; border-radius:8px; text-align:center; font-weight:bold; cursor:pointer;" onclick="newChat()">+ New Chat</div>
-            <div style="flex:1; overflow-y:auto; margin-top:20px;" id="history-list"></div>
-            <div style="margin-top:20px; color:#ff5555; cursor:pointer; text-align:center;" onclick="handleLogout()">Log Out</div>
-        </div>
-        <header>
-            <i class="fas fa-bars" style="color:#fff; font-size:20px; cursor:pointer;" onclick="toggleSidebar()"></i>
-            <div style="font-weight:bold; font-size:18px; color:#fff;">Student's AI</div>
-            <div style="width:20px;"></div>
-        </header>
         <div id="chat-box"></div>
         <div class="input-wrap">
-            <div class="input-bar">
-                <textarea id="input" placeholder="Type here..." rows="1" oninput="resizeInput(this)"></textarea>
-                <button id="sendBtn" class="send-btn" onclick="send(event)"><i class="fas fa-arrow-up"></i></button>
-            </div>
+            <input id="input" placeholder="Type here..." onkeydown="if(event.key==='Enter') send()">
+            <button onclick="send()">Send</button>
         </div>
     </div>
     <script>
-        let currentUser = null; let currentChatId = null;
-        function resizeInput(el) { el.style.height = 'auto'; el.style.height = Math.min(el.scrollHeight, 100) + 'px'; }
-        document.getElementById("input").addEventListener("keydown", function(e) { if (e.key === "Enter" && !e.shiftKey) { e.preventDefault(); send(e); }});
-        function checkLogin() {
-            const stored = localStorage.getItem("student_ai_user");
-            if (stored) { currentUser = stored; showApp(); } else { document.getElementById("login-overlay").style.display = "flex"; }
-        }
-        function handleLogin() {
-            const name = document.getElementById("username-input").value.trim();
-            if(name) { localStorage.setItem("student_ai_user", name); currentUser = name; showApp(); }
-        }
-        function handleLogout() { localStorage.removeItem("student_ai_user"); location.reload(); }
-        function showApp() {
-            document.getElementById("login-overlay").style.display = "none";
-            document.getElementById("app-container").style.display = "flex";
-            document.getElementById("display-name").innerText = currentUser;
-            document.getElementById("avatar").innerText = currentUser[0].toUpperCase();
-            loadHistory();
-            if(!currentChatId) {
-                 const box = document.getElementById("chat-box");
-                 if(box.innerHTML.trim() === "") { box.innerHTML = `<div class="msg ai-msg"><div class="ai-text"><strong>Welcome ${currentUser}!</strong><br>I am ready.</div></div>`; }
-            }
-        }
-        const chatBox = document.getElementById('chat-box');
-        const input = document.getElementById('input');
-        async function send(event) {
-            if(event) event.preventDefault();
+        let currentChatId = null;
+        let currentUser = localStorage.getItem("user") || "User";
+        
+        async function send() {
+            const input = document.getElementById('input');
             const text = input.value.trim();
             if(!text) return;
-            input.value = ''; input.style.height = 'auto'; input.focus();
-            const userDiv = document.createElement('div'); userDiv.className = 'msg user-msg';
-            userDiv.innerHTML = `<div class="bubble">${text}</div>`;
-            chatBox.appendChild(userDiv); chatBox.scrollTop = chatBox.scrollHeight;
-            const tempId = "ai-" + Date.now();
-            const aiDiv = document.createElement('div'); aiDiv.className = 'msg ai-msg'; aiDiv.id = tempId;
-            aiDiv.innerHTML = `<div class="ai-text"><i class="fas fa-circle-notch fa-spin"></i> Thinking...</div>`;
-            chatBox.appendChild(aiDiv); chatBox.scrollTop = chatBox.scrollHeight;
+            
+            const chatBox = document.getElementById('chat-box');
+            chatBox.innerHTML += `<div class="msg user-msg">${text}</div>`;
+            input.value = '';
+            
+            // Loading
+            const loadId = "load-"+Date.now();
+            chatBox.innerHTML += `<div id="${loadId}" class="msg ai-msg">Thinking...</div>`;
+            chatBox.scrollTop = chatBox.scrollHeight;
+
             try {
                 if(!currentChatId) {
-                    const res = await fetch('/new_chat', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ username: currentUser }) });
-                    if(!res.ok) throw new Error("Net Error");
-                    const data = await res.json();
-                    currentChatId = data.chat_id;
+                    const r = await fetch('/new_chat', {method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({username:currentUser})});
+                    const d = await r.json(); currentChatId = d.chat_id;
                 }
-                const res = await fetch('/chat', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ message: text, username: currentUser, chat_id: currentChatId }) });
-                if(!res.ok) throw new Error("API Error");
+                const res = await fetch('/chat', {
+                    method: 'POST', headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({message: text, username: currentUser, chat_id: currentChatId})
+                });
                 const data = await res.json();
-                document.getElementById(tempId).remove();
-                const finalAiDiv = document.createElement('div'); finalAiDiv.className = 'msg ai-msg';
-                finalAiDiv.innerHTML = `<div class="ai-text">${data.response || "Error"}</div>`;
-                chatBox.appendChild(finalAiDiv);
-                if(data.new_title) loadHistory();
-            } catch(e) { document.getElementById(tempId).innerHTML = `<div class="ai-text" style="color:#ff5555">⚠️ Failed: ${e.message}</div>`; }
+                document.getElementById(loadId).innerHTML = data.response;
+            } catch(e) {
+                document.getElementById(loadId).innerHTML = "Error: " + e.message;
+            }
             chatBox.scrollTop = chatBox.scrollHeight;
         }
-        async function loadHistory() {
-            try {
-                const res = await fetch('/get_history', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ username: currentUser }) });
-                const data = await res.json();
-                const list = document.getElementById("history-list"); list.innerHTML = "";
-                if(data.chats) Object.keys(data.chats).reverse().forEach(cid => { list.innerHTML += `<div style="padding:10px; color:#888; cursor:pointer;" onclick="loadChat('${cid}')"><i class="far fa-comment"></i> ${data.chats[cid].title}</div>`; });
-            } catch(e) {}
-        }
-        async function newChat(silent=false) {
-            try {
-                const res = await fetch('/new_chat', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ username: currentUser }) });
-                const data = await res.json(); currentChatId = data.chat_id;
-                if(!silent) { chatBox.innerHTML = ''; loadHistory(); toggleSidebar(); }
-            } catch(e) {}
-        }
-        async function loadChat(cid) {
-            currentChatId = cid;
-            const res = await fetch('/get_chat', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ username: currentUser, chat_id: cid }) });
-            const data = await res.json(); chatBox.innerHTML = '';
-            data.messages.forEach(msg => {
-                const div = document.createElement('div'); div.className = msg.role === 'user' ? 'msg user-msg' : 'msg ai-msg';
-                div.innerHTML = msg.role === 'user' ? `<div class="bubble">${msg.content}</div>` : `<div class="ai-text">${msg.content}</div>`;
-                chatBox.appendChild(div);
-            });
-            toggleSidebar(); chatBox.scrollTop = chatBox.scrollHeight;
-        }
-        function toggleSidebar() { document.getElementById('sidebar').classList.toggle('open'); document.getElementById('overlay').classList.toggle('active'); }
-        checkLogin();
     </script>
 </body>
 </html>
 """
 
-# --- BACKEND ROUTES ---
+# --- ROUTES ---
 @app.route("/", methods=["GET"])
 def home(): return render_template_string(HTML_TEMPLATE)
 
-@app.route("/get_history", methods=["POST"])
-def get_history():
-    user = request.json.get("username")
-    return jsonify({"chats": user_db.get(user, {})})
-
 @app.route("/new_chat", methods=["POST"])
 def new_chat():
     user = request.json.get("username")
     if user not in user_db: user_db[user] = {}
     nid = str(uuid.uuid4())
-    user_db[user][nid] = {"title": "New Chat", "messages": []}
+    user_db[user][nid] = {"title": "New", "messages": []}
     save_db(user_db)
-    return jsonify({"status": "success", "chat_id": nid})
-
-@app.route("/get_chat", methods=["POST"])
-def get_chat():
-    data = request.json
-    return jsonify({"messages": user_db.get(data.get("username"), {}).get(data.get("chat_id"), {}).get("messages", [])})
+    return jsonify({"chat_id": nid})
 
 @app.route("/chat", methods=["POST"])
 def chat():
-    data = request.json
-    msg = data.get("message")
-    user = data.get("username")
-    cid = data.get("chat_id")
+    d = request.json
+    u, cid, msg = d.get("username"), d.get("chat_id"), d.get("message")
     
-    if user not in user_db: user_db[user] = {}
-    if cid not in user_db[user]: 
-        user_db[user][cid] = {"title": "New Chat", "messages": []}
-        save_db(user_db)
+    if u not in user_db: user_db[u] = {}
+    if cid not in user_db[u]: user_db[u][cid] = {"messages": []}
 
-    user_db[user][cid]["messages"].append({"role": "user", "content": msg})
-    new_title = False
-    if len(user_db[user][cid]["messages"]) == 1:
-        user_db[user][cid]["title"] = " ".join(msg.split()[:4]) + "..."
-        new_title = True
+    user_db[u][cid]["messages"].append({"role": "user", "content": msg})
+    
+    # CALL DEBUG GENERATOR
+    reply = generate_with_retry(msg, user_db[u][cid]["messages"][:-1])
     
-    save_db(user_db) 
+    # Check if it's the specific error report, don't markdown it
+    if "DEBUG Report" in reply or "CRITICAL ERROR" in reply:
+        formatted_html = reply
+    else:
+        formatted_html = markdown.markdown(reply, extensions=['fenced_code'])
 
-    try:
-        previous_messages = user_db[user][cid]['messages'][:-1]
-        
-        # Call the SMART generation function
-        raw_response = generate_with_retry(msg, previous_messages)
-        
-        formatted_html = markdown.markdown(raw_response, extensions=['fenced_code', 'codehilite'])
-        
-        user_db[user][cid]["messages"].append({"role": "ai", "content": formatted_html})
-        save_db(user_db) 
-        
-        return jsonify({"response": formatted_html, "new_title": new_title})
-        
-    except Exception as e:
-        return jsonify({"error": str(e)})
+    user_db[u][cid]["messages"].append({"role": "model", "content": formatted_html})
+    save_db(user_db)
+    return jsonify({"response": formatted_html})
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)