Update app.py

app.py

@@ -56,15 +56,14 @@ document_texts_path = "document_texts.pkl"
 
 document_texts = []
 
-if os.path.exists(index_path):
+if os.path.exists(index_path) and os.path.exists(document_texts_path):
     try:
         with open(index_path, "rb") as f:
             index = pickle.load(f)
             print("Loaded FAISS index from faiss_index.pkl")
-        if os.path.exists(document_texts_path):
-            with open(document_texts_path, "rb") as f:
-                document_texts = pickle.load(f)
-                print("Loaded document texts from document_texts.pkl")
+        with open(document_texts_path, "rb") as f:
+            document_texts = pickle.load(f)
+            print("Loaded document texts from document_texts.pkl")
     except Exception as e:
         print(f"Error loading FAISS index or document texts: {e}")
 else:
@@ -76,8 +75,8 @@ else:
 
 def upload_files(files):
     global index, document_texts
-    for file in files:
-        try:
+    try:
+        for file in files:
             content = file.read()
             if file.name.endswith('.pdf'):
                 with open("temp.pdf", "wb") as f:
@@ -95,23 +94,19 @@ def upload_files(files):
             embeddings = embedding_model.encode(sentences)
             index.add(np.array(embeddings))
             document_texts.append(text)
-        except Exception as e:
-            print(f"Error processing file {file.name}: {e}")
-            return f"Error processing file {file.name}: {e}"
 
-    # Save the updated index and documents
-    try:
+        # Save the updated index and documents
         with open(index_path, "wb") as f:
             pickle.dump(index, f)
             print("Saved updated FAISS index to faiss_index.pkl")
         with open(document_texts_path, "wb") as f:
             pickle.dump(document_texts, f)
             print("Saved updated document texts to document_texts.pkl")
+        
+        return "Files processed successfully"
     except Exception as e:
-        print(f"Error saving FAISS index or document texts: {e}")
-        return f"Error saving FAISS index or document texts: {e}"
-
-    return "Files processed successfully"
+        print(f"Error processing files: {e}")
+        return f"Error processing files: {e}"
 
 def query_text(text):
     try:
@@ -152,6 +147,7 @@ demo.launch()
 
 
 
+