Update app.py

app.py

@@ -83,7 +83,8 @@ def generate_question(context, answer, num_beams=5):
 # Function to export questions to CSV
 def export_to_csv(data):
     df = pd.DataFrame(data, columns=["Context", "Answer", "Question"])
-    df.to_csv('questions.csv', index=False)
+    csv = df.to_csv('questions.csv', index=False)
+    return csv
 
 # Function to export questions to PDF
 def export_to_pdf(data):
@@ -97,7 +98,8 @@ def export_to_pdf(data):
         pdf.multi_cell(0, 10, f"Question: {question}")
         pdf.ln(10)
     
-    pdf.output("questions.pdf")
+    # pdf.output("questions.pdf")
+    return pdf.output(name='questions.pdf',dest='S').encode('latin1')
 
 # Streamlit interface
 st.title(":blue[Question Generator from Text]")
@@ -108,8 +110,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)
+downlaod_csv = st.toggle('Download CSV',value=True)
+download_pdf = st.toggle('Download PDF',value=True)
 if st.button("Generate Questions"):
     if text:
         keywords = extract_keywords(text)
@@ -133,12 +135,14 @@ if st.button("Generate Questions"):
         # Export buttons
         # if st.button("Export to CSV"):
         if downlaod_csv:
-            export_to_csv(data)
+            csv_data = export_to_csv(data)
             st.success("Questions exported to questions.csv")
+            st.download_button(label="Download CSV", data=csv_data, file_name='questions.csv', mime='text/csv')
 
         # if st.button("Export to PDF"):
         if download_pdf:
-            export_to_pdf(data)
+            pdf_data = export_to_pdf(data)
             st.success("Questions exported to questions.pdf")
+            st.download_button(label="Download PDF", data=pdf_data, file_name='questions.pdf', mime='application/pdf')
     else:
         st.write("Please enter some text to generate questions.")