Update app.py

app.py

@@ -1,11 +1,16 @@
 import streamlit as st
 import requests
 import json
+import os
 
 # Constants
 SPACE_URL = "https://abanm-dubs.hf.space/api/generate"
 API_KEY = "s3cr3t_k3y"  # Replace with your actual API key
 EOS_TOKEN = "<|end|>"
+CHAT_HISTORY_DIR = "chat_histories"  # Directory to save chat histories
+
+# Ensure the directory exists
+os.makedirs(CHAT_HISTORY_DIR, exist_ok=True)
 
 # Streamlit Configurations (must come first)
 st.set_page_config(page_title="🐶 DUBSChat", layout="wide")
@@ -14,8 +19,30 @@ st.set_page_config(page_title="🐶 DUBSChat", layout="wide")
 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("🔄 New Chat"):
+    
+    # Save chat history
+    if st.button("💾 Save Chat"):
+        if "messages" in st.session_state:
+            file_name = st.text_input("Enter filename to save chat:", "chat_history.json")
+            if file_name:
+                file_path = os.path.join(CHAT_HISTORY_DIR, file_name)
+                with open(file_path, "w") as f:
+                    json.dump(st.session_state["messages"], f)
+                st.success(f"Chat saved as {file_name}!")
+
+    # Load chat history
+    file_to_load = st.selectbox("Select a chat to load:", os.listdir(CHAT_HISTORY_DIR))
+    if st.button("📂 Load Chat"):
+        file_path = os.path.join(CHAT_HISTORY_DIR, file_to_load)
+        if os.path.exists(file_path):
+            with open(file_path, "r") as f:
+                st.session_state["messages"] = json.load(f)
+            st.success(f"Loaded chat from {file_to_load}!")
+        else:
+            st.error(f"File {file_to_load} does not exist.")
+
+    # Reset chat
+    if st.button("🔄 Reset Chat"):
         st.session_state["messages"] = [
             {"role": "system", "content": "You are DUBS, a helpful assistant capable of conversing in a friendly and knowledgeable way."},
             {"role": "assistant", "content": "Hello! How can I assist you today?"}