Update app.py

app.py

@@ -48,6 +48,11 @@ else:
     d = embedding_model.get_sentence_embedding_dimension()  # Dimension of the embeddings
     faiss_index = faiss.IndexFlatL2(d)  # Using IndexFlatL2 for simplicity
 
+state = {
+    "conversation": [],
+    "sentences": []
+}
+
 def extract_text_from_pdf(pdf_path):
     text = ""
     try:
@@ -73,9 +78,8 @@ def preprocess_text(text):
     return sentences
 
 def upload_files(files):
+    global state, faiss_index
     try:
-        global faiss_index
-        
         for file in files:
             try:
                 file_path = file.name
@@ -90,6 +94,7 @@ def upload_files(files):
                 embeddings = embedding_model.encode(sentences)
                 
                 faiss_index.add(np.array(embeddings).astype(np.float32))  # Add embeddings
+                state["sentences"].extend(sentences)
 
             except Exception as e:
                 print(f"Error processing file '{file.name}': {e}")
@@ -104,7 +109,8 @@ def upload_files(files):
         print(f"General error processing files: {e}")
         return {"error": str(e)}
 
-def process_and_query(state, files, question):
+def process_and_query(files, question):
+    global state
     if files:
         upload_result = upload_files(files)
         if "error" in upload_result:
@@ -139,13 +145,12 @@ with gr.Blocks() as demo:
         upload = gr.File(file_count="multiple", label="Upload PDF or DOCX files")
         upload_button = gr.Button("Upload")
         upload_output = gr.Textbox()
-        upload_button.click(fn=upload_files, inputs=upload, outputs=upload_output)
+        upload_button.click(fn=upload_files, inputs=[upload], outputs=upload_output)
 
     with gr.Tab("Query"):
         query = gr.Textbox(label="Enter your query")
         query_button = gr.Button("Search")
         query_output = gr.Textbox()
-        query_button.click(fn=process_and_query, inputs=[query], outputs=query_output)
+        query_button.click(fn=process_and_query, inputs=[upload, query], outputs=query_output)
 
 demo.launch()
-