Spaces:

Shirpi
/

Student-s_AI

Sleeping

App Files Files Community

Shirpi commited on Dec 19, 2025

Commit

1f29ed1

verified ·

1 Parent(s): 03c12d0

Create app.py

Browse files

Files changed (1) hide show

app.py +331 -0

app.py ADDED Viewed

	@@ -0,0 +1,331 @@

+import os
+import uuid
+import time
+import json
+import markdown
+from flask import Flask, request, jsonify, render_template_string
+import google.generativeai as genai # 👈 CLASSIC LIBRARY (STABLE)
+# ==========================================
+# 👇 API KEYS SETUP 👇
+# ==========================================
+keys_string = os.environ.get("API_KEYS", "")
+API_KEYS = [k.strip() for k in keys_string.replace(',', ' ').replace('\n', ' ').split() if k.strip()]
+print(f"✅ Loaded {len(API_KEYS)} API Keys.") # Debug Log
+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}")
+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.
+"""
+def generate_with_retry(prompt, history_messages=[]):
+    """
+    Uses Standard Library (google-generativeai)
+    Model: gemini-1.5-flash (Proven & Stable)
+    """
+    global current_key_index
+    if not API_KEYS:
+        return "❌ Error: API Keys Missing in Secrets."
+    # CONVERT HISTORY TO OLD SDK FORMAT
+    # Old SDK expects: [{'role': 'user', 'parts': ['text']}, {'role': 'model', 'parts': ['text']}]
+    recent_history = history_messages[-10:]
+    formatted_history = []
+    for msg in recent_history:
+        role = "user" if msg["role"] == "user" else "model"
+        formatted_history.append({
+            "role": role,
+            "parts": [msg["content"]]
+        })
+    last_error = ""
+    # RETRY LOOP
+    for attempt in range(len(API_KEYS) * 2):
+        key = API_KEYS[current_key_index]
+        try:
+            # 1. Configure Key
+            genai.configure(api_key=key)
+            # 2. Initialize Model (Classic Way)
+            model = genai.GenerativeModel(
+                model_name="gemini-1.5-flash",
+                system_instruction=SYSTEM_INSTRUCTION
+            )
+            # 3. Start Chat & Send Message
+            chat = model.start_chat(history=formatted_history)
+            response = chat.send_message(prompt)
+            return response.text
+        except Exception as e:
+            error_msg = str(e)
+            print(f"⚠️ Key #{current_key_index+1} Failed: {error_msg}")
+            if "404" in error_msg:
+                # If 404 happens here, it's purely a Model Name issue, but gemini-1.5-flash IS correct.
+                # So it usually means key access issue.
+                pass
+            last_error = error_msg
+            current_key_index = (current_key_index + 1) % len(API_KEYS)
+            time.sleep(1)
+            continue
+    return f"⚠️ **Connection Error:** {last_error}<br>Try restarting the Space."
+# --- UI TEMPLATE (KEEPING THE WORKING UI) ---
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <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">
+    <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; }
+    </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>
+        </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();
+            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;
+            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 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 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>`; }
+            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 ---
+@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": []}
+    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", [])})
+@app.route("/chat", methods=["POST"])
+def chat():
+    data = request.json
+    msg = data.get("message")
+    user = data.get("username")
+    cid = data.get("chat_id")
+    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)
+    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
+    save_db(user_db)
+    try:
+        previous_messages = user_db[user][cid]['messages'][:-1]
+        # Call the classic 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)})
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)