update app.py

app.py

@@ -6,16 +6,15 @@ from langchain.llms import Replicate
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.vectorstores import FAISS
 from langchain.memory import ConversationBufferMemory
-from langchain.document_loaders import PyPDFLoader
-from langchain.document_loaders import TextLoader
-from langchain.document_loaders import Docx2txtLoader
+from langchain.document_loaders import PyPDFLoader, TextLoader, Docx2txtLoader
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 import os
 from dotenv import load_dotenv
 import tempfile
 
-# Set REPLICATE_API_TOKEN environment variable
-os.environ["REPLICATE_API_TOKEN"] = "r8_AA3K1fhDykqLa5M74E5V0w5ss1z0P9S3foWJl"
+# Set replicate API token as environment variable
+replicate_api_token = "r8_AA3K1fhDykqLa5M74E5V0w5ss1z0P9S3foWJl"
+os.environ["REPLICATE_API_TOKEN"] = replicate_api_token
 
 load_dotenv()
 
@@ -53,16 +52,12 @@ def display_chat_history(chain):
                     message(st.session_state["generated"][i], key=str(i), avatar_style="fun-emoji")
 
 def create_conversational_chain(vector_store):
-    load_dotenv()
-
-    replicate_api_token = "r8_AA3K1fhDykqLa5M74E5V0w5ss1z0P9S3foWJl"  # Replace with your actual token
-    os.environ["REPLICATE_API_TOKEN"] = replicate_api_token
-
     llm = Replicate(
         streaming=True,
         model="replicate/llama-2-70b-chat:58d078176e02c219e11eb4da5a02a7830a283b14cf8f94537af893ccff5ee781",
         callbacks=[StreamingStdOutCallbackHandler()],
-        input={"temperature": 0.01, "max_length": 500, "top_p": 1}
+        input={"temperature": 0.01, "max_length": 500, "top_p": 1},
+        replicate_api_token=replicate_api_token
     )
     memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
@@ -72,7 +67,6 @@ def create_conversational_chain(vector_store):
     return chain
 
 def main():
-    load_dotenv()
     initialize_session_state()
     st.title("Chat With Your Doc")
     st.sidebar.title("Document Processing")
@@ -89,7 +83,7 @@ def main():
             loader = None
             if file_extension == ".pdf":
                 loader = PyPDFLoader(temp_file_path)
-            elif file_extension == ".docx" or file_extension == ".doc":
+            elif file_extension in (".docx", ".doc"):
                 loader = Docx2txtLoader(temp_file_path)
             elif file_extension == ".txt":
                 loader = TextLoader(temp_file_path)