Spaces:

Shirpi
/

Student-s_AI

Sleeping

App Files Files Community

Shirpi commited on Dec 20, 2025

Commit

c5d0bfa

verified ·

1 Parent(s): db04039

Update app.py

Browse files

Files changed (1) hide show

app.py +165 -57

app.py CHANGED Viewed

@@ -35,8 +35,8 @@ user_db = load_db()
 current_key_index = 0
 app = Flask(__name__)
-# --- 🧠 SYSTEM INSTRUCTION ---
-SYSTEM_INSTRUCTION = """
 ROLE: You are "Student's AI", a professional academic tutor.
 RULES:
 1. **MATH:** Use LaTeX for formulas ($$ ... $$).
@@ -68,7 +68,7 @@ def process_image(image_data):
         return Image.open(io.BytesIO(image_bytes))
     except: return None
-def generate_with_retry(prompt, image_data=None, file_text=None, history_messages=[]):
     global current_key_index
     if not API_KEYS: return "🚨 API Keys Missing."
@@ -78,8 +78,15 @@ def generate_with_retry(prompt, image_data=None, file_text=None, history_message
         formatted_history.append({"role": role, "parts": [m["content"]]})
     current_parts = []
     if file_text: current_parts.append(f"analyzing file:\n{file_text}\n\n")
-    current_parts.append(prompt)
     if image_data:
         img = process_image(image_data)
         if img: current_parts.append(img)
@@ -94,13 +101,13 @@ def generate_with_retry(prompt, image_data=None, file_text=None, history_message
         try:
             genai.configure(api_key=key)
-            model = genai.GenerativeModel(model_name=model_name, system_instruction=SYSTEM_INSTRUCTION)
             if image_data or file_text:
                 response = model.generate_content(current_parts)
             else:
                 chat = model.start_chat(history=formatted_history)
-                response = chat.send_message(prompt)
             return response.text
         except Exception as e:
             current_key_index = (current_key_index + 1) % len(API_KEYS)
@@ -108,7 +115,7 @@ def generate_with_retry(prompt, image_data=None, file_text=None, history_message
     return "⚠️ System Busy. Please try again."
-# --- UI TEMPLATE (Fixed Quotes & Logic) ---
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en">
@@ -149,10 +156,10 @@ HTML_TEMPLATE = """
             -webkit-user-select: none; user-select: none;
         }
-        textarea, input { -webkit-user-select: text !important; user-select: text !important; }
         .user-content, .ai-content, code, pre { -webkit-user-select: none !important; user-select: none !important; }
-        /* --- APP CONTAINER (Pushed down for Header) --- */
         #app-container {
             display: flex; flex-direction: column;
             height: 100dvh; width: 100%;
@@ -176,6 +183,7 @@ HTML_TEMPLATE = """
         }
         .menu-btn:active { transform: scale(0.95); background: #222; }
         .app-title { font-size: 24px; font-weight: 800; letter-spacing: -0.5px; color: #fff; }
         /* --- SIDEBAR ANIMATION --- */
         #sidebar {
@@ -234,7 +242,7 @@ HTML_TEMPLATE = """
         /* --- LOCKED INTRO TEXT --- */
         #intro-container {
             position: absolute;
-            top: 140px; /* Fixed from top so it won't move up with keyboard */
             left: 50%;
             transform: translateX(-50%);
             width: 90%; max-width: 600px;
@@ -278,23 +286,25 @@ HTML_TEMPLATE = """
         .preview-img { width: 100%; height: 100%; object-fit: cover; }
         .remove-preview { position: absolute; top: -8px; right: -8px; background: red; color: white; border-radius: 50%; width: 20px; height: 20px; font-size: 12px; cursor: pointer; border: none; display: flex; align-items: center; justify-content: center; }
-        /* --- LOGIN PAGE LOCKED (FIXED) --- */
-        #login-overlay {
             position: fixed; inset: 0; background: #000; z-index: 2000;
-            display: flex;
-            /* Align to START (Top) to prevent centering jump */
-            align-items: flex-start; justify-content: center;
-            /* Force padding from top so it never moves */
-            padding-top: 180px;
-            transition: opacity 0.8s ease; opacity: 1; pointer-events: auto;
-        }
-        #login-overlay.hidden { opacity: 0; pointer-events: none; }
         .login-box {
             width: 90%; max-width: 350px; text-align: center;
-            padding: 40px; border: 1px solid var(--border); border-radius: 20px; background: #0a0a0a;
-            /* No auto margins */
         }
         #image-modal { position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.9); z-index: 3000; display: none; align-items: center; justify-content: center; opacity: 0; transition: opacity 0.3s; }
         #image-modal img { max-width: 95%; max-height: 90%; border-radius: 8px; box-shadow: 0 0 20px rgba(0,0,0,0.8); }
@@ -303,11 +313,32 @@ HTML_TEMPLATE = """
 </head>
 <body>
-    <div id="login-overlay">
         <div class="login-box">
             <h1 class="app-title" style="margin-bottom:10px;">Student's AI</h1>
-            <input type="text" id="username-input" placeholder="Your Name" style="width:100%; padding:15px; border-radius:12px; border:1px solid #333; background:#111; color:#fff; text-align:center; outline:none; margin-bottom:20px; font-size: 16px;" onkeydown="if(event.key==='Enter') handleLogin()">
-            <button onclick="handleLogin()" style="width:100%; padding:15px; border-radius:12px; border:none; background:#fff; font-weight:800; cursor:pointer; font-size: 16px;">Start Learning</button>
         </div>
     </div>
@@ -333,7 +364,7 @@ HTML_TEMPLATE = """
         <header>
             <div class="menu-btn" onclick="toggleSidebar()"><i class="fas fa-bars"></i></div>
             <span class="app-title">Student's AI</span>
-            <div style="width:40px;"></div>
         </header>
         <div id="chat-box"></div>
@@ -361,22 +392,34 @@ HTML_TEMPLATE = """
     <script>
         let currentUser = null;
         let currentChatId = null;
         let currentAttachment = { type: null, data: null, name: null };
         let longPressTimer;
         // --- FIXED INTRO: LOCKED CONTAINER ---
         function getIntroHtml(name) {
-            return `<div id="intro-container"><div class="msg ai-msg"><div class="ai-content"><h1>Hi ${name},</h1><p>Ready to master your studies today?</p></div></div></div>`;
         }
-        // --- AUTH LOGIC (SMOOTH TRANSITION) ---
         function checkLogin() {
             try {
-                const stored = localStorage.getItem("student_ai_user");
-                if (stored) {
-                    currentUser = stored;
                     document.getElementById("login-overlay").classList.add('hidden');
-                    showApp();
                 }
             } catch(e) { console.log("Storage access denied"); }
         }
@@ -387,27 +430,58 @@ HTML_TEMPLATE = """
             if(name) {
                 try { localStorage.setItem("student_ai_user", name); } catch(e){}
                 currentUser = name;
-                // FORCE HIDE DIRECTLY
-                const overlay = document.getElementById("login-overlay");
-                overlay.classList.add('hidden');
-                setTimeout(() => overlay.style.display = 'none', 500);
-                showApp();
             } else {
                 input.style.border = "1px solid red";
                 setTimeout(() => input.style.border = "1px solid #333", 2000);
             }
         }
         function handleLogout() {
-            try { localStorage.removeItem("student_ai_user"); } catch(e){}
-            const overlay = document.getElementById("login-overlay");
-            overlay.style.display = 'flex';
-            setTimeout(() => overlay.classList.remove('hidden'), 10);
-            document.getElementById('sidebar').classList.remove('open');
-            setTimeout(() => {
-                document.getElementById('chat-box').innerHTML = "";
-                currentChatId = null;
-                document.getElementById("username-input").value = "";
-            }, 500);
         }
         function showApp() {
@@ -500,7 +574,14 @@ HTML_TEMPLATE = """
                 }
                 const res = await fetch('/chat', {
                     method: 'POST', headers: {'Content-Type': 'application/json'},
-                    body: JSON.stringify({ message: promptText, image: imgData, file_text: fileText, username: currentUser, chat_id: currentChatId })
                 });
                 const data = await res.json();
@@ -655,13 +736,36 @@ HTML_TEMPLATE = """
             box.scrollTop = box.scrollHeight;
         }
-        checkLogin();
-    </script>
-</body>
-</html>
 """
-# --- BACKEND ROUTES ---
 @app.route("/", methods=["GET"])
 def home(): return render_template_string(HTML_TEMPLATE)
@@ -708,12 +812,16 @@ def chat():
     u, cid, msg = d.get("username"), d.get("chat_id"), d.get("message")
     img_data = d.get("image")
     file_text = d.get("file_text")
     if u not in user_db: user_db[u] = {}
     if cid not in user_db[u]: user_db[u][cid] = {"messages": []}
     user_db[u][cid]["messages"].append({"role": "user", "content": msg})
-    reply = generate_with_retry(msg, img_data, file_text, user_db[u][cid]["messages"][:-1])
     user_db[u][cid]["messages"].append({"role": "model", "content": reply})
     new_title = False
@@ -728,7 +836,7 @@ def chat():
 def manifest():
     data = {
         "name": "Student's AI",
-        "short_name": "Student's AI",
         "start_url": "/",
         "display": "standalone",
         "orientation": "portrait",
@@ -736,12 +844,12 @@ def manifest():
         "theme_color": "#09090b",
         "icons": [
             {
-                "src": "https://huggingface.co/spaces/Shirpi/Student-s_AI/resolve/main/1000177401.png",
                 "sizes": "192x192",
                 "type": "image/png"
             },
             {
-                "src": "https://huggingface.co/spaces/Shirpi/Student-s_AI/resolve/main/1000177401.png",
                 "sizes": "512x512",
                 "type": "image/png"
             }

 current_key_index = 0
 app = Flask(__name__)
+# --- 🧠 SYSTEM INSTRUCTION (Dynamic Context) ---
+BASE_INSTRUCTION = """
 ROLE: You are "Student's AI", a professional academic tutor.
 RULES:
 1. **MATH:** Use LaTeX for formulas ($$ ... $$).
         return Image.open(io.BytesIO(image_bytes))
     except: return None
+def generate_with_retry(prompt, image_data=None, file_text=None, history_messages=[], user_context=""):
     global current_key_index
     if not API_KEYS: return "🚨 API Keys Missing."
         formatted_history.append({"role": role, "parts": [m["content"]]})
     current_parts = []
+    # Add User Context to the prompt (Hidden from user)
+    full_prompt = prompt
+    if user_context:
+        full_prompt = f"[Context: Student is studying {user_context}]\nQuestion: {prompt}"
     if file_text: current_parts.append(f"analyzing file:\n{file_text}\n\n")
+    current_parts.append(full_prompt)
     if image_data:
         img = process_image(image_data)
         if img: current_parts.append(img)
         try:
             genai.configure(api_key=key)
+            model = genai.GenerativeModel(model_name=model_name, system_instruction=BASE_INSTRUCTION)
             if image_data or file_text:
                 response = model.generate_content(current_parts)
             else:
                 chat = model.start_chat(history=formatted_history)
+                response = chat.send_message(full_prompt)
             return response.text
         except Exception as e:
             current_key_index = (current_key_index + 1) % len(API_KEYS)
     return "⚠️ System Busy. Please try again."
+# --- UI TEMPLATE ---
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en">
             -webkit-user-select: none; user-select: none;
         }
+        textarea, input, select { -webkit-user-select: text !important; user-select: text !important; }
         .user-content, .ai-content, code, pre { -webkit-user-select: none !important; user-select: none !important; }
+        /* --- APP CONTAINER --- */
         #app-container {
             display: flex; flex-direction: column;
             height: 100dvh; width: 100%;
         }
         .menu-btn:active { transform: scale(0.95); background: #222; }
         .app-title { font-size: 24px; font-weight: 800; letter-spacing: -0.5px; color: #fff; }
+        .settings-icon { font-size: 18px; color: #aaa; cursor: pointer; }
         /* --- SIDEBAR ANIMATION --- */
         #sidebar {
         /* --- LOCKED INTRO TEXT --- */
         #intro-container {
             position: absolute;
+            top: 140px;
             left: 50%;
             transform: translateX(-50%);
             width: 90%; max-width: 600px;
         .preview-img { width: 100%; height: 100%; object-fit: cover; }
         .remove-preview { position: absolute; top: -8px; right: -8px; background: red; color: white; border-radius: 50%; width: 20px; height: 20px; font-size: 12px; cursor: pointer; border: none; display: flex; align-items: center; justify-content: center; }
+        /* --- LOGIN & SELECTION OVERLAYS (LOCKED & FIXED) --- */
+        .overlay {
             position: fixed; inset: 0; background: #000; z-index: 2000;
+            display: flex; align-items: flex-start; justify-content: center;
+            padding-top: 150px;
+            transition: opacity 0.5s ease; opacity: 1; pointer-events: auto;
+        }
+        .overlay.hidden { opacity: 0; pointer-events: none; }
         .login-box {
             width: 90%; max-width: 350px; text-align: center;
+            padding: 30px; border: 1px solid var(--border); border-radius: 20px; background: #0a0a0a;
         }
+        /* FORM ELEMENTS */
+        .form-label { display: block; text-align: left; font-size: 12px; color: #aaa; margin-bottom: 5px; margin-top: 10px; }
+        input, select { width: 100%; padding: 12px; margin-bottom: 5px; background: #18181b; border: 1px solid #333; color: #fff; border-radius: 8px; outline: none; font-family: 'Outfit', sans-serif; font-size: 16px; }
+        .start-btn { width: 100%; padding: 15px; border-radius: 12px; border: none; background: #fff; font-weight: 800; cursor: pointer; font-size: 16px; margin-top: 20px; }
         #image-modal { position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.9); z-index: 3000; display: none; align-items: center; justify-content: center; opacity: 0; transition: opacity 0.3s; }
         #image-modal img { max-width: 95%; max-height: 90%; border-radius: 8px; box-shadow: 0 0 20px rgba(0,0,0,0.8); }
 </head>
 <body>
+    <div id="login-overlay" class="overlay">
         <div class="login-box">
             <h1 class="app-title" style="margin-bottom:10px;">Student's AI</h1>
+            <input type="text" id="username-input" placeholder="Your Name" onkeydown="if(event.key==='Enter') handleLogin()">
+            <button class="start-btn" onclick="handleLogin()">Next</button>
+        </div>
+    </div>
+    <div id="selection-overlay" class="overlay hidden">
+        <div class="login-box">
+            <h2 style="color:#fff; margin-top:0;">Customize Profile</h2>
+            <span class="form-label">Education Level</span>
+            <select id="edu-level" onchange="updateEduOptions()">
+                <option value="school">School (6th - 12th)</option>
+                <option value="college">College (Arts/Engg)</option>
+            </select>
+            <span class="form-label">Class / Year</span>
+            <select id="edu-year">
+                </select>
+            <span class="form-label">Subject (Optional)</span>
+            <input type="text" id="edu-subject" placeholder="Ex: Maths, Physics...">
+            <button class="start-btn" onclick="handleSelection()">Start Learning</button>
         </div>
     </div>
         <header>
             <div class="menu-btn" onclick="toggleSidebar()"><i class="fas fa-bars"></i></div>
             <span class="app-title">Student's AI</span>
+            <div class="settings-icon" onclick="openSettings()"><i class="fas fa-cog"></i></div>
         </header>
         <div id="chat-box"></div>
     <script>
         let currentUser = null;
         let currentChatId = null;
+        let userContext = "";
         let currentAttachment = { type: null, data: null, name: null };
         let longPressTimer;
         // --- FIXED INTRO: LOCKED CONTAINER ---
         function getIntroHtml(name) {
+            return `<div id="intro-container"><div class="msg ai-msg"><div class="ai-content"><h1>Hi ${name},</h1><p>Ready to master ${userContext ? userContext.split(',')[1] : "studies"}?</p></div></div></div>`;
         }
+        // --- AUTH & ONBOARDING LOGIC ---
         function checkLogin() {
             try {
+                const storedUser = localStorage.getItem("student_ai_user");
+                const storedContext = localStorage.getItem("student_ai_context");
+                if (storedUser) {
+                    currentUser = storedUser;
                     document.getElementById("login-overlay").classList.add('hidden');
+                    if (storedContext) {
+                        userContext = storedContext;
+                        document.getElementById("selection-overlay").classList.add('hidden');
+                        showApp();
+                    } else {
+                        // User exists but context missing (New Feature)
+                        document.getElementById("selection-overlay").classList.remove('hidden');
+                        updateEduOptions();
+                    }
                 }
             } catch(e) { console.log("Storage access denied"); }
         }
             if(name) {
                 try { localStorage.setItem("student_ai_user", name); } catch(e){}
                 currentUser = name;
+                // Hide Login -> Show Selection
+                document.getElementById("login-overlay").classList.add('hidden');
+                document.getElementById("selection-overlay").classList.remove('hidden');
+                updateEduOptions();
             } else {
                 input.style.border = "1px solid red";
                 setTimeout(() => input.style.border = "1px solid #333", 2000);
             }
         }
+        function updateEduOptions() {
+            const level = document.getElementById('edu-level').value;
+            const yearSelect = document.getElementById('edu-year');
+            yearSelect.innerHTML = "";
+            let options = [];
+            if (level === 'school') {
+                options = ["6th Std", "7th Std", "8th Std", "9th Std", "10th Std", "11th Std", "12th Std"];
+            } else {
+                options = ["1st Year", "2nd Year", "3rd Year", "4th Year"];
+            }
+            options.forEach(opt => {
+                const el = document.createElement("option");
+                el.value = opt; el.innerText = opt;
+                yearSelect.appendChild(el);
+            });
+        }
+        function handleSelection() {
+            const level = document.getElementById('edu-level').value;
+            const year = document.getElementById('edu-year').value;
+            const subject = document.getElementById('edu-subject').value;
+            // Save Context
+            userContext = `${level}, ${year}, ${subject}`;
+            try { localStorage.setItem("student_ai_context", userContext); } catch(e){}
+            document.getElementById("selection-overlay").classList.add('hidden');
+            showApp();
+        }
+        function openSettings() {
+            document.getElementById("selection-overlay").classList.remove('hidden');
+        }
         function handleLogout() {
+            try {
+                localStorage.removeItem("student_ai_user");
+                localStorage.removeItem("student_ai_context");
+            } catch(e){}
+            location.reload(); // Simple reload to reset
         }
         function showApp() {
                 }
                 const res = await fetch('/chat', {
                     method: 'POST', headers: {'Content-Type': 'application/json'},
+                    body: JSON.stringify({
+                        message: promptText,
+                        image: imgData,
+                        file_text: fileText,
+                        username: currentUser,
+                        chat_id: currentChatId,
+                        user_context: userContext  // SENDING CONTEXT TO BACKEND
+                    })
                 });
                 const data = await res.json();
             box.scrollTop = box.scrollHeight;
         }
+        @app.route('/manifest.json')
+        def manifest():
+            data = {
+                "name": "Student's AI",
+                "short_name": "StudentAI",
+                "start_url": "/",
+                "display": "standalone",
+                "orientation": "portrait",
+                "background_color": "#09090b",
+                "theme_color": "#09090b",
+                "icons": [
+                    {
+                        "src": "https://cdn-icons-png.flaticon.com/512/4712/4712035.png",
+                        "sizes": "192x192",
+                        "type": "image/png"
+                    },
+                    {
+                        "src": "https://cdn-icons-png.flaticon.com/512/4712/4712035.png",
+                        "sizes": "512x512",
+                        "type": "image/png"
+                    }
+                ]
+            }
+            return Response(json.dumps(data), mimetype='application/json')
+        if __name__ == '__main__':
+            app.run(host='0.0.0.0', port=7860)
 """
+# --- BACKEND ROUTES START HERE ---
 @app.route("/", methods=["GET"])
 def home(): return render_template_string(HTML_TEMPLATE)
     u, cid, msg = d.get("username"), d.get("chat_id"), d.get("message")
     img_data = d.get("image")
     file_text = d.get("file_text")
+    user_context = d.get("user_context", "") # Capture User Context
     if u not in user_db: user_db[u] = {}
     if cid not in user_db[u]: user_db[u][cid] = {"messages": []}
     user_db[u][cid]["messages"].append({"role": "user", "content": msg})
+    # PASS USER CONTEXT TO GENERATION FUNCTION
+    reply = generate_with_retry(msg, img_data, file_text, user_db[u][cid]["messages"][:-1], user_context)
     user_db[u][cid]["messages"].append({"role": "model", "content": reply})
     new_title = False
 def manifest():
     data = {
         "name": "Student's AI",
+        "short_name": "StudentAI",
         "start_url": "/",
         "display": "standalone",
         "orientation": "portrait",
         "theme_color": "#09090b",
         "icons": [
             {
+                "src": "https://cdn-icons-png.flaticon.com/512/4712/4712035.png",
                 "sizes": "192x192",
                 "type": "image/png"
             },
             {
+                "src": "https://cdn-icons-png.flaticon.com/512/4712/4712035.png",
                 "sizes": "512x512",
                 "type": "image/png"
             }