harshith1411's picture
Create app.py
fdd04ac verified
raw
history blame
3.35 kB
import streamlit as st
from langchain_groq import ChatGroq
import json
# Application Title and Subheader
st.title('AI Quiz Generator')
st.subheader('Test Yourself!')
# Input Fields for Standard and Topic
A = st.text_input('Enter standard')
B = st.text_input('Enter a topic')
# Initialize the ChatGroq Model (replace with your actual API key)
model = ChatGroq(
temperature=0.6,
groq_api_key='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'
)
def generate_quiz(standard, topic, difficulty):
"""
Generate a quiz with 5 multiple choice questions based on the difficulty level.
Returns structured JSON output.
"""
prompt = (
f"Generate a quiz with exactly 5 multiple choice questions for a {standard} student on the topic of {topic}. "
"Each question should have four options and one correct answer. "
"Return the output in JSON format with the following structure: "
"{ \"questions\": [ {\"question\": \"...\", \"options\": [\"...\", \"...\", \"...\", \"...\"], \"answer\": \"...\"} ] }"
)
response = model.predict(prompt)
try:
return json.loads(response) # Safely parse JSON response
except json.JSONDecodeError:
st.error("Failed to parse quiz. Please try again.")
return None
def calculate_score(user_answers, correct_answers):
"""Compare user answers with the correct answers and calculate the score."""
return sum(1 for q, ans in user_answers.items() if correct_answers.get(q) == ans)
# Session state to store quiz data
if 'quiz_data' not in st.session_state:
st.session_state.quiz_data = None
if 'user_answers' not in st.session_state:
st.session_state.user_answers = {}
# Difficulty Selection
difficulty_levels = ["Low", "Medium", "Hard"]
difficulty = st.radio("Select Difficulty Level", difficulty_levels)
# Generate Quiz Button
if st.button("Generate Quiz"):
if A and B:
st.session_state.quiz_data = generate_quiz(A, B, difficulty)
if st.session_state.quiz_data:
st.session_state.user_answers = {} # Reset user answers
st.success("Quiz Generated!")
else:
st.error("Please enter both the standard and topic.")
# Display Quiz if Available
if st.session_state.quiz_data:
st.write("### Quiz Questions:")
questions = st.session_state.quiz_data.get("questions", [])
for i, q in enumerate(questions[:5]): # Limit to 5 questions
st.write(f"**Q{i+1}: {q['question']}**")
user_answer = st.radio(
f"Select your answer for Question {i+1}",
options=q["options"],
key=f"question_{i+1}"
)
st.session_state.user_answers[f"question_{i+1}"] = user_answer
# Submit Quiz Button
if st.button("Submit Quiz"):
if st.session_state.quiz_data:
correct_answers = {f"question_{i+1}": q["answer"] for i, q in enumerate(st.session_state.quiz_data["questions"][:5])}
score = calculate_score(st.session_state.user_answers, correct_answers)
st.write("### Quiz Results")
st.write(f"Your answers: {st.session_state.user_answers}")
st.write(f"Correct answers: {correct_answers}")
st.write(f"Your score: {score}/5")
st.success("Great job! Keep practicing!")
else:
st.error("Quiz data not available. Please regenerate the quiz.")