Update app.py

app.py

@@ -1,125 +1,186 @@
+import os
 import gradio as gr
-import time
-
-# ========================================
-# 💬 AI Response Function (Replace this with your real AI)
-# ========================================
-def ai_response(message, history):
-    # Example: Simulate AI typing delay
-    time.sleep(0.8)
-    # You can connect this to any API (e.g., DeepSeek, Hugging Face, OpenAI)
-    return f"🤖 AI: I understood your question — '{message}'. Let’s explore that together!"
-
-# ========================================
-# ⚙️ Chat Logic
-# ========================================
-def respond(message, history):
-    history = history or []
-    bot_message = ai_response(message, history)
-    history.append((message, bot_message))
-    return "", history
-
-
-# ========================================
-# 🎨 Gradio Interface
-# ========================================
-with gr.Blocks(css="""
-body {
-    background: linear-gradient(135deg, #0d0d0d 0%, #1a1a1a 100%);
-    color: #f0f0f0;
-    font-family: 'Roboto', sans-serif;
-}
-#chatbot {
-    background-color: transparent !important;
-    border: none !important;
-}
-.send-btn {
-    background: linear-gradient(135deg, #03dac6 0%, #00bfa5 100%) !important;
-    color: #0d0d0d !important;
-    border: none !important;
-    font-weight: bold;
-    border-radius: 25px !important;
-    padding: 10px 20px !important;
-    transition: 0.3s;
-}
-.send-btn:hover {
-    transform: scale(1.05);
-    box-shadow: 0 0 12px rgba(3, 218, 198, 0.6);
-}
-.sidebar {
-    background-color: #2a2a2a;
-    padding: 25px;
-    border-right: 2px solid #444;
-    box-shadow: inset -5px 0 10px rgba(0,0,0,0.4);
-    height: 100vh;
-}
-.sidebar h2 {
-    color: #03dac6;
-    text-align: center;
-    font-weight: 400;
-    margin-bottom: 25px;
-}
-.sidebar-btn {
-    display: block;
-    width: 100%;
-    background: transparent;
-    color: #f0f0f0;
-    border: none;
-    text-align: left;
-    padding: 12px 15px;
-    border-radius: 8px;
-    font-size: 15px;
-    margin-bottom: 5px;
-    cursor: pointer;
-    transition: all 0.3s ease;
-}
-.sidebar-btn:hover {
-    background-color: #444;
-    transform: translateX(5px);
-}
-.footer {
-    font-size: 12px;
-    text-align: center;
-    margin-top: auto;
-    color: #aaa;
-}
-""") as demo:
-
-    # ========== Layout ==========
+import requests
+import json
+
+# 🌍 Web search function
+def search_web(query):
+    try:
+        url = "https://api.duckduckgo.com/"
+        params = {"q": query, "format": "json", "no_html": 1, "skip_disambig": 1}
+        response = requests.get(url, params=params)
+        data = response.json()
+
+        if data.get("AbstractText"):
+            return data["AbstractText"]
+        elif data.get("RelatedTopics"):
+            topics = [t.get("Text", "") for t in data["RelatedTopics"] if "Text" in t]
+            return " ".join(topics[:3])
+        else:
+            return "No useful information found."
+    except Exception as e:
+        return f"Search error: {e}"
+
+# 🧠 Memory setup
+HF_TOKEN = os.getenv("HF_TOKEN")
+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
+def chat_with_model(message, history, context):
+    if not isinstance(history, list):
+        history = []
+
+    # 🌍 Web search mode
+    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}"))
+        save_memory(history)
+        return history, history
+
+    # 🧠 Build conversation
+    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."
+        )}
+    ]
+
+    for past_user, past_bot in history[-5:]:
+        conversation.append({"role": "user", "content": past_user})
+        conversation.append({"role": "assistant", "content": past_bot})
+
+    conversation.append({"role": "user", "content": message})
+
+    # 🚀 Send to Hugging Face model
+    try:
+        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
+            }
+        )
+
+        data = response.json()
+        reply = data["choices"][0]["message"]["content"]
+
+        # 🧮 Clean up math formatting
+        reply = reply.replace("Step", "\n\n**Step")
+        reply = reply.replace(":", ":**")
+        reply = reply.replace("\\[", "\n\n\\[")
+        reply = reply.replace("\\]", "\\]\n\n")
+
+        if "\\" in reply or "log_" in reply or "^" in reply:
+            reply = f"{reply}"
+
+        history.append((message, reply))
+        save_memory(history)
+        return history, history
+
+    except Exception as e:
+        print("Error:", e)
+        history.append((message, "😅 EduAI is having trouble connecting right now. Please try again later!"))
+        return history, history
+
+# 📘 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
+def clear_memory():
+    if os.path.exists(MEMORY_FILE):
+        os.remove(MEMORY_FILE)
+    return [], "🧹 Chat memory cleared! Start fresh."
+
+# 🎨 Gradio Interface (UI Improved)
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
+    gr.Markdown(
+        """
+        # 🎓 **EduAI — Your Smart Study Companion**
+        Welcome to **EduAI**, your friendly study assistant! 💬  
+        Get help in **Science, ICT, English, Mathematics**, and more.  
+        """
+    )
+
     with gr.Row():
-        # Sidebar Section
-        with gr.Column(scale=1, elem_classes="sidebar"):
-            gr.HTML("<h2>AI Study Agent</h2>")
-            tutor_btn = gr.Button("📘 Study Tutor", elem_classes="sidebar-btn")
-            planner_btn = gr.Button("🗓️ Study Planner", elem_classes="sidebar-btn")
-            advisor_btn = gr.Button("🧠 Study Advisor", elem_classes="sidebar-btn")
-            gr.Markdown("<hr>")
-            gr.Markdown("✨ <small>Designed by Cherry</small>", elem_classes="footer")
-
-        # Main Chat Section
+        with gr.Column(scale=1, min_width=230):
+            gr.Markdown("### 🧭 **Main Menu**")
+
+            with gr.Accordion("📚 Subject Tutor", open=False):
+                subj = gr.Radio(
+                    ["Science 🧪", "ICT 💻", "English 📘", "Mathematics ➗"],
+                    label="Choose a subject"
+                )
+
+            with gr.Accordion("🗓 Study Planner", open=False):
+                planner = gr.Radio(
+                    ["View Plan 📅", "Add Task ✏️", "Study Tips 💡"],
+                    label="Planner Options"
+                )
+
+            with gr.Accordion("🌐 Languages", open=False):
+                lang = gr.Radio(
+                    ["Learn Sinhala 🇱🇰", "Learn Tamil 🇮🇳", "Learn English 🇬🇧", "Learn Spanish 🇪🇸"],
+                    label="Language Options"
+                )
+
+            with gr.Accordion("⚙️ Settings", open=False):
+                clear_btn = gr.Button("🧹 Clear Memory")
+
+            with gr.Accordion("👩‍🎓 About", open=False):
+                gr.Markdown(
+                    """
+                    EduAI was designed and fine-tuned by **Wafa Fazly**,  
+                    a passionate Sri Lankan student 👩‍💻  
+                    to help learners explore **Science, ICT, English, and more** —  
+                    in a smart and friendly way! 🌟
+                    """
+                )
+
         with gr.Column(scale=4):
-            chatbot = gr.Chatbot(label="", elem_id="chatbot", height=550)
-            with gr.Row():
-                msg = gr.Textbox(placeholder="Type your question here...", show_label=False, scale=5)
-                send = gr.Button("Send", elem_classes="send-btn", scale=1)
-
-            # Chat connections
-            msg.submit(respond, [msg, chatbot], [msg, chatbot])
-            send.click(respond, [msg, chatbot], [msg, chatbot])
-
-            # Sidebar button responses
-            def sidebar_action(choice, history):
-                response = f"✨ You selected **{choice}** — How can I assist with that?"
-                history = history or []
-                history.append((f"{choice}", response))
-                return history
-
-            tutor_btn.click(sidebar_action, ["Study Tutor", chatbot], chatbot)
-            planner_btn.click(sidebar_action, ["Study Planner", chatbot], chatbot)
-            advisor_btn.click(sidebar_action, ["Study Advisor", chatbot], chatbot)
-
-
-# ========================================
-# 🚀 Launch App
-# ========================================
-demo.launch()
+            context_display = gr.Markdown("📘 **You are in General Mode.** Ask EduAI anything about your studies!")
+            chatbot = gr.Chatbot(
+                label="💬 EduAI Chat Window",
+                height=450,
+                render_markdown=True,
+                bubble_full_width=False,
+                latex_delimiters=[
+                    {"left": "$$", "right": "$$", "display": True},
+                    {"left": "\\[", "right": "\\]", "display": True}
+                ]
+            )
+            msg = gr.Textbox(
+                label="💭 Type your question here...",
+                placeholder="Ask EduAI anything about your studies..."
+            )
+            send = gr.Button("✨ Send Message")
+
+    # 🪄 Event handlers
+    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)
+    send.click(chat_with_model, inputs=[msg, chatbot, context_display], outputs=[chatbot, chatbot])
+    clear_btn.click(clear_memory, outputs=[chatbot, context_display])
+
+iface.launch()