Update app.py

app.py

@@ -11,6 +11,7 @@ from langchain_openai import ChatOpenAI
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain.chains import SequentialChain
+import pdfkit as pdf
 
 with open('Response.json', 'r') as file:
     RESPONSE_JSON = json.load(file)
@@ -71,6 +72,7 @@ with st.form("user_inputs"):
 
 def download(df):
     col1, col2 = st.columns(2)
+    df.to_html('mcq.html').encode("utf-8")
 
     if button: 
         col1.download_button(label="Download CSV",
@@ -78,9 +80,11 @@ def download(df):
                                file_name="generated_mcqs.csv",
                                key="csv-download",
                                help="Click to download as CSV"
-                            )                   
+                            )    
+        
+        
         col2.download_button(label="Download PDF",
-                               data=df.to_html().encode("utf-8"),
+                               data=pdf.from_file('mcq.html'),
                                file_name="generated_mcqs.pdf",
                                key="pdf-download",
                                help="Click to download as PDF"