Update app.py

app.py

@@ -8,21 +8,19 @@ load_dotenv()
 api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
+# ✨ Global variable to store last topic
+last_user_question = ""
+
 # 🐍 Python Tutor Function
 def python_tutor(user_input):
+    global last_user_question
+    last_user_question = user_input  # store latest question for quiz
     response = client.chat.completions.create(
         model="gpt-4.1-mini",
         messages=[
             {
                 "role": "system",
-                "content": """You are a Python tutor. Help users understand Python with short examples. Be kind, clear, and only answer Python-related questions.
-
-1. Confirm if the question is Python-related. If not, politely inform the user and refrain from answering.
-2. Provide a concise initial answer:
-   - Include a brief explanation.
-   - Provide a straightforward code example.
-3. Keep the quiz simple and give options to choose. If wrong answers are selected, correct them with an explanation.
-"""
+                "content": """You are a Python tutor. Help users understand Python with short examples. Be kind, clear, and only answer Python-related questions."""
             },
             {"role": "user", "content": user_input}
         ],
@@ -31,16 +29,30 @@ def python_tutor(user_input):
     )
     return response.choices[0].message.content
 
-# 📚 Simple Quiz Generator (fixed question for now)
-def ask_question():
-    return "What is the output of print(2 + 3 * 4)?"
+# 📚 Dynamic Quiz Generator Based on User Topic
+def generate_quiz():
+    global last_user_question
+    if not last_user_question:
+        return "Please ask a Python question first in the tutor tab."
+
+    quiz_prompt = f"""Generate one simple multiple-choice quiz question based on this Python topic/question: "{last_user_question}". 
+Format: 
+Question: ...
+Options: A. ... B. ... C. ... D. ...
+Answer: ..."""
+
+    response = client.chat.completions.create(
+        model="gpt-4.1-mini",
+        messages=[
+            {"role": "system", "content": "You are a quiz generator for Python topics."},
+            {"role": "user", "content": quiz_prompt}
+        ],
+        temperature=0.3,
+        max_tokens=300
+    )
 
-def evaluate_answer(user_answer):
-    correct_answer = "14"
-    if user_answer.strip() == correct_answer:
-        return "✅ Correct! Great job!"
-    else:
-        return "❌ Incorrect. Try again. Hint: Use BODMAS."
+    quiz_text = response.choices[0].message.content.strip()
+    return quiz_text
 
 # 🎨 Gradio UI with Two Pages
 with gr.Blocks() as app:
@@ -51,10 +63,9 @@ with gr.Blocks() as app:
         gr.Button("Ask").click(python_tutor, inputs=question_input, outputs=answer_output)
 
     with gr.Tab("🧪 Mini Quiz"):
-        gr.Markdown("### Let's test your Python basics!")
-        quiz_question = gr.Textbox(label="Question", interactive=False, value=ask_question())
-        user_response = gr.Textbox(label="Your Answer")
-        result_output = gr.Textbox(label="Result")
-        gr.Button("Submit Answer").click(evaluate_answer, inputs=user_response, outputs=result_output)
+        gr.Markdown("### Your Custom Python Quiz Based on Tutor Topic")
+        quiz_box = gr.Textbox(label="Quiz Question", lines=4)
+        generate_btn = gr.Button("🔄 Generate Quiz")
+        generate_btn.click(fn=generate_quiz, outputs=quiz_box)
 
 app.launch()