Update app.py

app.py

@@ -1,65 +1,118 @@
 import gradio as gr
 import openai
-import os
 import random
-import re
+import os
 
 # Set your OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_mcqs(topic, num_questions=3):
-    """
-    Generate MCQs dynamically using OpenAI's GPT.
-    Returns HTML with checkboxes and Show Answers button.
-    """
-    prompt = f"""
-    Generate {num_questions} multiple-choice questions on "{topic}".
-    Format the output as follows for each question:
-    Question: ...
-    Options: A) ... B) ... C) ... D) ...
-    Answer: ...
-    """
+# -------------------- AI Functions --------------------
 
+def chat_tutor(message, history=[]):
+    """AI Study Tutor Chat"""
+    prompt = f"You are EduAI, a smart student-friendly tutor. Answer clearly and concisely:\n\nUser: {message}\nAI:"
     response = openai.Completion.create(
         engine="text-davinci-003",
         prompt=prompt,
-        max_tokens=500,
+        max_tokens=200,
         temperature=0.7
     )
+    reply = response.choices[0].text.strip()
+    history.append((message, reply))
+    return history, history
 
-    text = response.choices[0].text.strip()
+def generate_mcqs(topic, num_questions=3):
+    """AI generates MCQs dynamically"""
+    prompt = f"Create {num_questions} multiple choice questions on '{topic}' with 4 options (A,B,C,D) and correct answer. Format each as Question: ... Options: ... Answer: ..."
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=500,
+        temperature=0.7
+    )
+    text = response.choices[0].text.strip().split("\n")
     
-    # Parse questions
-    pattern = r"Question:\s*(.*?)\nOptions:\s*A\)\s*(.*?)\s*B\)\s*(.*?)\s*C\)\s*(.*?)\s*D\)\s*(.*?)\nAnswer:\s*(.*)"
-    matches = re.findall(pattern, text, re.DOTALL)
-
-    html = "<div style='font-family:Poppins,sans-serif;color:#222;'>"
-    for i, m in enumerate(matches, 1):
-        q_text, opt1, opt2, opt3, opt4, ans = m
-        html += f"<div style='margin:15px;padding:15px;border-radius:10px;background:#fff9ff;box-shadow:0 3px 10px rgba(0,0,0,0.08);'>"
-        html += f"<b style='color:#5b21b6;'>{i}. {q_text}</b><br><br>"
-        for opt in [opt1, opt2, opt3, opt4]:
-            html += f"<label style='display:block;margin-bottom:5px;'><input type='checkbox' name='q{i}' value='{opt}' style='accent-color:#7c3aed;'> {opt}</label>"
-        html += f"<p style='color:#065f46;display:none;' id='ans{i}'><b>✅ Correct Answer:</b> {ans.strip()}</p>"
-        html += "</div>"
-
-    html += """
-    <button onclick="document.querySelectorAll('[id^=ans]').forEach(a=>a.style.display='block');" 
-            style='margin-top:10px;background:#7c3aed;color:white;border:none;padding:10px 20px;border-radius:8px;cursor:pointer;'>
-        Show Answers
-    </button>
-    </div>
-    """
+    html = "<div style='font-family:Poppins,sans-serif;color:#eee;'>"
+    for i, line in enumerate(text):
+        if line.lower().startswith("question:"):
+            html += f"<div style='margin:15px;padding:15px;border-radius:10px;background:#222;box-shadow:0 3px 10px rgba(0,0,0,0.3);'>"
+            html += f"<b style='color:#7c3aed;'>{i+1}. {line[9:].strip()}</b><br>"
+        elif line.lower().startswith("options:"):
+            options = line[8:].split(",")
+            for opt in options:
+                html += f"<label style='display:block;margin:5px 0;'><input type='checkbox' style='accent-color:#7c3aed;'> {opt.strip()}</label>"
+        elif line.lower().startswith("answer:"):
+            html += f"<p style='color:#22c55e;display:none;' id='ans{i}'><b>✅ Correct Answer:</b> {line[7:].strip()}</p>"
+            html += "</div>"
+    html += "<button onclick=\"document.querySelectorAll('[id^=ans]').forEach(a=>a.style.display='block');\" style='margin:10px;padding:10px 20px;border:none;border-radius:8px;background:#7c3aed;color:white;cursor:pointer;'>Show Answers</button></div>"
     return html
 
-# Gradio UI
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as demo:
-    gr.Markdown("<h1 style='text-align:center;color:#5b21b6;'>🧠 EduAI — AI MCQ Quiz Generator</h1>")
-    topic_input = gr.Textbox(label="Enter a topic (e.g., Physics, History)")
-    num_questions_input = gr.Slider(1, 10, value=3, step=1, label="Number of Questions")
-    gen_btn = gr.Button("🎯 Generate Quiz")
-    quiz_output = gr.HTML("<i>Enter a topic and click Generate to start!</i>")
+def study_adviser(topic):
+    """AI gives study advice"""
+    prompt = f"You are EduAI. Give 5 practical, concise study tips for the topic: {topic}"
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=150,
+        temperature=0.7
+    )
+    return response.choices[0].text.strip().replace("\n", "<br>")
+
+def study_planner(task):
+    """AI helps plan study schedule"""
+    prompt = f"You are EduAI. Create a 1-day study plan for: {task}. Suggest timing and order of activities."
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=150,
+        temperature=0.7
+    )
+    return response.choices[0].text.strip().replace("\n", "<br>")
+
+# -------------------- Gradio Interface --------------------
+
+with gr.Blocks(css="""
+    .gradio-container {background-color:#121212;color:#eee;font-family:Poppins,sans-serif;}
+    button {font-weight:600;border-radius:8px;}
+    button.primary {background:#7c3aed;color:white;border:none;}
+    button.primary:hover {background:#9333ea;}
+    h1 {color:#7c3aed;text-align:center;}
+""") as demo:
+
+    gr.Markdown("<h1>🎓 EduAI — Smart Study Assistant</h1>")
+
+    with gr.Row():
+        with gr.Column(scale=1, min_width=250):
+            gr.Markdown("### 🧭 Main Menu")
+
+            with gr.Accordion("📚 Study Tutor", open=True):
+                chat_history = gr.Chatbot(label="EduAI Tutor", height=350)
+                user_msg = gr.Textbox(label="Ask EduAI", placeholder="Type your question...")
+                send_btn = gr.Button("Send", elem_classes="primary")
+
+            with gr.Accordion("🎯 MCQ Generator", open=False):
+                topic_mcq = gr.Textbox(label="Topic for MCQs")
+                num_q = gr.Slider(1, 10, value=3, step=1, label="Number of Questions")
+                mcq_btn = gr.Button("Generate MCQs", elem_classes="primary")
+                mcq_output = gr.HTML("<i>MCQs will appear here...</i>")
+
+            with gr.Accordion("💡 Study Adviser", open=False):
+                topic_advice = gr.Textbox(label="Topic for advice")
+                advice_btn = gr.Button("Get Advice", elem_classes="primary")
+                advice_output = gr.HTML("<i>Advice will appear here...</i>")
+
+            with gr.Accordion("🗓 Study Planner", open=False):
+                task_plan = gr.Textbox(label="Task / Subject")
+                plan_btn = gr.Button("Generate Plan", elem_classes="primary")
+                plan_output = gr.HTML("<i>Plan will appear here...</i>")
+
+        with gr.Column(scale=3):
+            gr.Markdown("<h2 style='color:#7c3aed;'>📘 Your Study Workspace</h2>")
 
-    gen_btn.click(generate_mcqs, inputs=[topic_input, num_questions_input], outputs=quiz_output)
+    # -------- Events --------
+    send_btn.click(chat_tutor, inputs=[user_msg, chat_history], outputs=[chat_history, chat_history])
+    mcq_btn.click(generate_mcqs, inputs=[topic_mcq, num_q], outputs=mcq_output)
+    advice_btn.click(study_adviser, inputs=topic_advice, outputs=advice_output)
+    plan_btn.click(study_planner, inputs=task_plan, outputs=plan_output)
 
 demo.launch()