Update app.py

app.py

@@ -278,7 +278,19 @@ def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temp
 def get_web_search_database():
     embed = get_embeddings()
     temp_dir = tempfile.mkdtemp()
-    database = WebSearchFAISS.from_documents([], embed)
+    
+    # Create a dummy document to initialize the database
+    dummy_doc = WebSearchDocument(page_content="Dummy content", metadata={"source": "dummy"})
+    
+    try:
+        database = WebSearchFAISS.from_documents([dummy_doc], embed)
+        # Remove the dummy document after initialization
+        database.delete([dummy_doc.metadata["source"]])
+    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.embed_query, None, {}, {}, None)
+    
     return database, temp_dir
 
 def cleanup_web_search_database(temp_dir):
@@ -286,21 +298,26 @@ def cleanup_web_search_database(temp_dir):
 
 def duckduckgo_search(query):
     logging.info(f"Performing DuckDuckGo search for query: {query}")
-    with DDGS() as ddgs:
-        results = list(ddgs.text(query, max_results=5))
-    
-    logging.info(f"Number of search results: {len(results)}")
-    
-    database, temp_dir = get_web_search_database()
-    documents = []
-    for result in results:
-        content = f"{result['title']}\n{result['body']}"
-        doc = WebSearchDocument(page_content=content, metadata={"source": result['href']})
-        documents.append(doc)
-    
-    logging.info(f"Number of documents created: {len(documents)}")
-    database.add_documents(documents)
-    return database, temp_dir, results
+    try:
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=5))
+        
+        logging.info(f"Number of search results: {len(results)}")
+        
+        database, temp_dir = get_web_search_database()
+        documents = []
+        for result in results:
+            content = f"{result['title']}\n{result['body']}"
+            doc = WebSearchDocument(page_content=content, metadata={"source": result['href']})
+            documents.append(doc)
+        
+        logging.info(f"Number of documents created: {len(documents)}")
+        if documents:
+            database.add_documents(documents)
+        return database, temp_dir, results
+    except Exception as e:
+        logging.error(f"Error in duckduckgo_search: {str(e)}", exc_info=True)
+        return None, None, []
 
 def retrieve_web_search_results(database, query):
     logging.info(f"Retrieving web search results for query: {query}")
@@ -356,10 +373,8 @@ def respond(message: str, history: List[Tuple[str, str]], model: str, temperatur
     try:
         if use_web_search:
             database, temp_dir, search_results = duckduckgo_search(message)
-            logging.info(f"Number of search results: {len(search_results)}")
-            
-            if not search_results:
-                yield "I'm sorry, but I couldn't find any search results for your query. Could you please rephrase or ask a different question?"
+            if database is None:
+                yield "I'm sorry, but I encountered an error while searching. Please try again later."
                 return
             
             context = retrieve_web_search_results(database, message)