Update app.py

app.py

@@ -81,16 +81,19 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
-# Function to save prompt
+# Function to save prompt to a file
 def save_prompt(prompt):
-    if "saved_prompts" not in st.session_state:
-        st.session_state.saved_prompts = []
-    st.session_state.saved_prompts.append(prompt)
+    with open("saved_prompts.txt", "a") as file:
+        file.write(prompt + "\n")
 
-# Function to delete prompt
+# Function to delete prompt from file
 def delete_prompt(prompt_index):
-    if "saved_prompts" in st.session_state:
-        del st.session_state.saved_prompts[prompt_index]
+    with open("saved_prompts.txt", "r") as file:
+        prompts = file.readlines()
+    with open("saved_prompts.txt", "w") as file:
+        for i, p in enumerate(prompts):
+            if i != prompt_index:
+                file.write(p)
 
 
 def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
@@ -105,6 +108,9 @@ if prompt := st.chat_input("Enter your prompt here..."):
     with st.chat_message("user", avatar="❓"):  
         st.markdown(prompt)
 
+    if st.button("Save"):
+        save_prompt(prompt)
+
     # Fetch response from Groq API
     try:
         chat_completion = client.chat.completions.create(
@@ -138,9 +144,11 @@ if prompt := st.chat_input("Enter your prompt here..."):
 
 # Collapsible section for prompt management
 with st.expander("Prompt Management", expanded=False):
-    if "saved_prompts" in st.session_state:
-        for i, prompt in enumerate(st.session_state.saved_prompts):
+    if os.path.exists("saved_prompts.txt"):
+        with open("saved_prompts.txt", "r") as file:
+            saved_prompts = file.readlines()
+        for i, prompt in enumerate(saved_prompts):
             delete_button = st.button("Delete", key=f"delete_{i}")
-            st.write(f"{i + 1}. {prompt}")
+            st.write(f"{i + 1}. {prompt.strip()}")
             if delete_button:
                 delete_prompt(i)