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,10 +28,9 @@ 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 = {
@@ -78,7 +77,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"])
 
@@ -88,32 +87,10 @@ def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
         if chunk.choices[0].delta.content:
             yield chunk.choices[0].delta.content
 
-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:
+if prompt := st.chat_input("Enter your prompt here..."):
     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
@@ -129,11 +106,11 @@ if prompt:
         )
 
         # 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):