Update app.py

app.py

@@ -7,7 +7,7 @@ from langchain.vectorstores import FAISS
 from langchain.chains import ConversationalRetrievalChain
 from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
-import os
+
 openai_api_key = os.environ.get("OPENAI_API_KEY")
 
 class AdvancedPdfChatbot:
@@ -18,12 +18,17 @@ class AdvancedPdfChatbot:
         self.llm = ChatOpenAI(temperature=0, model_name="gpt-3.5-turbo")
         self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
         self.qa_chain = None
+        self.db = None
 
     def load_and_process_pdf(self, pdf_path):
         loader = PyPDFLoader(pdf_path)
         documents = loader.load()
         texts = self.text_splitter.split_documents(documents)
-        self.db = FAISS.from_documents(texts, self.embeddings)
+        if self.db is None:
+            self.db = FAISS.from_documents(texts, self.embeddings)
+        else:
+            new_db = FAISS.from_documents(texts, self.embeddings)
+            self.db.merge_from(new_db)
         self.setup_conversation_chain()
 
     def setup_conversation_chain(self):
@@ -40,14 +45,15 @@ class AdvancedPdfChatbot:
         return result['answer']
 
 # Initialize the chatbot
-
 pdf_chatbot = AdvancedPdfChatbot(openai_api_key)
 
 def upload_pdf(pdf_file):
     if pdf_file is None:
         return "Please upload a PDF file."
     file_path = pdf_file.name
+    pdf_file.save(file_path)  # Ensure the file is saved locally
     pdf_chatbot.load_and_process_pdf(file_path)
+    os.remove(file_path)  # Clean up the uploaded file after processing
     return "PDF uploaded and processed successfully. You can now start chatting!"
 
 def respond(message, history):
@@ -58,7 +64,7 @@ def respond(message, history):
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# PDF Chatbot")
-    
+
     with gr.Row():
         pdf_upload = gr.File(label="Upload PDF", file_types=[".pdf"])
         upload_button = gr.Button("Process PDF")
@@ -74,4 +80,4 @@ with gr.Blocks() as demo:
     clear.click(lambda: None, None, chatbot_interface, queue=False)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()