Update python-chatGPT course

Signed-off-by: Manuel Vergara <manuel@vergaracarmona.es>
This commit is contained in:
2023-06-10 17:48:19 +02:00
parent bd11fc17e2
commit c3fdcdec37
8 changed files with 236 additions and 4 deletions

View File

@@ -0,0 +1,47 @@
"""
Programa para crear contenido
y resumirlo con la API de OpenAI
"""
import openai
import os
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("OPENAI_API_KEY")
openai.api_key = api_key
def crear_contenido(tema, tokens, temperatura, modelo="text-davinci-002"):
"""
Crea contenido con OpenAI GPT-3
"""
prompt = f"Escribe un artículo corto sobre el tema: {tema}"
respuesta = openai.Completion.create(
engine=modelo,
prompt=prompt,
n=1,
temperature=temperatura,
max_tokens=tokens
)
return respuesta.choices[0].text.strip()
# Bienvenida
print("Bienvenido a la aplicación de creación de contenido. \n Necesito que me des algunos datos.")
# Pedir datos
tema = input("Elige un tema para tu artículo: ")
tokens = int(input("Tokens máximos: "))
temperatura = int(
input("Del 1 al 10, ¿Cuánto quieres que sea de creativo el artículo?: ")) / 10
# Crear contenido
articulo_creado = crear_contenido(tema, tokens, temperatura)
print(articulo_creado)