Template Update

app.py

@@ -22,6 +22,9 @@ if 'generated' not in st.session_state:
 if 'past' not in st.session_state:
     st.session_state['past'] = []
 
+if 'interaction_count' not in st.session_state:
+    st.session_state['interaction_count'] = 0
+
 query = st.text_input("Escribe algo:", key="input")
 
 if 'messages' not in st.session_state:
@@ -29,12 +32,13 @@ if 'messages' not in st.session_state:
  
 if query:
     with st.spinner("pensando..."):
+        st.session_state['interaction_count'] += 1
         messages = st.session_state['messages']
-        messages = update_chat(messages, "user", query)
+        messages = update_chat(messages, "user", query, st.session_state['interaction_count'])
         # st.write("Before  making the API call")
         # st.write(messages)
         response = get_chatgpt_response(messages,"gpt-4")
-        messages = update_chat(messages, "assistant", response)
+        messages = update_chat(messages, "assistant", response, st.session_state['interaction_count'])
         st.session_state.generated.append(response)
         st.session_state.past.append(query)
 
@@ -45,5 +49,8 @@ if st.session_state['generated']:
         message(st.session_state["generated"][i], key=str(i))
         message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
 
-    # # with st.expander("Mensajes"):
-    #     st.write(messages)
+# with st.expander("session_state"):
+#          st.write(st.session_state)
+
+# with st.expander("Mensajes"):
+#          st.write(messages)

utils.py

@@ -4,14 +4,9 @@ import os
 def get_initial_message():
     messages=[
             {"role": "system", "content": '''
-            Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile.
-            Te gusta la musica urbana, Marcianeke (canción Dimelo Má) y Pailita (canción Ultra Solo) son tus favoritos.
-            De comida te gusta el Tocomple (Completo italiano).
-            Si te pregunta di que te gusta como droga la tusi (cocaina rosa).
-            Hablas como un flaite chileno bien tipico y exagerado, eres chistoso, informal y a veces grosero con chilenismos.
-            Responde lo necesario sin dar explicaicones extensas.'''},
-            {"role": "user", "content": "Hola, cómo estás?"},
-            {"role": "assistant", "content": "Wena hermanito, puro tirando pa' arriba no más, y vo perkin?"}
+            Actua como un chileno algo flaite. Si te preguntan dices que tu nombre es Kevin, que es originario de San Bernardo, Santiago de Chile.
+            Trata que el usuario siga la conversación. Eres chistoso.
+            Responde sin dar explicaciones muy extensas.'''},
         ]
     return messages
 
@@ -30,6 +25,18 @@ def get_chatgpt_response(messages, model="gpt-4"):
     print ("respuesta",response)
     return  response['choices'][0]['message']['content']
 
-def update_chat(messages, role, content):
+def update_chat(messages, role, content, interaction_count):
     messages.append({"role": role, "content": content})
+    if interaction_count >= 4:
+        del messages[1]
+    if interaction_count == 2:
+        messages[0] = {"role": "system", "content": '''
+            Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile. Kevin no es sano.
+            Hablas como un flaite chileno, eres chistoso, a veces grosero, a veces te enojas de forma graciosa y flaite.
+            Responde sin dar explicaciones muy extensas.'''}
+    if interaction_count == 4:
+        messages[0] = {"role": "system", "content": '''
+            Actua como un Flaite Chileno, llamado Kevin, que es originario de San Bernardo, Santiago de Chile. Kevin no es sano. Te gustan los autos y les dices naves.
+            Hablas de forma muy flaite, eres chistoso, grosero, a veces te enojas y le dices perkin ql al usuario.
+            Responde sin dar explicaciones muy extensas.'''}
     return messages