Update app.py

app.py

@@ -1,70 +1,378 @@
+import os
 import gradio as gr
-from huggingface_hub import InferenceClient
+import requests
+import json
+import time
 
+# Test HF_TOKEN
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    print("Error: HF_TOKEN not set. Please set your Hugging Face API token.")
+else:
+    print("HF_TOKEN loaded successfully.")
 
-def respond(
-    message,
-    history: list[dict[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-    hf_token: gr.OAuthToken,
-):
-    """
-    For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-    """
-    client = InferenceClient(token=hf_token.token, model="openai/gpt-oss-20b")
+# 🌍 Web search function (unchanged)
+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}"
 
-    messages = [{"role": "system", "content": system_message}]
+# 🧠 Memory setup (unchanged)
+MEMORY_FILE = "memory.json"
 
-    messages.extend(history)
+def load_memory():
+    if os.path.exists(MEMORY_FILE):
+        with open(MEMORY_FILE, "r") as f:
+            return json.load(f)
+    return []
 
-    messages.append({"role": "user", "content": message})
+def save_memory(memory):
+    with open(MEMORY_FILE, "w") as f:
+        json.dump(memory, f)
 
-    response = ""
+memory = load_memory()
 
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        choices = message.choices
-        token = ""
-        if len(choices) and choices[0].delta.content:
-            token = choices[0].delta.content
+# 💬 Chat function (removed file_input handling)
+def chat_with_model(message, history, context):
+    if not isinstance(history, list):
+        history = []
 
-        response += token
-        yield response
+    if message.lower().startswith("search "):
+        query = message[7:]
+        search_result = search_web(query)
+        timestamp = time.strftime("%H:%M")
+        history.append((f"{message} <span class='timestamp'>{timestamp}</span>", f"🔎 Here's what I found online:\n\n{search_result} <span class='timestamp'>{timestamp}</span>"))
+        save_memory(history)
+        return history, history
 
+    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.split(' <span')[0]})  # Strip timestamp for API
+        conversation.append({"role": "assistant", "content": past_bot.split(' <span')[0]})
+
+    conversation.append({"role": "user", "content": message})
+
+    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
+            }
+        )
+        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 = reply.replace("Step", "\n\n**Step").replace(":", ":**").replace("\\[", "\n\n\\[").replace("\\]", "\\]\n\n")
+
+        timestamp = time.strftime("%H:%M")
+        history.append((f"{message} <span class='timestamp'>{timestamp}</span>", f"{reply} <span class='timestamp'>{timestamp}</span>"))
+        save_memory(history)
+        return history, history
+
+    except Exception as e:
+        print("Backend Error:", e)
+        timestamp = time.strftime("%H:%M")
+        error_msg = f"😅 EduAI is having trouble connecting right now. Check your HF_TOKEN or try again later! <span class='timestamp'>{timestamp}</span>"
+        history.append((f"{message} <span class='timestamp'>{timestamp}</span>", error_msg))
+        return history, history
+
+# 📘 Sidebar context update (unchanged)
+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 (unchanged)
+def clear_memory():
+    if os.path.exists(MEMORY_FILE):
+        os.remove(MEMORY_FILE)
+    return [], "🧹 Chat memory cleared! Start fresh."
+
+# 🖌 Custom CSS (unchanged, but input-row simplified)
+custom_css = """
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&display=swap');
+
+body { 
+    font-family: 'Inter', sans-serif; 
+    margin: 0; 
+    padding: 0; 
+    background: #0f0f0f; 
+    color: #e0e0e0; 
+    min-height: 100vh; 
+}
+.gradio-container { 
+    background: transparent; 
+    box-shadow: none; 
+}
+.header-title { 
+    font-size: 28px; 
+    text-align: center; 
+    margin-bottom: 20px; 
+    font-weight: 600; 
+    color: #007BFF; 
+    text-shadow: 0 0 10px rgba(0, 123, 255, 0.5); 
+}
+.sidebar { 
+    background: #1a1a1a; 
+    padding: 20px; 
+    border-radius: 12px; 
+    box-shadow: 0 4px 12px rgba(0,0,0,0.5); 
+    border: 1px solid #333; 
+    width: 280px; 
+}
+.menu-title { 
+    font-size: 18px; 
+    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: #2a2a2a; 
+    color: #e0e0e0; 
+    border-radius: 10px; 
+    padding: 12px; 
+    font-weight: 500; 
+    cursor: pointer; 
+    transition: background 0.3s; 
+}
+.accordion-header:hover { 
+    background: #007BFF; 
+    color: #0f0f0f; 
+}
+.main-chat { 
+    padding: 20px; 
+    background: #0f0f0f; 
+    display: flex; 
+    flex-direction: column; 
+    height: calc(100vh - 100px); 
+}
+.context-box { 
+    background: #1a1a1a; 
+    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; 
+    font-size: 14px; 
+}
+.chatbox { 
+    flex: 1; 
+    border-radius: 10px; 
+    box-shadow: 0 2px 6px rgba(0,0,0,0.3); 
+    background: #1a1a1a; 
+    border: 1px solid #333; 
+    padding: 15px; 
+    overflow-y: auto; 
+}
+.message { 
+    display: flex; 
+    margin: 15px 0; 
+    animation: fadeIn 0.5s ease-in; 
+}
+@keyframes fadeIn { 
+    from { opacity: 0; transform: translateY(10px); } 
+    to { opacity: 1; transform: translateY(0); } 
+}
+.message.user { 
+    justify-content: flex-end; 
+}
+.message.ai { 
+    justify-content: flex-start; 
+}
+.message .avatar { 
+    width: 40px; 
+    height: 40px; 
+    border-radius: 50%; 
+    margin: 0 10px; 
+    background: #007BFF; 
+    display: flex; 
+    align-items: center; 
+    justify-content: center; 
+    color: white; 
+    font-weight: 600; 
+    font-size: 16px; 
+}
+.message.ai .avatar { 
+    background: #333; 
+}
+.message .bubble { 
+    max-width: 70%; 
+    padding: 12px 16px; 
+    border-radius: 18px; 
+    position: relative; 
+}
+.message.user .bubble { 
+    background: #007BFF; 
+    color: white; 
+}
+.message.ai .bubble { 
+    background: #2a2a2a; 
+    color: #e0e0e0; 
+    border: 1px solid #333; 
+}
+.timestamp { 
+    font-size: 10px; 
+    opacity: 0.7; 
+    margin-top: 5px; 
+    display: block; 
+}
+.typing { 
+    display: none; 
+    font-style: italic; 
+    color: #aaa; 
+    padding: 10px; 
+    animation: pulse 1.5s infinite; 
+}
+@keyframes pulse { 
+    0%, 100% { opacity: 0.5; } 
+    50% { opacity: 1; } 
+}
+.input-row { 
+    margin-top: 15px; 
+}
+.chat-input { 
+    width: 100%; 
+    border-radius: 20px; 
+    padding: 12px 16px; 
+    border: 1px solid #555; 
+    background: #1a1a1a; 
+    color: #e0e0e0; 
+    transition: border-color 0.3s; 
+    font-size: 14px; 
+}
+.chat-input:focus { 
+    border-color: #007BFF; 
+    outline: none; 
+}
+.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; 
+}
 """
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-chatbot = gr.ChatInterface(
-    respond,
-    type="messages",
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
-
-with gr.Blocks() as demo:
-    with gr.Sidebar():
-        gr.LoginButton()
-    chatbot.render()
-
-
-if __name__ == "__main__":
-    demo.launch()
+
+# 🎨 Gradio Interface without file uploader
+with gr.Blocks(theme=gr.themes.Base(), css=custom_css) as iface:
+    gr.Markdown("# 🤖 **EduAI**", elem_classes="header-title")
+
+    with gr.Row():
+        with gr.Column(scale=1, elem_classes="sidebar"):
+            gr.Markdown("### 🧭 **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"
+                )
+
+            with gr.Accordion("🗓 Study Planner", open=False):
+                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"
+                )
+
+            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")
+
+        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="",
+                height=500,
+                render_markdown=True,
+                type="messages",
+                latex_delimiters=[{"left": "$$", "right": "$$", "display": True}, {"left": "\\[", "right": "\\]", "display": True}]
+            )
+
+            typing_indicator = gr.Markdown("", elem_classes="typing")
+
+            with gr.Row(elem_classes="input-row"):
+                msg = gr.Textbox(placeholder="Message EduAI...", elem_classes="chat-input", show_label=False)
+
+    # Event Handlers with typing indicator
+    def start_typing():
+        return "EduAI is typing..."
+
+    def stop_typing():
+        return ""
+
+    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(start_typing, outputs=typing_indicator).then(
+        chat_with_model, inputs=[msg, chatbot, context_display], outputs=[chatbot, chatbot]
+    ).then(stop_typing, outputs=typing_indicator)
+    clear_btn.click(clear_memory, outputs=[chatbot, context_display])
+
+iface.launch()