Update app.py

app.py

@@ -280,13 +280,17 @@ def get_web_search_database():
     temp_dir = tempfile.mkdtemp()
     
     try:
-        # Initialize with an empty list of documents
-        database = WebSearchFAISS.from_documents([], embed)
-        logging.info("Successfully initialized empty WebSearchFAISS database")
+        # Create a dummy document to initialize the database
+        dummy_doc = Document(page_content="Dummy content", metadata={"source": "dummy"})
+        database = FAISS.from_documents([dummy_doc], embed)
+        logging.info("Successfully initialized WebSearchFAISS database with dummy document")
+        # Remove the dummy document
+        database.delete(["dummy"])
+        logging.info("Removed dummy document from database")
     except Exception as e:
         logging.error(f"Error initializing WebSearchFAISS: {str(e)}", exc_info=True)
         # If initialization fails, create an empty database
-        database = WebSearchFAISS(embed, None, {}, {}, None)
+        database = FAISS(embed, None, {}, {}, None)
         logging.info("Created empty WebSearchFAISS database manually")
     
     return database, temp_dir
@@ -306,7 +310,7 @@ def duckduckgo_search(query):
         documents = []
         for result in results:
             content = f"{result['title']}\n{result['body']}"
-            doc = WebSearchDocument(page_content=content, metadata={"source": result['href']})
+            doc = Document(page_content=content, metadata={"source": result['href']})
             documents.append(doc)
         
         logging.info(f"Number of documents created: {len(documents)}")
@@ -317,7 +321,7 @@ def duckduckgo_search(query):
             except Exception as e:
                 logging.error(f"Error adding documents to database: {str(e)}", exc_info=True)
                 # If adding documents fails, create a new database with these documents
-                database = WebSearchFAISS.from_documents(documents, get_embeddings())
+                database = FAISS.from_documents(documents, get_embeddings())
                 logging.info("Created new WebSearchFAISS database with search results")
         return database, temp_dir, results
     except Exception as e: