Update app.py

app.py

@@ -63,25 +63,25 @@ def query_document(query, faiss_index, document_chunks, model_name="sentence-tra
 # Gradio interface
 def chatbot_interface():
     with gr.Blocks() as demo:
-        faiss_index = None
-        document_chunks = None
+        state = gr.State()  # This state will store the FAISS index and document chunks
 
         # Function to handle document upload
         def upload_file(file):
-            nonlocal faiss_index, document_chunks
             faiss_index, document_chunks = process_document(file.name)
+            state.value = (faiss_index, document_chunks)
             return "Document uploaded and indexed. You can now ask questions."
 
         # Function to handle user queries
         def ask_question(query):
-            if faiss_index and document_chunks:
+            if state.value is not None:
+                faiss_index, document_chunks = state.value
                 return query_document(query, faiss_index, document_chunks)
             return "Please upload a document first."
 
         # Gradio UI
         upload = gr.File(label="Upload a PDF document")
         question = gr.Textbox(label="Ask a question about the document")
-        answer = gr.Textbox(label="Answer", interactive=False)  # Updated to interactive=False
+        answer = gr.Textbox(label="Answer", interactive=False)
 
         # Layout for Gradio app
         gr.Markdown("# Document Chatbot")