Update app.py

app.py

@@ -11,11 +11,9 @@ from langchain.document_loaders import TextLoader
 from langchain.document_loaders import Docx2txtLoader
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 import os
-from dotenv import load_dotenv
 import tempfile
 
-load_dotenv()
-
+# Initialize session state
 def initialize_session_state():
     if 'history' not in st.session_state:
         st.session_state['history'] = []
@@ -26,11 +24,13 @@ def initialize_session_state():
     if 'past' not in st.session_state:
         st.session_state['past'] = ["Hey! 👋"]
 
+# Conversation chat function
 def conversation_chat(query, chain, history):
     result = chain({"question": query, "chat_history": history})
     history.append((query, result["answer"]))
     return result["answer"]
 
+# Display chat history
 def display_chat_history(chain):
     reply_container = st.container()
     container = st.container()
@@ -53,9 +53,8 @@ def display_chat_history(chain):
                 message(st.session_state["past"][i], is_user=True, key=str(i) + '_user', avatar_style="thumbs")
                 message(st.session_state["generated"][i], key=str(i), avatar_style="fun-emoji")
 
+# Create conversational chain
 def create_conversational_chain(vector_store):
-    load_dotenv()
-
     replicate_api_token = "r8_MgTUrfPJIluDoXUhG7JXuPAYr6PonOW4BJCj0"
     os.environ["REPLICATE_API_TOKEN"] = replicate_api_token
 
@@ -69,12 +68,12 @@ 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
 
+# Main function
 def main():
-    load_dotenv()
     initialize_session_state()
     st.title("Chat With Your Doc")
     st.sidebar.title("Document Processing")
@@ -110,4 +109,4 @@ def main():
         display_chat_history(chain)
 
 if __name__ == "__main__":
-    main()
+    main()