Update app.py

app.py

@@ -1,19 +1,25 @@
 import streamlit as st
 import requests
 import json
+from langchain.memory import ConversationBufferMemory
 
 # Constants
 SPACE_URL = "https://abanm-dubs.hf.space/api/generate"
-API_KEY = "s3cr3t_k3y"  # Replace with your actual API key
 
 # Streamlit Configurations (must come first)
 st.set_page_config(page_title="🐶 DUBSChat", layout="centered")
 
+# Initialize memory for context retention
+if "memory" not in st.session_state:
+    st.session_state["memory"] = ConversationBufferMemory()
+
 # Sidebar Configuration
 with st.sidebar:
     dubs_key = st.text_input("Enter Dubs Key", key="chatbot_api_key", type="password")
     st.markdown("Dubs Recall")
-    st.markdown("""Coming Soon!""")
+    if st.button("Show Conversation History"):
+        if "memory" in st.session_state:
+            st.text_area("Conversation History", st.session_state["memory"].buffer)
 
 def call_api(message):
     """
@@ -29,7 +35,10 @@ def call_api(message):
         "Authorization": f"Bearer {dubs_key}",
         "Content-Type": "application/json",
     }
-    payload = {"prompt": message}  # Match the API's expected input key
+    # Add context to the prompt
+    context = st.session_state["memory"].buffer
+    prompt = f"{context}\nUser: {message}\nAI:"
+    payload = {"prompt": prompt}  # Match the API's expected input key
 
     try:
         # Send request with streaming enabled
@@ -81,4 +90,5 @@ if prompt := st.chat_input():
         # Append the final assistant's response to the chat history
         st.session_state["messages"].append({"role": "assistant", "content": full_response})
 
-
+        # Save user input and bot response to memory for context retention
+        st.session_state["memory"].save_context({"User": prompt}, {"AI": full_response})