andreinigo commited on
Commit
9de4270
1 Parent(s): 7f0fe5f

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py CHANGED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ import gradio as gr
3
+ import os
4
+
5
+ openai.api_key = os.environ['API_KEY']
6
+
7
+ def generate_outline(premisa):
8
+ response = openai.ChatCompletion.create(
9
+ model="gpt-4",
10
+ messages=[
11
+ {"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"},
12
+ {"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}
13
+ ],
14
+ temperature=0.4,
15
+ )
16
+ return response.choices[0].message.content
17
+
18
+
19
+ demo = gr.Interface(
20
+ fn=generate_outline,
21
+ # input
22
+ inputs=gr.Textbox(label="Premisa"),
23
+ # output
24
+ outputs=[
25
+ # title
26
+ gr.Textbox(label="Outline")
27
+ ],
28
+ title="Dramamex Outlines",
29
+ description="Escribe la premisa para el outline de tu historia.")
30
+
31
+
32
+ if __name__ == "__main__":
33
+ demo.launch()