Update app.py

app.py

@@ -16,13 +16,11 @@ from nltk.tokenize import sent_tokenize
 
 # Load spaCy model
 nlp = spacy.load("en_core_web_sm")
-# wiki_wiki = wikipediaapi.Wikipedia('en')
 
 # Initialize Wikipedia API with a user agent
 user_agent = 'QGen/1.0 ([email protected])'
 wiki_wiki = wikipediaapi.Wikipedia(user_agent= user_agent,language='en')
 
-
 # Load T5 model and tokenizer
 model_name = "DevBM/t5-large-squad"
 model = T5ForConditionalGeneration.from_pretrained(model_name)
@@ -102,7 +100,7 @@ def export_to_pdf(data):
     pdf.output("questions.pdf")
 
 # Streamlit interface
-st.title("Question Generator from Text")
+st.title(":blue[Question Generator from Text]")
 text = st.text_area("Enter text here:", value="Joe Biden, the current US president is on a weak wicket going in for his reelection later this November against former President Donald Trump.")
 
 # Customization options
@@ -110,7 +108,8 @@ num_beams = st.slider("Select number of beams for question generation", min_valu
 context_window_size = st.slider("Select context window size (number of sentences before and after)", min_value=1, max_value=5, value=1)
 num_questions = st.slider("Select number of questions to generate", min_value=1, max_value=1000, value=5)
 question_complexity = st.selectbox("Select question complexity", ["Simple", "Intermediate", "Complex"])
-
+downlaod_csv = st.toggle('Download CSV',value=False)
+download_pdf = st.toggle('Download PDF',value=False)
 if st.button("Generate Questions"):
     if text:
         keywords = extract_keywords(text)
@@ -132,11 +131,13 @@ if st.button("Generate Questions"):
             data.append((context, keyword, question))
         
         # Export buttons
-        if st.button("Export to CSV"):
+        # if st.button("Export to CSV"):
+        if downlaod_csv:
             export_to_csv(data)
             st.success("Questions exported to questions.csv")
 
-        if st.button("Export to PDF"):
+        # if st.button("Export to PDF"):
+        if download_pdf:
             export_to_pdf(data)
             st.success("Questions exported to questions.pdf")
     else: