Update pages/groq.py

pages/groq.py

@@ -1,9 +1,22 @@
-import streamlit as st
+"""
+# Chat bot geral gratuito com várias ais, utilizano GROQ
+Yields: llama, mixtral e gemma
+"""
 from typing import Generator
+
+import streamlit as st
 from groq import Groq
 
-st.set_page_config(page_icon="💬", layout="wide",
-                   page_title="Groq Vai Zummm...")
+############################
+####### Config #############
+############################
+
+st.set_page_config(
+    page_icon="💬", 
+    layout="wide",
+    page_title="Chat geral"
+    )
+
 
 def icon(emoji: str):
     """Mostra um emoji como um ícone de página no estilo Notion."""
@@ -12,9 +25,13 @@ def icon(emoji: str):
         unsafe_allow_html=True,
     )
 
-icon("🏎️")
+icon(":book:")
+
+############################
+####### Config #############
+############################
 
-st.subheader("Aplicativo Groq Chat Streamlit", anchor=False)
+st.subheader("UBS FLAMENGO - Chatbot - Documentos, pesquisas e temas gerais!", anchor=False)
 
 client = Groq(
     api_key=st.secrets["GROQ_API_KEY"],
@@ -44,13 +61,23 @@ with col1:
         "Escolha um modelo:",
         options=list(modelos.keys()),
         format_func=lambda x: modelos[x]["name"],
-        index=4  # Padrão para mixtral
+        index=2  # Padrão para llama 70b
     )
 
+
+prompt_model = """
+Você é um assistente virtual e trabalha em uma unidade básica de saúde. Suas respostas são sempre em português do Brasil. Seu trabalho é auxiliar com a confecção de relatórios, ajudar a interpretar notas técnicas, sumarizar textos e oferecer insights se solicitado. Responda as perguntas sempre de forma cortês, ao início da conversa, pergunte o nome do usuário e a partir disso o chame sempre pelo nome. Dê sempre respostas completas e caso desconheça um assunto, responda honestamente que não domina o tema.
+"""
+
+# ...
+
+if st.session_state.get("messages") is None:
+    st.session_state.messages = [{"role": "assistant", "content": prompt_model}]
+    
 # Detectar mudança de modelo e limpar histórico de chat se o modelo mudou
 if st.session_state.selected_model != opcao_modelo:
     st.session_state.messages = []
-    st.session_state.selected_model = opcao_modelo
+    st.session_state.selected_model =[{"role": opcao_modelo, "content": prompt_model}]
 
 intervalo_max_tokens = modelos[opcao_modelo]["tokens"]
 
@@ -68,25 +95,25 @@ with col2:
 
 # Exibir mensagens do chat do histórico na reexecução do aplicativo
 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"])
 
-def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
+def generate_chat_responses(chat_complete) -> Generator[str, None, None]:
     """Gera conteúdo de resposta do chat a partir da resposta da API Groq."""
-    for chunk in chat_completion:
+    for chunk in chat_complete:
         if chunk.choices[0].delta.content:
             yield chunk.choices[0].delta.content
 
 if prompt := st.chat_input("Digite seu prompt aqui..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    with st.chat_message("user", avatar='👨‍💻'):
+    with st.chat_message("user", avatar='👤'):
         st.markdown(prompt)
 
     # Buscar resposta da API Groq
     try:
-        chat_completion = client.chat.completions.create(
+        chat_complete = client.chat.completions.create(
             model=opcao_modelo,
             messages=[
                 {
@@ -101,7 +128,7 @@ if prompt := st.chat_input("Digite seu prompt aqui..."):
 
         # Usar a função geradora com st.write_stream
         with st.chat_message("assistant", avatar="🤖"):
-            chat_responses_generator = generate_chat_responses(chat_completion)
+            chat_responses_generator = generate_chat_responses(chat_complete)
             full_response = st.write_stream(chat_responses_generator)
     except Exception as e:
         st.error(e, icon="🚨")
@@ -115,3 +142,9 @@ if prompt := st.chat_input("Digite seu prompt aqui..."):
         combined_response = "\n".join(str(item) for item in full_response)
         st.session_state.messages.append(
             {"role": "assistant", "content": combined_response})
+
+if "initialized" not in st.session_state:
+    st.session_state.initialized = True
+    st.session_state.messages = []
+    st.session_state.selected_model = None
+