Update app.py

app.py

@@ -3,14 +3,27 @@ import gradio as gr
 import requests
 import json
 
-# Test HF_TOKEN
+# Hugging Face Token
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
-    print("Error: HF_TOKEN not set. Please set your Hugging Face API token.")
+    print("Error: HF_TOKEN not set.")
 else:
     print("HF_TOKEN loaded successfully.")
 
-# 🌍 Web search function (unchanged)
+# Memory file
+MEMORY_FILE = "memory.json"
+
+def load_memory():
+    if os.path.exists(MEMORY_FILE):
+        with open(MEMORY_FILE, "r") as f:
+            return json.load(f)
+    return []
+
+def save_memory(memory):
+    with open(MEMORY_FILE, "w") as f:
+        json.dump(memory, f)
+
+# Web search
 def search_web(query):
     try:
         url = "https://api.duckduckgo.com/"
@@ -27,309 +40,119 @@ def search_web(query):
     except Exception as e:
         return f"Search error: {e}"
 
-# 🧠 Memory setup (unchanged)
-MEMORY_FILE = "memory.json"
-
-def load_memory():
-    if os.path.exists(MEMORY_FILE):
-        with open(MEMORY_FILE, "r") as f:
-            return json.load(f)
-    return []
-
-def save_memory(memory):
-    with open(MEMORY_FILE, "w") as f:
-        json.dump(memory, f)
-
-memory = load_memory()
-
-# 💬 Chat function with better error handling
+# Chat function with messages format
 def chat_with_model(message, history, context, file_input=None):
-    if not isinstance(history, list):
+    if history is None:
         history = []
 
     if message.lower().startswith("search "):
         query = message[7:]
         search_result = search_web(query)
-        history.append((message, f"🔎 Here's what I found online:\n\n{search_result}"))
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": f"🔎 Here's what I found online:\n\n{search_result}"})
         save_memory(history)
         return history, history
 
     if file_input:
-        file_name = file_input.name
-        message += f"\n\n📎 (User uploaded a file named '{file_name}')"
-
-    conversation = [
-        {"role": "system", "content": (
-            "You are EduAI, a multilingual educational AI assistant created by a Sri Lankan student named Wafa Fazly. "
-            "When solving math, explain step-by-step like a professional tutor. "
-            "Use Markdown and LaTeX formatting for equations (use \\[ and \\]). "
-            "Keep answers neat, structured, and student-friendly."
-        )}
-    ]
+        message += f"\n\n📎 (User uploaded a file named '{file_input.name}')"
 
-    for past_user, past_bot in history[-5:]:
-        conversation.append({"role": "user", "content": past_user})
-        conversation.append({"role": "assistant", "content": past_bot})
+    # Add user message
+    history.append({"role": "user", "content": message})
 
-    conversation.append({"role": "user", "content": message})
+    # Build conversation for API
+    conversation = [{"role": m["role"], "content": m["content"]} for m in history]
 
     try:
-        # Primary API call
         response = requests.post(
             "https://router.huggingface.co/v1/chat/completions",
             headers={
                 "Authorization": f"Bearer {HF_TOKEN}",
                 "Content-Type": "application/json"
             },
-            json={
-                "model": "deepseek-ai/DeepSeek-V3.2-Exp:novita",
-                "messages": conversation
-            }
+            json={"model": "deepseek-ai/DeepSeek-V3.2-Exp:novita", "messages": conversation}
         )
         if response.status_code != 200:
             raise Exception(f"API Error: {response.status_code} - {response.text}")
-        
-        data = response.json()
-        reply = data["choices"][0]["message"]["content"]
 
+        reply = response.json()["choices"][0]["message"]["content"]
+
+        # Format math nicely
         reply = reply.replace("Step", "\n\n**Step").replace(":", ":**").replace("\\[", "\n\n\\[").replace("\\]", "\\]\n\n")
 
-        history.append((message, reply))
+        # Add assistant reply
+        history.append({"role": "assistant", "content": reply})
         save_memory(history)
         return history, history
 
     except Exception as e:
         print("Backend Error:", e)
-        error_msg = "😅 EduAI is having trouble connecting right now. Check your HF_TOKEN or try again later!"
-        history.append((message, error_msg))
+        error_msg = "😅 EduAI is having trouble connecting. Check HF_TOKEN or try again later!"
+        history.append({"role": "assistant", "content": error_msg})
         return history, history
 
-# Rest of the code (unchanged, including UI and CSS)
-
-# 📘 Sidebar context update
+# Sidebar context update
 def update_context(choice):
     if not choice:
         return "📘 **You are in General Mode.** Ask EduAI anything about your studies!"
     return f"📘 **You selected {choice} mode.** Ask anything related to this topic!"
 
-# 🧹 Clear chat memory
+# Clear memory
 def clear_memory():
     if os.path.exists(MEMORY_FILE):
         os.remove(MEMORY_FILE)
     return [], "🧹 Chat memory cleared! Start fresh."
 
-# 🖌 Custom CSS (unchanged)
+# Custom CSS for professional look
 custom_css = """
-@import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;600&display=swap');
-
-body { 
-    font-family: 'Roboto', sans-serif; 
-    margin: 0; 
-    padding: 0; 
-    background: #1e1e1e; 
-    color: #e0e0e0; 
-    min-height: 100vh; 
-}
-.gradio-container { 
-    background: transparent; 
-    box-shadow: none; 
-}
-.header-title { 
-    font-size: 36px; 
-    text-align: center; 
-    margin-bottom: 20px; 
-    font-weight: 600; 
-    color: #007BFF; 
-    text-shadow: 0 0 10px rgba(0, 123, 255, 0.5); 
-}
-.sidebar { 
-    background: #2a2a2a; 
-    padding: 20px; 
-    border-radius: 12px; 
-    box-shadow: 0 4px 12px rgba(0,0,0,0.5); 
-    border: 1px solid #333; 
-}
-.menu-title { 
-    font-size: 20px; 
-    font-weight: 500; 
-    margin-bottom: 15px; 
-    color: #007BFF; 
-}
-.accordion { 
-    border-radius: 10px; 
-    margin-bottom: 10px; 
-    box-shadow: 0 2px 6px rgba(0,0,0,0.3); 
-}
-.accordion-header { 
-    background: #333; 
-    color: #e0e0e0; 
-    border-radius: 10px; 
-    padding: 12px; 
-    font-weight: 500; 
-    cursor: pointer; 
-    transition: background 0.3s; 
-}
-.accordion-header:hover { 
-    background: #007BFF; 
-    color: #1e1e1e; 
-}
-.main-chat { 
-    padding: 20px; 
-    background: #1e1e1e; 
-}
-.context-box { 
-    background: #2a2a2a; 
-    padding: 15px; 
-    border-radius: 10px; 
-    box-shadow: 0 2px 6px rgba(0,0,0,0.3); 
-    margin-bottom: 15px; 
-    border: 1px solid #333; 
-    color: #e0e0e0; 
-}
-.chatbox { 
-    border-radius: 10px; 
-    box-shadow: 0 2px 6px rgba(0,0,0,0.3); 
-    background: #252525; 
-    border: 1px solid #333; 
-}
-.chatbox .message.user { 
-    background: #007BFF; 
-    color: #1e1e1e; 
-    border-radius: 12px; 
-    padding: 12px 15px; 
-    margin: 8px 0; 
-    box-shadow: 0 2px 6px rgba(0, 123, 255, 0.3); 
-}
-.chatbox .message.bot { 
-    background: #333; 
-    color: #e0e0e0; 
-    border-radius: 12px; 
-    padding: 12px 15px; 
-    margin: 8px 0; 
-    box-shadow: 0 2px 6px rgba(0,0,0,0.3); 
-}
-.input-row { 
-    display: flex; 
-    align-items: center; 
-    gap: 10px; 
-}
-.chat-input { 
-    flex: 1; 
-    border-radius: 10px; 
-    padding: 12px; 
-    border: 1px solid #555; 
-    background: #2a2a2a; 
-    color: #e0e0e0; 
-    transition: border-color 0.3s; 
-}
-.chat-input:focus { 
-    border-color: #007BFF; 
-    outline: none; 
-}
-.file-upload { 
-    width: 40px; 
-    height: 40px; 
-    border: 1px solid #555; 
-    border-radius: 10px; 
-    background: #2a2a2a; 
-    color: #e0e0e0; 
-    display: flex; 
-    align-items: center; 
-    justify-content: center; 
-    cursor: pointer; 
-    transition: background 0.3s; 
-    font-size: 18px; 
-}
-.file-upload:hover { 
-    background: #007BFF; 
-    color: #1e1e1e; 
-}
-.btn-clear { 
-    background: #f44336; 
-    color: white; 
-    border-radius: 10px; 
-    padding: 10px 15px; 
-    border: none; 
-    font-weight: 500; 
-    cursor: pointer; 
-    transition: background 0.3s; 
-}
-.btn-clear:hover { 
-    background: #d32f2f; 
-}
-.about-text { 
-    font-size: 14px; 
-    color: #aaa; 
-    line-height: 1.5; 
-}
-.radio { 
-    margin: 10px 0; 
-}
-.radio label { 
-    color: #e0e0e0; 
-}
+@import url('https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&display=swap');
+body { font-family: 'Roboto', sans-serif; background:#1e1e1e; color:#e0e0e0; min-height:100vh; }
+.header-title { font-size:36px; text-align:center; margin-bottom:20px; color:#007BFF; font-weight:700; text-shadow:0 0 10px rgba(0,123,255,0.5); }
+.sidebar { background:#2a2a2a; padding:20px; border-radius:12px; box-shadow:0 4px 12px rgba(0,0,0,0.5); border:1px solid #333; }
+.menu-title { font-size:20px; font-weight:500; margin-bottom:15px; color:#007BFF; }
+.accordion { background:#2a2a2a; border-radius:12px; padding:10px; margin-bottom:10px; box-shadow:0 3px 8px rgba(0,0,0,0.5); }
+.accordion-header { background:#333; color:#e0e0e0; border-radius:10px; padding:12px; font-weight:500; cursor:pointer; transition:0.3s; }
+.accordion-header:hover { background:#007BFF; color:#1e1e1e; }
+.main-chat { padding:20px; background:#1e1e1e; }
+.context-box { background:#2a2a2a; padding:15px; border-radius:10px; box-shadow:0 2px 6px rgba(0,0,0,0.3); margin-bottom:15px; border:1px solid #333; color:#e0e0e0; }
+.chatbox .message.user { background:#007BFF; color:#fff; border-radius:20px; padding:12px 18px; margin:8px 0; }
+.chatbox .message.bot { background:#333; color:#e0e0e0; border-radius:20px; padding:12px 18px; margin:8px 0; }
+.input-row { display:flex; gap:10px; }
+.chat-input { flex:1; border-radius:12px; padding:12px; border:1px solid #555; background:#2a2a2a; color:#e0e0e0; }
+.chat-input:focus { border-color:#007BFF; outline:none; }
+.btn-send { background:#7c3aed; color:#fff; border-radius:12px; padding:12px 18px; border:none; cursor:pointer; transition:0.3s; }
+.btn-send:hover { background:#5b21b6; }
+.file-upload { width:50px; height:50px; border:1px solid #555; border-radius:12px; background:#2a2a2a; display:flex; align-items:center; justify-content:center; cursor:pointer; font-size:20px; color:#e0e0e0; transition:0.3s; }
+.file-upload:hover { background:#007BFF; color:#1e1e1e; }
+.btn-clear { background:#f44336; color:white; border-radius:12px; padding:10px 15px; border:none; cursor:pointer; transition:0.3s; }
+.btn-clear:hover { background:#d32f2f; }
+.about-text { font-size:14px; color:#aaa; line-height:1.5; }
+.radio label { color:#e0e0e0; margin:8px 0; }
 """
 
-# 🎨 Gradio Interface (unchanged)
+# Launch Gradio UI
 with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:
-    gr.Markdown("# 🎓 **EduAI — Your Professional Study Companion**", elem_classes="header-title")
+    gr.Markdown("# 🎓 **EduAI — Professional Study Companion**", elem_classes="header-title")
 
     with gr.Row():
+        # Sidebar
         with gr.Column(scale=1, min_width=250, elem_classes="sidebar"):
             gr.Markdown("### 🧭 **Main Menu**", elem_classes="menu-title")
 
             with gr.Accordion("📚 Subject Tutor", open=False):
-                subj = gr.Radio(
-                    ["Science 🧪", "ICT 💻", "English 📘", "Mathematics ➗"],
-                    label="Choose a subject",
-                    type="index"
-                )
+                subj = gr.Radio(["Science 🧪", "ICT 💻", "English 📘", "Mathematics ➗"], label="Choose a subject", type="index")
 
             with gr.Accordion("🗓 Study Planner", open=False):
-                planner = gr.Radio(
-                    ["View Plan 📅", "Add Task ✏️", "Study Tips 💡"],
-                    label="Planner Options",
-                    type="index"
-                )
+                planner = gr.Radio(["View Plan 📅", "Add Task ✏️", "Study Tips 💡"], label="Planner Options", type="index")
 
             with gr.Accordion("🌐 Languages", open=False):
-                lang = gr.Radio(
-                    ["Learn Sinhala 🇱🇰", "Learn Tamil 🇮🇳", "Learn English 🇬🇧", "Learn Spanish 🇪🇸"],
-                    label="Language Options",
-                    type="index"
-                )
+                lang = gr.Radio(["Learn Sinhala 🇱🇰","Learn Tamil 🇮🇳","Learn English 🇬🇧","Learn Spanish 🇪🇸"], label="Language Options", type="index")
 
             with gr.Accordion("⚙️ Settings", open=False):
                 clear_btn = gr.Button("🧹 Clear Memory", elem_classes="btn-clear")
 
             with gr.Accordion("👩‍🎓 About", open=False):
-                gr.Markdown("""
-                EduAI – developed by **Wafa Fazly** using a pre-trained AI model.  
-                Helps learners understand **Science, ICT, English, and more** in a simple, friendly way! 💬
-                """, elem_classes="about-text")
+                gr.Markdown("EduAI – developed by **Wafa Fazly** using a pre-trained AI model. Helps learners understand **Science, ICT, English, and more**.", elem_classes="about-text")
 
+        # Main chat
         with gr.Column(scale=4, elem_classes="main-chat"):
-            context_display = gr.Markdown(
-                "📘 **You are in General Mode.** Ask EduAI anything about your studies!",
-                elem_classes="context-box"
-            )
-
-            chatbot = gr.Chatbot(
-                label="EduAI Chat",
-                height=500,
-                render_markdown=True,
-                type="messages",
-                latex_delimiters=[{"left": "$$", "right": "$$", "display": True}, {"left": "\\[", "right": "\\]", "display": True}]
-            )
-
-            with gr.Row(elem_classes="input-row"):
-                msg = gr.Textbox(label="Ask EduAI:", placeholder="Type your question here...", elem_classes="chat-input", show_label=False)
-                file_input = gr.File(label="📎", file_types=[".pdf", ".docx", ".png", ".jpg"], elem_classes="file-upload", show_label=False)
-
-    subj.change(update_context, inputs=subj, outputs=context_display)
-    planner.change(update_context, inputs=planner, outputs=context_display)
-    lang.change(update_context, inputs=lang, outputs=context_display)
-    msg.submit(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, chatbot])
-    clear_btn.click(clear_memory, outputs=[chatbot, context_display])
-
-iface.launch()
+            context_display = gr.Markdown("📘 **You are in General Mode.** Ask EduAI anything