Update app.py

app.py

@@ -3,175 +3,120 @@ import gradio as gr
 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])  # just a few results
-        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)
+MODEL_URL = "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-V3.2-Exp:novita"
 
-memory = load_memory()
-
-# 💬 Chat function
-def chat_with_model(message, history, context, file_input=None):
+# 🌍 AI response function
+def ai_response(message, history, context, file_input=None):
     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
-
-    # 📂 If file is uploaded
     if file_input:
-        file_name = file_input.name
-        message += f"\n\n📎 (User uploaded a file named '{file_name}')"
+        message += f"\n\n📎 (User uploaded a file named '{file_input.name}')"
 
-    # 🧠 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."
+            "You are EduAI, a multilingual educational AI assistant. "
+            "Explain concepts step-by-step, generate MCQs dynamically when requested, "
+            "and provide study tips. Use Markdown and LaTeX for math. Keep answers professional and clear."
         )}
     ]
-
     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
-            }
+            MODEL_URL,
+            headers={"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"},
+            json={"inputs": conversation, "parameters":{"max_new_tokens":512}}
         )
-
         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")
-
-        # ✅ Add Markdown + LaTeX support
-        if "\\" in reply or "log_" in reply or "^" in reply:
-            reply = f"{reply}"
-
+        if isinstance(data, dict) and "error" in data:
+            reply = "❌ Error contacting model: " + data["error"]
+        else:
+            reply = data[0]["generated_text"] if isinstance(data, list) else str(data)
         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!"))
+        history.append((message, f"❌ Error contacting model: {e}"))
         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
+# 🧠 Clear chat memory
+MEMORY_FILE = "memory.json"
 def clear_memory():
     if os.path.exists(MEMORY_FILE):
         os.remove(MEMORY_FILE)
     return [], "🧹 Chat memory cleared! Start fresh."
 
-# 🎨 Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
-    gr.Markdown("# 🎓 **EduAI — Your Smart Study Companion**")
+# 📘 Sidebar context
+def update_context(choice):
+    if not choice:
+        return "📘 **General Mode.** Ask EduAI anything!"
+    return f"📘 **You selected {choice}.**"
+
+# 🧩 Dynamic MCQ generator via AI
+def generate_mcq(topic):
+    prompt = f"Create a short 3-question MCQ quiz about {topic} with 4 options each. Indicate the correct answer."
+    try:
+        response = requests.post(
+            MODEL_URL,
+            headers={"Authorization": f"Bearer {HF_TOKEN}", "Content-Type": "application/json"},
+            json={"inputs": prompt, "parameters":{"max_new_tokens":400}}
+        )
+        data = response.json()
+        text = data[0]["generated_text"] if isinstance(data, list) else str(data)
+        return f"<div style='color:white'>{text.replace(chr(10), '<br>')}</div>"
+    except Exception as e:
+        return f"❌ Error generating MCQs: {e}"
+
+# 🎨 Gradio UI
+custom_css = """
+body {background-color:#1e1e2f; color:white; font-family:'Poppins', sans-serif;}
+#chatbot {background: rgba(30,30,50,0.85); border-radius:12px; backdrop-filter: blur(8px);}
+button.primary {background: linear-gradient(90deg,#7c3aed,#a855f7); color:white; border:none; border-radius:8px;}
+button.primary:hover {background: linear-gradient(90deg,#6d28d9,#9333ea);}
+"""
+
+with gr.Blocks(css=custom_css) as iface:
+    gr.Markdown("<h1 style='text-align:center; color:#a855f7'>🎓 EduAI — Smart Study Companion</h1>")
 
     with gr.Row():
-        with gr.Column(scale=1, min_width=230):
-            gr.Markdown("### 🧭 **Main Menu**")
+        with gr.Column(scale=1, min_width=220):
+            gr.Markdown("### 🧭 Main Menu")
+
+            with gr.Accordion("📚 Study Tutor", open=False):
+                tutor_sub = gr.Radio(["Math", "Science", "ICT", "English"], label="Select Subject")
 
-            with gr.Accordion("📚 Subject Tutor", open=False):
-                subj = gr.Radio(
-                    ["Science 🧪", "ICT 💻", "English 📘", "Mathematics ➗"],
-                    label="Choose a subject"
-                )
+            with gr.Accordion("🧠 MCQ Generator", open=False):
+                mcq_topic = gr.Textbox(label="Topic for Quiz")
+                gen_mcq_btn = gr.Button("Generate MCQ", elem_classes="primary")
+                mcq_output = gr.HTML("<i>Enter a topic and click 'Generate MCQ'</i>")
 
             with gr.Accordion("🗓 Study Planner", open=False):
-                planner = gr.Radio(
-                    ["View Plan 📅", "Add Task ✏️", "Study Tips 💡"],
-                    label="Planner Options"
-                )
+                planner_opt = 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("📖 Study Advisor", open=False):
+                advisor_txt = gr.Textbox(label="Ask for Study Advice")
+                advisor_btn = gr.Button("Get Advice", elem_classes="primary")
+                advisor_output = gr.HTML("<i>Ask your study question here</i>")
 
             with gr.Accordion("⚙️ Settings", open=False):
-                clear_btn = gr.Button("🧹 Clear Memory")
-
-            with gr.Accordion("👩‍🎓 About", open=False):
-                gr.Markdown("""
-                EduAI – developed and fine-tuned by **Wafa Fazly** using a pre-trained AI model,  
-                to help learners understand **Science, ICT, English, and more** —  
-                in a simple and friendly way! 💬
-                """)
+                clear_btn = gr.Button("🧹 Clear Chat Memory")
 
         with gr.Column(scale=4):
-            context_display = gr.Markdown("📘 **You are in General Mode.** Ask EduAI anything about your studies!")
-            chatbot = gr.Chatbot(
-                label="EduAI Chat",
-                height=450,
-                render_markdown=True,
-                latex_delimiters=[{"left": "$$", "right": "$$", "display": True}, {"left": "\\[", "right": "\\]", "display": True}]
-            )
-            msg = gr.Textbox(label="Ask EduAI:")
-            file_input = gr.File(label="📂 Upload a study file (PDF, DOCX, or image):")
-            send = gr.Button("Send ✈️")
-
-    # 🪄 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, file_input], outputs=[chatbot, chatbot])
+            context_display = gr.Markdown("📘 **General Mode**")
+            chatbot = gr.Chatbot(elem_id="chatbot", height=480)
+            user_input = gr.Textbox(label="Type your message")
+            send_btn = gr.Button("Send ✈️", elem_classes="primary")
+
+    # Event bindings
+    send_btn.click(ai_response, inputs=[user_input, chatbot, context_display, None], outputs=[chatbot, chatbot])
     clear_btn.click(clear_memory, outputs=[chatbot, context_display])
+    gen_mcq_btn.click(generate_mcq, inputs=[mcq_topic], outputs=mcq_output)
+    advisor_btn.click(ai_response, inputs=[advisor_txt, chatbot, context_display, None], outputs=[chatbot, chatbot])
+    tutor_sub.change(update_context, inputs=tutor_sub, outputs=context_display)
+    planner_opt.change(update_context, inputs=planner_opt, outputs=context_display)
 
-iface.launch()
+iface.launch()