import streamlit as st
import nltk
from transformers import pipeline

# Initialize NLTK
nltk.download('punkt') 
# Instantiate the model
model = pipeline(model="dantedgp/question-generator", tokenizer="google-t5/t5-small")

placeholder = """The mitochondria is the powerhouse of the cell.
Glycolisis harnesses the energy in the bonds of glucose.
The nucleus stores the genetic code of a living being.
The peroxisome produces hydrogen peroxide used for lipid breakdown.
Ribosomes are the sites of protein synthesis.
Antibodies label and target pathogens for destruction.
Hemoglobin is a transport protein for oxygen."""

st.set_page_config(page_title='AI Quiz Generator')
st.title('AI Quiz Generator')

text_input = st.text_area('Enter your text', value=placeholder, height=200)
generate_button = st.button("Generate")

if generate_button:
    sentences = nltk.sent_tokenize(text_input)
    print(sentences)
    for sentence in sentences:
        output = model(f"ask: {sentence}")[0]['generated_text']
        st.write(output)