Multiple sentence support

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import nltk
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
 # Instantiate the model
@@ -16,9 +17,21 @@ def generate_questions(txt):
 
 st.set_page_config(page_title='AI Quiz Generator')
 st.title('AI Quiz Generator')
-txt_input = st.text_area('Enter your text', '', height=200)
+st.subtitle('It generates questions from your notes. How cool is that?')
+text_input = st.text_area(
+    'Enter your text', 
+    value='Algae cells can be used to synthesize lipids for the production of biofuels and foods.', 
+    height=200
+)
+
 generate_button = st.button("Generate Questions")
 
 if generate_button:
-    questions = generate_questions(txt_input)
-    st.write(questions)
+    if text_input != '':
+        sentences = nltk.sent_tokenize(text_input)
+        for sentence in sentences:
+          question = generate_questions(text_input)
+          st.write(question)
+    else:
+      st.warning('Please enter some text.')
+    

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-torch
+torch
+nltk