for marinating the chat history via pickle

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from huggingface_hub import InferenceClient
 import os
 import sys
+import pickle
 
 st.title("CODEFUSSION ☄")
 
@@ -53,6 +54,20 @@ def reset_conversation():
     st.session_state.messages = []
     return None
 
+def load_conversation_history():
+    history_file = "conversation_history.pickle"
+    if os.path.exists(history_file):
+        with open(history_file, "rb") as f:
+            conversation_history = pickle.load(f)
+    else:
+        conversation_history = []
+    return conversation_history
+
+def save_conversation_history(conversation_history):
+    history_file = "conversation_history.pickle"
+    with open(history_file, "wb") as f:
+        pickle.dump(conversation_history, f)
+
 models = [key for key in model_links.keys()]
 selected_model = st.sidebar.selectbox("Select Model", models)
 temp_values = st.sidebar.slider('Select a temperature value', 0.0, 1.0, (0.5))
@@ -76,8 +91,8 @@ reset_conversation()
 repo_id = model_links[selected_model]
 st.subheader(f'{selected_model}')
 
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+# Load the conversation history from the file
+st.session_state.messages = load_conversation_history()
 
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
@@ -104,4 +119,7 @@ if prompt := st.chat_input(f"Hi I'm {selected_model}, How can I help you today?"
             stream=True
         )
         response = st.write_stream(output)
-        st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.messages.append({"role": "assistant", "content": response})
+        
+        # Save the updated conversation history to the file
+        save_conversation_history(st.session_state.messages)