Update app.py

app.py

@@ -40,33 +40,35 @@ from helper import conversational_chat,pdf_loader,splitDoc,makeEmbeddings,create
 
 def ui(): 
     st.title('PDF Question Answer Bot')
-    # hugging_face_key = os.environ["HUGGINGFACE_HUB_TOKEN"]
+    hugging_face_key = os.environ["HUGGINGFACE_HUB_TOKEN"]
     llm = create_flan_t5_base(load_in_8bit=False)
     hf_llm = HuggingFacePipeline(pipeline=llm)
 
     uploaded_file = st.file_uploader("Choose a PDF file", type=["pdf"]) 
     #saving the uploaded pdf file 
-    save_path = "./uploaded_file.pdf"
-    with open(save_path, "wb") as f:
-        f.write(uploaded_file.read())
+    if uploaded_file is not None:
+        save_path = "./uploaded_file.pdf"
+        with open(save_path, "wb") as f:
+            f.write(uploaded_file.read())
     
     #loading the pdf file 
-    pdf_doc=pdf_loader('./uploaded_file.pdf')
-    vector_database = makeEmbeddings(pdf_doc)
-    #making the retriever of the vector database 
-    retriever = vector_database.as_retriever(search_kwargs={"k":4})
-    qa_chain = RetrievalQA.from_chain_type(llm=hf_llm, chain_type="stuff",retriever=retriever)
-
-    # Create an empty container to hold the PDF loader section
-    pdf_loader_container = st.empty()
-    
-    # Check if the PDF file is uploaded or not
-    if uploaded_file is not None:
-        print("The file has been uploaded successfully")
-        # Hide the PDF loader interface when the file is uploaded
-        pdf_loader_container.empty()
-        # Show the chat interface
-        show_chat_interface(qa_chain)
+        pdf_doc=pdf_loader('./uploaded_file.pdf')
+        vector_database = makeEmbeddings(pdf_doc)
+        #making the retriever of the vector database 
+        retriever = vector_database.as_retriever(search_kwargs={"k":4})
+        qa_chain = RetrievalQA.from_chain_type(llm=hf_llm, chain_type="stuff",retriever=retriever)
+
+        # Create an empty container to hold the PDF loader section
+        pdf_loader_container = st.empty()
+        
+        # Check if the PDF file is uploaded or not
+        if uploaded_file is not None:
+            print("The file has been uploaded successfully")
+            # Hide the PDF loader interface when the file is uploaded
+            pdf_loader_container.empty()
+            # Show the chat interface
+            show_chat_interface(qa_chain)
+
 
 def show_chat_interface(qa_chain):
     if 'history' not in st.session_state: