Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 from transformers import pipeline
 from PyPDF2 import PdfReader
-import PyPDF2
-import os
 import nltk
 
 nltk.download('punkt')
+st.title(body='7 - Question Generation')
+
 
 def get_pdf_text(pdf_docs):
     text = ""
@@ -15,35 +15,38 @@ def get_pdf_text(pdf_docs):
             text += page.extract_text()
     return text
 
-def main():
+
+########################################################
+st.subheader(body='Proposition 1',divider='orange')
+
+if st.toggle(label='Show Proposition 1'):
     st.title('Question Generator from PDFs')
+    if st.checkbox('Show Caption'):
+        st.caption('Hugging Face Model used: ramsrigouthamg/t5_squad_v1')
     pipe = pipeline(
         task = 'text2text-generation',
         model = 'ramsrigouthamg/t5_squad_v1'
     )
     file = st.file_uploader(label='Upload',accept_multiple_files=True)
-    pr = st.button(label='Start')
-    if pr:
-        st.write('Hi')
-        raw_text = get_pdf_text(file)
-        sentences = nltk.sent_tokenize(text=raw_text)
-        # st.write(sts)
-        # for i in sentences:
-        #     st.write(i)
-        questions = []
-        st.subheader("Generated Questions are: ")
-        s = pipe(sentences)
-        for i in s:
-            questions.append(i['generated_text'][10:])
-            st.write(i['generated_text'][10:])
-        if st.toggle(label='Show Pipeline Output'):
-            st.write(s)
-        if st.toggle(label='Show Questions list'):
-            st.write(questions)
-        # for i in sts:
-        #     x = pipe(i)
-        #     questions.append(x)
-        #     st.write(x)
+    # pr = st.button(label='Process')
+    raw_text = get_pdf_text(file)
+    sentences = nltk.sent_tokenize(text=raw_text)
+    s = pipe(sentences)
+    questions = []
+    for i in s:
+        x = i['generated_text'][10:]
+        questions.append(x)
+        # st.write(f':blue[{x}]')
 
-if __name__ == '__main__':
-    main()
+    if st.toggle(label='Show Questions'):
+        st.subheader("*Generated Questions are:*")
+        for i in s:
+            x = i['generated_text'][10:]
+            questions.append(x)
+            st.write(f':blue[{x}]')
+    if st.toggle('Show Text'):
+        st.write(raw_text)
+    if st.toggle(label='Show Pipeline Output'):
+        st.write(s)
+    if st.toggle(label='Show Questions list'):
+        st.write(questions)