Update app.py

app.py

@@ -72,21 +72,32 @@ with st.form("user_inputs"):
 
 def download(df):
     col1, col2 = st.columns(2)
-    html_content=df.to_html(index=False)
-    pdf_path= "generated_mcqs.pdf"
+    html_content = df.to_html(index=False)
+    pdf_path = "generated_mcqs.pdf"
+
+    if button:
+        # Download CSV
+        col1.download_button(
+            label="Download CSV",
+            data=df.to_csv(index=False).encode("utf-8"),
+            file_name="generated_mcqs.csv",
+            key="csv-download",
+            help="Click to download as CSV",
+        )
+
+        # Download PDF
+        pdf_data = HTML(string=html_content).write_pdf(pdf_path)
+
+        if pdf_data is not None:
+            col2.download_button(
+                label="Download PDF",
+                data=pdf_data,
+                key="pdf-download",
+                help="Click to download as PDF",
+            )
+        else:
+            st.warning("PDF generation failed. Please try again.")
 
-    if button: 
-        col1.download_button(label="Download CSV",
-                               data=df.to_csv(index=False).encode("utf-8"),
-                               file_name="generated_mcqs.csv",
-                               key="csv-download",
-                               help="Click to download as CSV"
-                            )    
-        
-        col2.download_button(label="Download PDF",
-                    data= HTML(string=html_content).write_pdf(pdf_path),
-                    key="pdf-download",
-                    help="Click to download as PDF")