Update app.py

app.py

@@ -10,10 +10,10 @@ def save_docs(docs):
     import shutil
     import os
 
-    destination_dir = "/kaggle/working/docs/"
+    destination_dir = "/home/user/app/docs/"
     os.makedirs(destination_dir, exist_ok=True)
 
-    output_dir="/kaggle/working/docs/"
+    output_dir="/home/user/app/docs/"
 
     for doc in docs:
       shutil.copy(doc.name, output_dir)
@@ -30,13 +30,13 @@ def process_docs():
     from langchain.embeddings.openai import OpenAIEmbeddings
     from langchain.text_splitter import RecursiveCharacterTextSplitter
 
-    loader1 = DirectoryLoader('/kaggle/working/docs/', glob="./*.pdf", loader_cls=PyPDFLoader)
+    loader1 = DirectoryLoader('/home/user/app/docs/', glob="./*.pdf", loader_cls=PyPDFLoader)
     document1 = loader1.load()
 
-    loader2 = DirectoryLoader('/kaggle/working/docs/', glob="./*.txt", loader_cls=TextLoader)
+    loader2 = DirectoryLoader('/home/user/app/docs/', glob="./*.txt", loader_cls=TextLoader)
     document2 = loader2.load()
 
-    loader3 = DirectoryLoader('/kaggle/working/docs/', glob="./*.docx", loader_cls=Docx2txtLoader)
+    loader3 = DirectoryLoader('/home/user/app/docs/', glob="./*.docx", loader_cls=Docx2txtLoader)
     document3 = loader3.load()
 
     document1.extend(document2)
@@ -52,7 +52,7 @@ def process_docs():
     embeddings = OpenAIEmbeddings()
 
     docs_db = FAISS.from_documents(docs, embeddings)
-    docs_db.save_local("/kaggle/working/docs_db/")
+    docs_db.save_local("/home/user/app/docs_db/")
 
     return "File(s) processed successfully!"
 
@@ -82,7 +82,7 @@ def search_docs(question):
     from langchain.chat_models import ChatOpenAI
 
     embeddings = OpenAIEmbeddings()
-    docs_db = FAISS.load_local("/kaggle/working/docs_db/", embeddings)
+    docs_db = FAISS.load_local("/home/user/app/docs_db/", embeddings)
     docs = docs_db.similarity_search(question)
 
     llm = ChatOpenAI(model_name='gpt-3.5-turbo')
@@ -98,8 +98,8 @@ def delete_docs():
 
     import shutil
 
-    path1 = "/kaggle/working/docs/"
-    path2 = "/kaggle/working/docs_db/"
+    path1 = "/home/user/app/docs/"
+    path2 = "/home/user/app/docs_db/"
 
     try:
         shutil.rmtree(path1)