Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import requests
 import json
-import re
+import random
 
 # 🌍 Web search function
 def search_web(query):
@@ -43,20 +43,17 @@ def chat_with_model(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, ""
+        return history, history
 
-    # 📂 If file is uploaded
     if file_input:
-        file_name = os.path.basename(file_input.name)
+        file_name = file_input.name
         message += f"\n\n📎 (User uploaded a file named '{file_name}')"
 
-    # 🧠 Build conversation
     conversation = [
         {"role": "system", "content": (
             "You are EduAI, a multilingual educational AI assistant created by a Sri Lankan student named Wafa Fazly. "
@@ -72,7 +69,6 @@ def chat_with_model(message, history, context, file_input=None):
 
     conversation.append({"role": "user", "content": message})
 
-    # 🚀 Send to Hugging Face model
     try:
         response = requests.post(
             "https://router.huggingface.co/v1/chat/completions",
@@ -81,7 +77,7 @@ def chat_with_model(message, history, context, file_input=None):
                 "Content-Type": "application/json"
             },
             json={
-                "model": "deepseek-ai/DeepSeek-V2.5",
+                "model": "deepseek-ai/DeepSeek-V3.2-Exp:novita",
                 "messages": conversation
             }
         )
@@ -89,19 +85,23 @@ def chat_with_model(message, history, context, file_input=None):
         data = response.json()
         reply = data["choices"][0]["message"]["content"]
 
-        # 🧮 Clean up math formatting
-        reply = re.sub(r"\bStep (\d+):", r"\n\n**Step \1:**", reply)
+        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, ""
+        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, ""
+        return history, history
+
 
 # 📘 Sidebar context update
 def update_context(choice):
@@ -109,81 +109,34 @@ def update_context(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."
 
-# 🧩 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}]
-            }
-        )
+# 🧠 MCQ QUIZ GENERATOR
+quiz_data = [
+    {"question": "Which planet is known as the Red Planet?", "options": ["Earth", "Mars", "Venus", "Jupiter"], "answer": "Mars"},
+    {"question": "What is the chemical symbol for water?", "options": ["H2O", "O2", "CO2", "NaCl"], "answer": "H2O"},
+    {"question": "What is 5 × 6?", "options": ["11", "30", "25", "36"], "answer": "30"},
+    {"question": "Which gas do humans breathe in to survive?", "options": ["Carbon dioxide", "Oxygen", "Nitrogen", "Hydrogen"], "answer": "Oxygen"},
+    {"question": "What is the largest planet in our Solar System?", "options": ["Earth", "Mars", "Jupiter", "Saturn"], "answer": "Jupiter"}
+]
 
-        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
+def generate_quiz():
+    selected = random.sample(quiz_data, 3)
+    quiz_html = ""
+    for i, q in enumerate(selected, 1):
+        quiz_html += f"<div><b>{i}. {q['question']}</b><br>"
+        for opt in q['options']:
+            quiz_html += f"<input type='checkbox' name='q{i}' value='{opt}'> {opt}<br>"
+        quiz_html += "<br></div>"
+    quiz_html += "<br><b>✅ Select your answers and review!</b>"
+    return quiz_html
 
-    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:
@@ -211,11 +164,9 @@ 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("🧠 MCQ Quiz Generator", open=False):
+                gen_btn = gr.Button("🎯 Generate Quiz")
+                quiz_output = gr.HTML("<i>Click 'Generate Quiz' to start!</i>")
 
             with gr.Accordion("⚙️ Settings", open=False):
                 clear_btn = gr.Button("🧹 Clear Memory")
@@ -243,9 +194,8 @@ 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, msg])
-    msg.submit(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, msg])
+    send.click(chat_with_model, inputs=[msg, chatbot, context_display, file_input], outputs=[chatbot, chatbot])
     clear_btn.click(clear_memory, outputs=[chatbot, context_display])
-    generate_quiz_btn.click(generate_dynamic_quiz, inputs=quiz_topic, outputs=quiz_output)
+    gen_btn.click(generate_quiz, outputs=quiz_output)
 
 iface.launch()