Inicio de Repositorio ChileanGPT

.gitignore

@@ -0,0 +1,4 @@
+# .gitignore
+
+# Ignora el archivo .env
+.env

README.md

@@ -1,8 +1,8 @@
 ---
 title: ChileanGPT
-emoji: 🌖
-colorFrom: pink
-colorTo: green
+emoji: 🤙
+colorFrom: red
+colorTo: blue
 sdk: streamlit
 sdk_version: 1.19.0
 app_file: app.py

app.py

@@ -0,0 +1,42 @@
+import streamlit as st
+from streamlit_chat import message
+from 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.title("Conversa con un Chileno")
+
+if 'generated' not in st.session_state:
+    st.session_state['generated'] = []
+if 'past' not in st.session_state:
+    st.session_state['past'] = []
+
+query = st.text_input("Di algo:", key="input")
+
+if 'messages' not in st.session_state:
+    st.session_state['messages'] = get_initial_message()
+ 
+if query:
+    with st.spinner("pensando..."):
+        messages = st.session_state['messages']
+        messages = update_chat(messages, "user", query)
+        # st.write("Before  making the API call")
+        # st.write(messages)
+        response = get_chatgpt_response(messages,"gpt-3.5-turbo")
+        messages = update_chat(messages, "assistant", response)
+        st.session_state.generated.append(response)
+        st.session_state.past.append(query)
+
+        
+if st.session_state['generated']:
+
+    for i in range(len(st.session_state['generated'])-1, -1, -1):
+        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)

requirements.txt

@@ -0,0 +1,4 @@
+streamlit
+streamlit_chat
+openai
+python-dotenv

utils.py

@@ -0,0 +1,22 @@
+import openai
+
+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 escuchar musica urbana, en especial a Marcianeke y Pailita. Hablas como un flaite chileno bien tipico y exagerado, eres chistoso, informal y a veces un poco grosero. 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?"}
+        ]
+    return messages
+
+def get_chatgpt_response(messages, model="gpt-3.5-turbo"):
+    print("Pregunta: ", messages)
+    response = openai.ChatCompletion.create(
+    model=model,
+    messages=messages
+    )
+    print ("respuesta",response)
+    return  response['choices'][0]['message']['content']
+
+def update_chat(messages, role, content):
+    messages.append({"role": role, "content": content})
+    return messages