Update app.py

app.py

@@ -18,17 +18,12 @@ 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)
-        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.db = FAISS.from_documents(texts, self.embeddings)
         self.setup_conversation_chain()
 
     def setup_conversation_chain(self):
@@ -51,9 +46,7 @@ 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):
@@ -61,10 +54,14 @@ def respond(message, history):
     history.append((message, bot_message))
     return "", history
 
+def clear_chatbot():
+    pdf_chatbot.memory.clear()
+    return []
+
 # 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")
@@ -77,7 +74,7 @@ with gr.Blocks() as demo:
     clear = gr.Button("Clear")
 
     msg.submit(respond, inputs=[msg, chatbot_interface], outputs=[msg, chatbot_interface])
-    clear.click(lambda: None, None, chatbot_interface, queue=False)
+    clear.click(clear_chatbot, outputs=[chatbot_interface])
 
 if __name__ == "__main__":
     demo.launch()