Update app.py

app.py

@@ -1,70 +1,65 @@
 import gradio as gr
-import requests
+import openai
 import os
 import random
+import re
 
-# 🧠 Function to call a Hugging Face model (or any LLM endpoint)
-HF_TOKEN = os.getenv("HF_TOKEN")
+# Set your OpenAI API key
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_mcqs(topic="science", num_questions=3):
-    """Generate MCQs using a model API"""
-    prompt = f"Create {num_questions} multiple-choice questions on {topic}. " \
-             f"Each question should have 4 options (A–D) and specify the correct answer."
+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: ...
+    """
 
-    try:
-        response = requests.post(
-            "https://api-inference.huggingface.co/models/google/gemma-2b",
-            headers={"Authorization": f"Bearer {HF_TOKEN}"},
-            json={"inputs": prompt}
-        )
-        data = response.json()
-        text = data[0]["generated_text"]
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=500,
+        temperature=0.7
+    )
 
-        # 🪄 Simple parsing: try to split questions
-        questions = []
-        for block in text.split("\n\n"):
-            if "." in block:
-                parts = block.split("\n")
-                q_line = parts[0]
-                opts = [p.strip() for p in parts[1:5] if p.strip()]
-                ans = ""
-                for p in parts:
-                    if "Answer" in p or "Correct" in p:
-                        ans = p.split(":")[-1].strip()
-                if len(opts) == 4:
-                    questions.append({"question": q_line, "options": opts, "answer": ans})
-        return questions if questions else [{"question": "⚠️ Couldn’t parse any question.", "options": [], "answer": ""}]
-    except Exception as e:
-        return [{"question": f"Error: {e}", "options": [], "answer": ""}]
+    text = response.choices[0].text.strip()
+    
+    # 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)
 
-
-# 🎯 Build quiz HTML with checkboxes
-def build_quiz_html(topic):
-    mcqs = generate_mcqs(topic)
     html = "<div style='font-family:Poppins,sans-serif;color:#222;'>"
-    for i, q in enumerate(mcqs, 1):
+    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['question']}</b><br><br>"
-        for opt in q["options"]:
+        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:#7e22ce;display:none;' id='ans{i}'><b>✅ Correct Answer:</b> {q['answer']}</p>"
+        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>"""
+    </div>
+    """
     return html
 
-
-# ⚙️ Gradio Interface
+# Gradio UI
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet")) as demo:
-    gr.Markdown("<h1 style='text-align:center;color:#5b21b6;'>🧠 EduAI — Auto MCQ Quiz Generator</h1>")
-    topic = gr.Textbox(label="Enter a topic (e.g. Physics, History, Chemistry)")
+    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>")
 
-    gen_btn.click(build_quiz_html, inputs=topic, outputs=quiz_output)
+    gen_btn.click(generate_mcqs, inputs=[topic_input, num_questions_input], outputs=quiz_output)
 
 demo.launch()