Update app.py

app.py

@@ -240,7 +240,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
             # Convert web search results to Document format
             web_docs = [Document(page_content=result["text"], metadata={"source": result["link"]}) for result in search_results if result["text"]]
             
-            # Load or create the vector database
+            # Check if the FAISS database exists
             if os.path.exists("faiss_database"):
                 database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
                 database.add_documents(web_docs)
@@ -259,7 +259,12 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
             prompt_val = ChatPromptTemplate.from_template(prompt_template)
             formatted_prompt = prompt_val.format(context=context_str, question=question)
         else:
-            database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
+            # Check if the FAISS database exists
+            if os.path.exists("faiss_database"):
+                database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
+            else:
+                return "No FAISS database found. Please upload documents to create the vector store."
+            
             history_str = "\n".join([f"Q: {item['question']}\nA: {item['answer']}" for item in conversation_history])
             
             if is_related_to_history(question, conversation_history):