Update app.py

app.py

@@ -2,15 +2,13 @@ import os
 import gradio as gr
 import requests
 import json
-import random
 
-# Hugging Face API token
-HF_TOKEN = os.getenv("HF_TOKEN")
+HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face token
 
-# =========================
+# ----------------------------
 # AI response via DeepSeek
-# =========================
-def ai_response(message, history=None):
+# ----------------------------
+def ai_response(message, history):
     if history is None:
         history = []
 
@@ -18,10 +16,9 @@ def ai_response(message, history=None):
         "You are EduAI, a professional multilingual educational assistant. "
         "Explain clearly with examples, use step-by-step for math, and Markdown/LaTeX for equations."
     )}]
-
-    for past_user, past_bot in history[-5:]:
-        conversation.append({"role": "user", "content": past_user})
-        conversation.append({"role": "assistant", "content": past_bot})
+    for item in history[-5:]:
+        conversation.append({"role": "user", "content": item["user"]})
+        conversation.append({"role": "assistant", "content": item["bot"]})
 
     conversation.append({"role": "user", "content": message})
 
@@ -37,26 +34,29 @@ def ai_response(message, history=None):
                 "messages": conversation
             }
         )
+        response.raise_for_status()
         data = response.json()
         reply = data["choices"][0]["message"]["content"]
-        history.append((message, reply))
-        return history, reply
     except Exception as e:
-        history.append((message, f"❌ Error contacting model: {e}"))
-        return history, history[-1][1]
+        reply = f"❌ Error: {e}"
+
+    history.append({"user": message, "bot": reply})
+    return history, reply
 
-# =========================
-# MCQ Generator (AI-based)
-# =========================
+# ----------------------------
+# MCQ Generator
+# ----------------------------
 def generate_mcq(subject):
-    prompt = f"Create 3 multiple-choice questions with 4 options each and the correct answer for {subject}. Return as JSON like: [{{'question':'...','options':['..'],'answer':'..'}}, ...]"
-    history, reply = ai_response(prompt)
+    prompt = f"Create 3 multiple-choice questions with 4 options each and the correct answer for {subject}. Return as JSON like: [{'question':'...','options':['..'],'answer':'..'}, ...]"
+    _, reply = ai_response(prompt, [])
     try:
         mcqs = json.loads(reply)
+        if not isinstance(mcqs, list):
+            raise ValueError("Invalid JSON")
     except:
-        return f"❌ Error generating MCQs: {reply}"
+        return f"<p style='color:red;'>❌ Error generating MCQs: AI output invalid.</p>"
 
-    quiz_html = "<div style='font-family:Poppins, sans-serif; color:#EEE;'>"
+    quiz_html = "<div style='font-family:Poppins,sans-serif; color:#EEE;'>"
     for i, q in enumerate(mcqs, 1):
         quiz_html += f"<div style='margin-bottom:15px; padding:12px; border-radius:10px; background:#2a2a2a;'>"
         quiz_html += f"<b style='color:#7C3AED;'>{i}. {q['question']}</b><br>"
@@ -67,20 +67,23 @@ def generate_mcq(subject):
     quiz_html += "</div>"
     return quiz_html
 
-# =========================
+# ----------------------------
 # Study Adviser
-# =========================
+# ----------------------------
 def study_adviser(question):
-    history, answer = ai_response(f"Give a professional study advice: {question}")
+    if not question.strip():
+        return "Please enter a question for study advice."
+    _, answer = ai_response(f"Give a professional study advice: {question}", [])
     return answer
 
-# =========================
+# ----------------------------
 # Study Planner
-# =========================
+# ----------------------------
 planner_tasks = []
 
 def add_task(task):
-    planner_tasks.append(task)
+    if task.strip():
+        planner_tasks.append(task.strip())
     return "\n".join([f"- {t}" for t in planner_tasks])
 
 def view_plan():
@@ -88,11 +91,11 @@ def view_plan():
         return "No tasks added yet."
     return "\n".join([f"- {t}" for t in planner_tasks])
 
-# =========================
+# ----------------------------
 # Gradio UI
-# =========================
+# ----------------------------
 custom_css = """
-body {background:#121212; color:#EEE; font-family:Poppins, sans-serif;}
+body {background:#121212; color:#EEE; font-family:Poppins,sans-serif;}
 button.primary {background:#7C3AED; color:white; border-radius:8px; padding:5px 12px;}
 button.primary:hover {background:#6D28D9;}
 .gradio-container {background:#121212;}
@@ -132,23 +135,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Dark()) as iface:
             user_input = gr.Textbox(label="Type your question:")
             send_btn = gr.Button("Send", elem_classes="primary")
 
-    # =========================
-    # Event Bindings
-    # =========================
-    # Study Tutor
-    tutor_btn.click(ai_response, inputs=[tutor_input, gr.State(value=[])], outputs=[gr.State(value=[]), tutor_output])
-
-    # MCQ Generator
+    # Bind buttons
+    tutor_btn.click(lambda msg: ai_response(msg, [])[1], inputs=[tutor_input], outputs=[tutor_output])
     mcq_btn.click(generate_mcq, inputs=[mcq_subject], outputs=[mcq_output])
-
-    # Study Adviser
     adviser_btn.click(study_adviser, inputs=[adviser_input], outputs=[adviser_output])
-
-    # Study Planner
     add_task_btn.click(add_task, inputs=[task_input], outputs=[planner_output])
-    view_plan_btn.click(view_plan, outputs=[planner_output])
-
-    # Chatbot
-    send_btn.click(ai_response, inputs=[user_input, gr.State(value=[])], outputs=[gr.State(value=[]), chatbot_display])
+    view_plan_btn.click(lambda: view_plan(), outputs=[planner_output])
+    send_btn.click(lambda msg: ai_response(msg, []) , inputs=[user_input], outputs=[chatbot_display])
 
 iface.launch()