dromerosm commited on
Commit
69b90ae
β€’
1 Parent(s): e3c319b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -76,7 +76,7 @@ if st.button("Clear Chat"):
76
 
77
  # Display chat messages from history on app rerun
78
  for message in st.session_state.messages:
79
- avatar = "πŸ€–" if message["role"] == "assistant" else "πŸ•Ί"
80
  with st.chat_message(message["role"], avatar=avatar):
81
  st.markdown(message["content"])
82
 
@@ -91,7 +91,7 @@ def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
91
  if prompt := st.chat_input("Enter your prompt here..."):
92
  st.session_state.messages.append({"role": "user", "content": prompt})
93
 
94
- with st.chat_message("user", avatar="πŸ•Ί"):
95
  st.markdown(prompt)
96
 
97
  # Fetch response from Groq API
@@ -107,7 +107,7 @@ if prompt := st.chat_input("Enter your prompt here..."):
107
  )
108
 
109
  # Use the generator function with st.write_stream
110
- with st.chat_message("assistant", avatar="πŸ€–"):
111
  chat_responses_generator = generate_chat_responses(chat_completion)
112
  full_response = st.write_stream(chat_responses_generator)
113
  except Exception as e:
 
76
 
77
  # Display chat messages from history on app rerun
78
  for message in st.session_state.messages:
79
+ avatar = "πŸ”‹" if message["role"] == "assistant" else "πŸ§‘β€πŸ’»"
80
  with st.chat_message(message["role"], avatar=avatar):
81
  st.markdown(message["content"])
82
 
 
91
  if prompt := st.chat_input("Enter your prompt here..."):
92
  st.session_state.messages.append({"role": "user", "content": prompt})
93
 
94
+ with st.chat_message("user", avatar="πŸ§‘β€πŸ’»"):
95
  st.markdown(prompt)
96
 
97
  # Fetch response from Groq API
 
107
  )
108
 
109
  # Use the generator function with st.write_stream
110
+ with st.chat_message("assistant", avatar="πŸ”‹"):
111
  chat_responses_generator = generate_chat_responses(chat_completion)
112
  full_response = st.write_stream(chat_responses_generator)
113
  except Exception as e: