Update python-chatGPT course

Signed-off-by: Manuel Vergara <manuel@vergaracarmona.es>

python-chatgpt/src/08_filtrar_respuestas.py

@@ -0,0 +1,100 @@
+"""
+Chatbot con OpenAI GPT-3
+"""
+
+import openai
+import os
+import spacy
+from dotenv import load_dotenv
+from colorama import init, Fore
+
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
+
+openai.api_key = api_key
+
+preguntas_anteriores = []
+respuestas_anteriores = []
+modelo_spacy = spacy.load("es_core_news_md")
+palabras_prohibidas = ["palabra1", "palabra2", "paella"]
+
+
+# Inicializar colorama
+init()
+
+
+def clearConsole():
+    # Función limpiar consola
+    os.system('clear')
+
+
+def filtrar_lista_negra(texto, lista_negra):
+    """
+    Filtra palabras de una lista negra
+    """
+
+    # Separar el texto en tokens
+    token = modelo_spacy(texto)
+    # Crear una lista vacía para guardar los tokens permitidos
+    resultado = []
+
+    for t in token:
+        # Si el token no está en la lista negra, agregarlo al resultado
+        if t.text not in lista_negra:
+            resultado.append(t.text)
+
+        else:
+            # Si el token está en la lista negra, agregarlo al resultado
+            resultado.append("[xxxxx]")
+
+    return " ".join(resultado)
+
+
+def preguntar_chat_gpt(prompt, modelo="text-davinci-002"):
+    """
+    Pregunta a la API de OpenAI GPT-3
+    """
+
+    respuesta = openai.Completion.create(
+        engine=modelo,
+        prompt=prompt,
+        n=1,
+        temperature=1,
+        max_tokens=150
+    )
+
+    respuesta_sin_filtrar = respuesta.choices[0].text.strip()
+
+    respuesta_filtrada = filtrar_lista_negra(
+        respuesta_sin_filtrar, palabras_prohibidas)
+
+    return respuesta_filtrada
+
+
+# Bienvenida
+clearConsole()
+print(Fore.RED + "Bienvenido al chatbot de OpenAI GPT-3." + Fore.RESET)
+print(Fore.RED + "Escribe \"salir\" cuando quieras terminar la conversación." + Fore.RESET)
+
+# Loop para controlar el flujo de la conversación
+while True:
+
+    conversacion_historica = ""
+
+    ingreso_usuario = input(Fore.MAGENTA + "Tú: " + Fore.RESET)
+
+    if ingreso_usuario == "salir":
+        break
+
+    for pregunta, respuesta in zip(preguntas_anteriores, respuestas_anteriores):
+        conversacion_historica += f"{Fore.BLUE}Usuario pregunta: {Fore.RESET}{pregunta}"
+        conversacion_historica += f"{Fore.GREEN}Bot responde: {Fore.RESET}{respuesta}\n"
+
+    prompt = f"{Fore.CYAN}Usuario pregunta: {Fore.RESET}{ingreso_usuario}"
+    conversacion_historica += prompt
+    respuesta_gpt = preguntar_chat_gpt(conversacion_historica)
+
+    print(f"{respuesta_gpt}")
+
+    preguntas_anteriores.append(ingreso_usuario)
+    respuestas_anteriores.append(respuesta_gpt)