import streamlit as st
from transformers import T5Tokenizer, T5ForConditionalGeneration

# Load Model
model_name = "google/flan-t5-small" #"YOUR_USERNAME/quiz-generator"
tokenizer = T5Tokenizer.from_pretrained(model_name)
model = T5ForConditionalGeneration.from_pretrained(model_name)

# Streamlit App
st.title("AI Quiz Generator")
topic = st.text_input("Enter Topic:")
difficulty = st.selectbox("Select Difficulty:", ["Easy", "Medium", "Hard"])
num_questions = st.slider("Number of Questions:", 1, 10, 5)

if st.button("Generate Quiz"):
    prompt = f"Generate {num_questions} {difficulty} questions on {topic}"
    inputs = tokenizer(prompt, return_tensors="pt", padding=True, truncation=True)
    outputs = model.generate(**inputs)
    questions = tokenizer.decode(outputs[0], skip_special_tokens=True)
    st.write("### Generated Questions:")
    st.write(questions)