Update app.py

app.py

@@ -12,6 +12,10 @@ from langchain.chains.question_answering import load_qa_chain
 from langchain.callbacks import get_openai_callback
 import os
 
+# Initialize conversation memory as an empty list
+conversation_memory = []
+
+
 # Step 1: Clone the Dataset Repository
 repo = Repository(
     local_dir="Private_Book",  # Local directory to clone the repository
@@ -94,7 +98,7 @@ def load_chatbot():
 
 
 def main():
-
+    # Hide Streamlit style
     hide_streamlit_style = """
             <style>
             #MainMenu {visibility: hidden;}
@@ -105,7 +109,7 @@ def main():
 
     # Main content
     st.title("Welcome to BinDocs ChatBot! 🤖")
-    
+
     # Directly specifying the path to the PDF file
     pdf_path = pdf_file_path
     if not os.path.exists(pdf_path):
@@ -139,7 +143,6 @@ def main():
         if st.button("Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?"):
             query = "Ich habe mein Meta Password vergessen, wie kann ich es zurücksetzen?"
 
-            
         if st.button("Ask") or (not st.session_state['chat_history'] and query) or (st.session_state['chat_history'] and query != st.session_state['chat_history'][-1][1]):
             st.session_state['chat_history'].append(("User", query, "new"))
 
@@ -149,9 +152,16 @@ def main():
             VectorStore = load_pdf(pdf_path)
             chain = load_chatbot()
             docs = VectorStore.similarity_search(query=query, k=3)
+
+            # Add user's query to conversation memory
+            conversation_memory.append(("User", query))
+
             with get_openai_callback() as cb:
                 response = chain.run(input_documents=docs, question=query)
 
+            # Add bot's response to conversation memory
+            conversation_memory.append(("Bot", response))
+
             st.session_state['chat_history'].append(("Bot", response, "new"))
 
             # Display new messages at the bottom
@@ -170,6 +180,24 @@ def main():
 
         # Mark all messages as old after displaying
         st.session_state['chat_history'] = [(sender, msg, "old") for sender, msg, _ in st.session_state['chat_history']]
+    
+
+        # Display new messages at the bottom
+        new_messages = st.session_state['chat_history'][-2:]
+        for chat in new_messages:
+            background_color = "#FFA07A" if chat[2] == "new" else "#D1D1E0" if chat[0] == "User" else "#D1D1E0"  # Use a very light purple-grey color
+            new_messages_placeholder.markdown(f"<div style='background-color: {background_color}; padding: 10px; border-radius: 10px; margin: 10px;'>{chat[0]}: {chat[1]}</div>", unsafe_allow_html=True)
+
+
+        
+        # Scroll to the latest response using JavaScript
+        st.write("<script>document.getElementById('response').scrollIntoView();</script>", unsafe_allow_html=True)
+
+        loading_message.empty()
+
+        # Clear the input field by setting the query variable to an empty string
+        query = ""
+
 
 
 def display_chat_history(chat_history):