Update utils.py

utils.py

@@ -2,13 +2,8 @@ from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 import json
 from prompts import *
-
-# from langchain.text_splitter import RecursiveCharacterTextSplitter
-# from PyPDF2 import PdfReader
-# from langchain_google_genai import GoogleGenerativeAIEmbeddings
-# from langchain.vectorstores import FAISS
-# import PyPDF2
-
+import streamlit as st
+import os
 
 # utils function to generate QA Pairs
 def util(context, numPairs, inputPrompt,model):
@@ -46,3 +41,20 @@ def getMcqQAPairs(context, numPairs,model):
         template=prompt_template, input_variables=["context", "numPairs"]
     )
     return util(context, numPairs, prompt,model)
+
+#download file
+def downloadFile(response,FileName):
+    with open(FileName, "w") as outfile:
+        json.dump(response, outfile, indent=4)
+    with open(FileName, "rb") as file:
+        st.download_button(
+            label="Download File",
+            data=file,
+            file_name=FileName,
+            mime="text/json",
+            type='primary'
+        )
+    os.remove(FileName)
+    os.remove('temp.pdf')
+
+