Update app.py

app.py

@@ -81,6 +81,18 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
+# Function to save prompt
+def save_prompt(prompt):
+    if "saved_prompts" not in st.session_state:
+        st.session_state.saved_prompts = []
+    st.session_state.saved_prompts.append(prompt)
+
+# Function to delete prompt
+def delete_prompt(prompt_index):
+    if "saved_prompts" in st.session_state:
+        del st.session_state.saved_prompts[prompt_index]
+
+
 def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
     """Yield chat response content from the Groq API response."""
     for chunk in chat_completion:
@@ -122,4 +134,13 @@ if prompt := st.chat_input("Enter your prompt here..."):
         combined_response = "\n".join(str(item) for item in full_response)
         st.session_state.messages.append(
             {"role": "assistant", "content": combined_response}
-        )
+        )
+
+# 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):
+            delete_button = st.button("Delete", key=f"delete_{i}")
+            st.write(f"{i + 1}. {prompt}")
+            if delete_button:
+                delete_prompt(i)