Update app.py

app.py

@@ -45,7 +45,7 @@ def export_to_excel(study_plan, filename="study_plan.xlsx"):
 
 # Streamlit UI
 st.title("Personalized Study Assistant Chatbot")
-st.write("Generate a tailored study plan with actionable insights and export options.")
+st.write("Generate a tailored study plan with actionable insights and download options.")
 
 # Input fields
 subject = st.text_input("Study Topic or Exam Subject", "IELTS")
@@ -58,19 +58,13 @@ if st.button("Generate Study Plan"):
     st.subheader("Generated Study Plan")
     st.text_area("Your Study Plan", study_plan, height=300)
 
-    # Export options
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        if st.button("Export to PDF"):
-            pdf_file = export_to_pdf(study_plan)
-            st.success(f"Study plan exported as {pdf_file}")
-            with open(pdf_file, "rb") as file:
-                st.download_button("Download PDF", file, file_name="study_plan.pdf", key="pdf_download")
-    
-    with col2:
-        if st.button("Export to Excel"):
-            excel_file = export_to_excel(study_plan)
-            st.success(f"Study plan exported as {excel_file}")
-            with open(excel_file, "rb") as file:
-                st.download_button("Download Excel", file, file_name="study_plan.xlsx", key="excel_download")
+    # File generation after study plan is created
+    pdf_file = export_to_pdf(study_plan)
+    excel_file = export_to_excel(study_plan)
+
+    # Create download buttons for both PDF and Excel
+    with open(pdf_file, "rb") as file:
+        st.download_button("Download PDF", file, file_name="study_plan.pdf", key="pdf_download")
+
+    with open(excel_file, "rb") as file:
+        st.download_button("Download Excel", file, file_name="study_plan.xlsx", key="excel_download")