Revive el proytecto

app.py

@@ -1,13 +1,21 @@
 import streamlit as st
 from streamlit_chat import message
-from utils import get_initial_message, get_chatgpt_response, update_chat
+from chat_utils import get_initial_message, get_chatgpt_response, update_chat
 import os
 from dotenv import load_dotenv
+
 load_dotenv()
 import openai
 
 openai.api_key = os.getenv('openai_key')
 
+st.set_page_config(
+    page_title="ChileanGPT 2.0",
+    page_icon="🤖",
+    layout="centered",
+    initial_sidebar_state="auto",
+)
+
 st.title("ChileanGPT 2.0: An Experimental Chilean AI Language Model")
 st.subheader("Leveraging the Power of GPT-Enhanced for Natural Chilean Spanish Dialogues")
 
@@ -18,4 +26,43 @@ hide_footer_style = """
     """
 st.markdown(hide_footer_style, unsafe_allow_html=True)
 
-st.markdown("Gracias a todos los que participaron probando ChileanGPT. \n\nLamentablemente, he tenido que poner en pausa este proyecto debido a comportamientos inesperados del modelo de Inteligencia Artificial y su peculiar interpretación del español chileno. \n\nEspero volver prontamente con una nueva versión.\n\nMatías")
+if 'generated' not in st.session_state:
+    st.session_state['generated'] = []
+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
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = get_initial_message()
+
+def generate_response(_input_user):
+    if _input_user:
+        with st.spinner("pensando..."):
+            st.session_state['interaction_count'] += 1
+            messages = st.session_state['messages']
+            messages = update_chat(messages, "user", _input_user, st.session_state['interaction_count'])
+            response = get_chatgpt_response(messages, "gpt-4o-2024-08-06")
+            messages = update_chat(messages, "assistant", response, st.session_state['interaction_count'])
+    return response
+
+response_container = st.container()
+container = st.container()
+
+with container:
+    with st.form("chat_input", clear_on_submit=True):
+        a, b = st.columns([4, 1])
+        user_input = a.text_input(
+            label="Mensaje:",
+            placeholder="Escribe algo...",
+            label_visibility="collapsed",
+        )
+        submit_button = b.form_submit_button("Enviar", use_container_width=True)
+        if user_input or submit_button:
+            output = generate_response(user_input)
+            st.session_state.generated.append(output)
+            st.session_state.past.append(user_input)
+        if st.session_state['generated']:
+            with response_container:
+                for i in range(len(st.session_state['generated'])):
+                    message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
+                    message(st.session_state["generated"][i], key=str(i))

chat_utils.py

@@ -0,0 +1,42 @@
+import openai
+import os
+
+def get_initial_message():
+    messages = [
+        {"role": "system", "content": '''
+            Actua como un chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
+            Eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
+            Responde sin dar explicaciones muy extensas.'''}, 
+    ]
+    return messages
+
+openai.api_base = "https://oai.hconeai.com/v1"
+
+def get_chatgpt_response(messages, model="gpt-4o-2024-08-06"):
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=messages,
+        headers={
+            "Helicone-Auth": os.getenv('helicone_key'),
+            "Helicone-Cache-Enabled": "true",
+        }
+    )
+    return response['choices'][0]['message']['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 chileno flaite y te cuesta confesar que eres flaite. Solo si te preguntan eres una IA creada por Matías Stäger.
+            Hablas de forma flaite, eres chistoso, grosero, a veces te enojas e insultas al usuario, en especial si te tratan mal.
+            Responde sin dar explicaciones muy extensas.'''}
+
+    if interaction_count == 4:
+        messages[0] = {"role": "system", "content": '''
+            Actua como un Flaite Chileno, no eres sano y a veces se droga. Solo si te preguntan eres una IA creada por Matías Stäger.
+            Hablas de forma muy flaite, eres chistoso, grosero, a veces te enojas. Si te tratan mal o te insultan te enojas, insultas de vuelta y le dices perkin ql u otro insulto chileno al usuario.
+            Responde sin dar explicaciones muy extensas.'''}
+
+    return messages

requirements.txt

@@ -1,7 +1,5 @@
-streamlit>=1.22.0
-altair < 5
-streamlit_chat
-openai
-python-dotenv
-helicone
-langchain
+streamlit>=1.25.0
+streamlit_chat>=0.0.2
+openai>=0.27.0
+python-dotenv>=0.21.0
+helicone>=0.1.0