Update app.py

app.py

@@ -505,16 +505,22 @@ import gradio as gr
 def chatbot_interface(user_query, response_style, selected_retrieval_methods, selected_reranking_methods, pdf_file, chunk_size, overlap):
     vector_db_path = "vector_db.pkl"
     annoy_index_path = "vector_index.ann"
-    if pdf_file is not None:
-        with tempfile.NamedTemporaryFile(mode="wb", suffix=".pdf", delete=False) as tmp:
-            tmp.write(pdf_file)  # Write the uploaded file content to the temporary file
-            pdf_path = tmp.name  
-
-        # Create vector DB and Annoy index 
-        create_vector_db_and_annoy_index(pdf_path, vector_db_path, annoy_index_path)
-        store_embeddings_in_vector_db(pdf_path, 'vector_db.pkl', 'vector_index.ann', chunk_size, overlap)
-    else:
-        return "Please upload a PDF file." 
+
+    pdf_path = pdf_file.name
+    create_vector_db_and_annoy_index(pdf_path, 'vector_db.pkl', 'vector_index.ann')
+    store_embeddings_in_vector_db(pdf_path, 'vector_db.pkl', 'vector_index.ann', chunk_size, overlap)
+    
+    
+    # if pdf_file is not None:
+    #     with tempfile.NamedTemporaryFile(mode="wb", suffix=".pdf", delete=False) as tmp:
+    #         tmp.write(pdf_file)  # Write the uploaded file content to the temporary file
+    #         pdf_path = tmp.name  
+
+    #     # Create vector DB and Annoy index 
+    #     create_vector_db_and_annoy_index(pdf_path, vector_db_path, annoy_index_path)
+    #     store_embeddings_in_vector_db(pdf_path, 'vector_db.pkl', 'vector_index.ann', chunk_size, overlap)
+    # else:
+    #     return "Please upload a PDF file." 
     # if pdf_file is not None:
     #     pdf_path = pdf_file.name  # Get the path of the uploaded file
     #     create_vector_db_and_annoy_index(pdf_path, 'vector_db.pkl', 'vector_index.ann')