Update app.py

app.py

@@ -24,6 +24,7 @@ class AdvancedPdfChatbot:
         self.llm =  ChatOpenAI(temperature=0.5,model_name='gpt-4o',max_tokens=3000)
         self.memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
         self.qa_chain = None
+        self.pdf_path = None
         self.template = """
 You are a file-based knowledge assistant that interacts with users like ChatGPT. Your primary source of knowledge comes from user-uploaded files, such as PDFs. You do not rely on general knowledge or the internet. Instead, you extract, analyze, and synthesize information directly from the content of the provided file(s).  
 **1. Personality and Tone**  
@@ -81,6 +82,7 @@ NOTE : DESCRIBE/SUMMARY should always return the overall summary of the document
             texts = self.text_splitter.split_documents(documents)
             self.db = FAISS.from_documents(texts, self.embeddings)
             self.setup_conversation_chain()
+            self.pdf_path = pdf_path
         except Exception as e:
             return f"An error occurred while processing the PDF: {e}"
 
@@ -98,6 +100,13 @@ NOTE : DESCRIBE/SUMMARY should always return the overall summary of the document
         result = self.qa_chain({"question": query})
         return result['answer']
 
+    def get_pdf_path(self):
+    # Return the stored PDF path
+    if self.pdf_path:
+        return self.pdf_path
+    else:
+        return "No PDF uploaded yet."
+
 # Initialize the chatbot
 pdf_chatbot = AdvancedPdfChatbot(openai_api_key)
 
@@ -123,6 +132,10 @@ def clear_chatbot():
     pdf_chatbot.memory.clear()
     return []
 
+def get_pdf_path():
+    # Call the method to return the current PDF path
+    return pdf_chatbot.get_pdf_path()
+
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# PDF Chatbot")
@@ -134,6 +147,9 @@ with gr.Blocks() as demo:
     upload_status = gr.Textbox(label="Upload Status")
     upload_button.click(upload_pdf, inputs=[pdf_upload], outputs=[upload_status])
 
+    path_button = gr.Button("Get PDF Path")
+    pdf_path_display = gr.Textbox(label="Current PDF Path")
+    
     chatbot_interface = gr.Chatbot()
     msg = gr.Textbox()
     clear = gr.Button("Clear")