You've already forked Curso-lenguaje-python
48 lines
1.0 KiB
Python
48 lines
1.0 KiB
Python
"""
|
|
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)
|