Update app.py

app.py

@@ -56,8 +56,9 @@ def display_chat_history(chain):
 def create_conversational_chain(vector_store):
     load_dotenv()
 
-    replicate_api_token = "r8_MgTUrfPJIluDoXUhG7JXuPAYr6PonOW4BJCj0"
-    os.environ["REPLICATE_API_TOKEN"] = replicate_api_token
+    replicate_api_token = os.getenv("REPLICATE_API_TOKEN")
+    if not replicate_api_token:
+        raise ValueError("Replicate API token is not set. Please set the REPLICATE_API_TOKEN environment variable.")
 
     llm = Replicate(
         streaming=True,
@@ -69,8 +70,8 @@ def create_conversational_chain(vector_store):
     memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
     chain = ConversationalRetrievalChain.from_llm(llm=llm, chain_type='stuff',
-                                                 retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
-                                                 memory=memory)
+                                                  retriever=vector_store.as_retriever(search_kwargs={"k": 2}),
+                                                  memory=memory)
     return chain
 
 def main():