Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 import requests
 import json
+import re
 
 # 🌍 Web search function
 def search_web(query):
@@ -15,7 +16,7 @@ def search_web(query):
             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
+            return " ".join(topics[:3])
         else:
             return "No useful information found."
     except Exception as e:
@@ -48,11 +49,11 @@ def chat_with_model(message, history, context, file_input=None):
         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
+        return history, ""
 
     # 📂 If file is uploaded
     if file_input:
-        file_name = file_input.name
+        file_name = os.path.basename(file_input.name)
         message += f"\n\n📎 (User uploaded a file named '{file_name}')"
 
     # 🧠 Build conversation
@@ -80,7 +81,7 @@ def chat_with_model(message, history, context, file_input=None):
                 "Content-Type": "application/json"
             },
             json={
-                "model": "deepseek-ai/DeepSeek-V3.2-Exp:novita",
+                "model": "deepseek-ai/DeepSeek-V2.5",
                 "messages": conversation
             }
         )
@@ -89,23 +90,18 @@ def chat_with_model(message, history, context, file_input=None):
         reply = data["choices"][0]["message"]["content"]
 
         # 🧮 Clean up math formatting
-        reply = reply.replace("Step", "\n\n**Step")
-        reply = reply.replace(":", ":**")
+        reply = re.sub(r"\bStep (\d+):", r"\n\n**Step \1:**", reply)
         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}"
-
         history.append((message, reply))
         save_memory(history)
-        return history, history
+        return 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
+        return history, ""
 
 # 📘 Sidebar context update
 def update_context(choice):
@@ -119,6 +115,76 @@ def clear_memory():
         os.remove(MEMORY_FILE)
     return [], "🧹 Chat memory cleared! Start fresh."
 
+# 🧩 Dynamic MCQ Quiz Generator
+def generate_dynamic_quiz(topic):
+    try:
+        if not topic.strip():
+            return [gr.Markdown("⚠️ Please enter a topic to generate the quiz.")]
+
+        prompt = f"""
+        Create 5 multiple-choice questions about "{topic}".
+        Each question should have 4 options (A, B, C, D) and only one correct answer.
+        Format the response in JSON like this:
+        [
+          {{
+            "question": "...",
+            "options": ["...", "...", "...", "..."],
+            "answer": "..."
+          }},
+          ...
+        ]
+        """
+
+        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-V2.5",
+                "messages": [{"role": "user", "content": prompt}]
+            }
+        )
+
+        data = response.json()
+        raw_text = data["choices"][0]["message"]["content"]
+
+        # 🧩 Try parsing JSON
+        try:
+            quiz_data = json.loads(raw_text)
+        except:
+            quiz_data = []
+            for block in raw_text.split("\n\n"):
+                if "?" in block:
+                    parts = block.split("\n")
+                    q = parts[0]
+                    opts = [p.strip("- ").strip() for p in parts[1:5] if p.strip()]
+                    if len(opts) == 4:
+                        quiz_data.append({
+                            "question": q,
+                            "options": opts,
+                            "answer": ""
+                        })
+
+        # 🎯 Build UI dynamically
+        quiz_elements = []
+        for q in quiz_data:
+            quiz_elements.append(gr.Markdown(f"### ❓ {q['question']}"))
+            quiz_elements.append(gr.CheckboxGroup(
+                choices=q["options"],
+                label="Select your answer:",
+                interactive=True
+            ))
+
+        if not quiz_elements:
+            return [gr.Markdown("😅 Couldn't generate a proper quiz. Try another topic!")]
+
+        return quiz_elements
+
+    except Exception as e:
+        return [gr.Markdown(f"Error generating quiz: {e}")]
+
 # 🎨 Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
     gr.Markdown("# 🎓 **EduAI — Your Smart Study Companion**")
@@ -145,6 +211,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
                     label="Language Options"
                 )
 
+            # 🧩 MCQ Quiz Generator
+            with gr.Accordion("🧩 MCQ Quiz Generator", open=False):
+                quiz_topic = gr.Textbox(label="Enter a topic (e.g., Photosynthesis, AI, Newton's Laws):")
+                generate_quiz_btn = gr.Button("Generate Quiz 🧠")
+                quiz_output = gr.Column()
+
             with gr.Accordion("⚙️ Settings", open=False):
                 clear_btn = gr.Button("🧹 Clear Memory")
 
@@ -171,7 +243,9 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as iface:
     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])
+    send.click(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, msg])
+    msg.submit(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, msg])
     clear_btn.click(clear_memory, outputs=[chatbot, context_display])
+    generate_quiz_btn.click(generate_dynamic_quiz, inputs=quiz_topic, outputs=quiz_output)
 
-iface.launch()
+iface.launch()