import openai
import gradio as gr
import os

openai.api_key = os.environ['API_KEY']

def generate_outline(premisa):
    print("Premisa:"+premisa)
    response = openai.ChatCompletion.create(
        model="gpt-4",
        messages=[
            {"role": "system", "content": "Eres un guionista de comedia de televisión especializado en crear conflictos interesantes y la comedia de situación. Separas cada episodio en 4 partes: 1. Incidente incitante \n2. Desarrollo del problema \n3. Clímax \n4. Resolución"},
            {"role": "user", "content": "Desarrolla creativamente las cuatro partes para una situación cómica de un episodio de una serie. La serie tiene la siguiente premisa:"+ premisa} 
        ],
        temperature=0.4,
    )
    print(response.choices[0].message.content)
    return response.choices[0].message.content


demo = gr.Interface(
    fn=generate_outline,
    # input
    inputs=gr.Textbox(label="Premisa"),
    # output
    outputs=[
        # title
        gr.Textbox(label="Outline")
    ],
    title="Dramamex Outlines",
    description="Escribe la premisa para el outline de tu historia.")


if __name__ == "__main__":
    demo.launch()