Update app.py

app.py

@@ -7,7 +7,7 @@ import datetime
 import json
 
 _ = load_dotenv(find_dotenv())
-st.set_page_config(page_icon="💬", layout="wide", page_title="...")
+st.set_page_config(page_icon="", layout="wide", page_title="...")
 
 def icon(emoji: str):
     """Shows an emoji as a Notion-style page icon."""
@@ -28,9 +28,10 @@ client = Groq(
 # Initialize chat history and selected model
 if "messages" not in st.session_state:
     st.session_state.messages = []
-
 if "selected_model" not in st.session_state:
     st.session_state.selected_model = None
+if "prompts" not in st.session_state:
+    st.session_state.prompts = {}
 
 # Define model details
 models = {
@@ -77,7 +78,7 @@ with col2:
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
-    avatar = "🧠" if message["role"] == "assistant" else "❓"
+    avatar = "" if message["role"] == "assistant" else ""
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
@@ -87,10 +88,32 @@ def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
         if chunk.choices[0].delta.content:
             yield chunk.choices[0].delta.content
 
-if prompt := st.chat_input("Enter your prompt here..."):
+with st.expander("Save Prompt"):
+    save_prompt_button = st.button("Save Prompt")
+    if save_prompt_button:
+        prompt_name = st.text_input("Enter a name for the prompt:")
+        if prompt_name:
+            st.session_state.prompts[prompt_name] = st.session_state.messages[-1]["content"]
+            st.success(f"Prompt '{prompt_name}' saved!")
+
+with st.expander("Load Prompt"):
+    load_prompt_option = st.selectbox(
+        "Select a saved prompt:",
+        options=list(st.session_state.prompts.keys()),
+    )
+    if load_prompt_option:
+        st.session_state.messages[-1]["content"] = st.session_state.prompts[load_prompt_option]
+        st.write("Prompt loaded!")
+
+if load_prompt_option:
+    prompt = st.session_state.prompts[load_prompt_option]
+else:
+    prompt = st.chat_input("Enter your prompt here...")
+
+if prompt:
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    with st.chat_message("user", avatar="❓"):  
+    with st.chat_message("user", avatar=""):  
         st.markdown(prompt)
 
     # Fetch response from Groq API
@@ -106,11 +129,11 @@ if prompt := st.chat_input("Enter your prompt here..."):
         )
 
         # Use the generator function with st.write_stream
-        with st.chat_message("assistant", avatar="🧠"):
+        with st.chat_message("assistant", avatar=""):
             chat_responses_generator = generate_chat_responses(chat_completion)
             full_response = st.write_stream(chat_responses_generator)
     except Exception as e:
-        st.error(e, icon="🚨")
+        st.error(e, icon="")
 
     # Append the full response to session_state.messages
     if isinstance(full_response, str):