Upload 2 files

app.py

@@ -3,8 +3,6 @@ import openai
 import os
 import gradio as gr
 
-openai.api_key = os.environ.get("OPENAI_API_KEY")
-
 # Generate the app using gradio, running dramatron which is in utils.py and has the outputs titulo, prem_ep, outline, chars, scenes, all_esc
 
 demo = gr.Interface(

utils.py

@@ -4,10 +4,15 @@ from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.pdfbase.pdfmetrics import stringWidth
 import backoff
+import os
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+client = openai.Client(api_key=os.environ.get("OPENAI_API_KEY"))
+
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_premisa_serie():
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un guionista de comedia de televisión chistoso, creativo y original. Eres especialista en sarcasmo. Sigue el formato Título: \n Premisa:"},
@@ -22,9 +27,11 @@ def generate_premisa_serie():
     premisa = output.split("Premisa:")[1]
     return titulo, premisa
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_premisa_episodio(prem_serie, keywords):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres el comediante Bill Burr, un guionista de comedia de televisión chistoso, creativo y original. Eres especialista en sit-coms de sarcasmo. Te dan una premisa de una serie y palabras clave para que generes la premisa de un episodio. Sigue el formato Título: \n Premisa:"},
@@ -39,9 +46,11 @@ def generate_premisa_episodio(prem_serie, keywords):
     premisa = output.split("Premisa:")[1]
     return titulo, premisa
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_outline(premisa):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un guionista de comedia de televisión especializado en crear conflictos interesantes y comedia de situación. No uses estilos como negritas, itálicas, subrayado. El usuario te dará una premisa y tú te encargas de desarrollarla con la siguiente estructura: 1. Incidente incitante:... \n2. Desarrollo del problema:... \n3. Clímax:... \n4. Resolución:..."},
@@ -60,9 +69,11 @@ def separar_escenas(outline):
 
     return outline_list
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_characters(outline):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor de una comedia situacional. Desarrolla en un enunciado a cada uno de los personajes dado un outline. El usuario te va a dar el outline al que te tienes que apegar. Usa un salto de línea para cada personaje."},
@@ -72,9 +83,11 @@ def generate_characters(outline):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_scenes(outline):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor de una comedia situacional. Enlista las escenas que formaran el episodio a partir de un outline. El usuario te va a dar el outline al que te tienes que apegar así como sus reglas de conteo de escenas. Usa un salto de línea para cada escena, no uses negritas y usa un renglón por escena separando por dos puntos el nombre de la escena y la descripción."},
@@ -84,21 +97,11 @@ def generate_scenes(outline):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
-def generate_dialogues_all(chars,scenes):
-    response = openai.ChatCompletion.create(
-        model="gpt-4-1106-preview",
-        messages=[
-            {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el guión de manera extensa con debates irónicos y sarcásticos para las escenas de un episodio. El guión debe durar 7 (siete) cuartillas. El usuario te da las escenas que tienes que desarrollar y las características de los personajes. Usa el formato de guión de Final Draft."},
-            {"role": "user", "content": scenes + "\n Los personajes son los siguientes:\n"+chars}
-        ],
-        temperature=0.3,
-    )
-    return response.choices[0].message.content
-
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues1(chars,scene1):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo de manera concisa, realista, irónico y sarcástico para la escena de un episodio. El usuario te da la escena que tienes que desarrollar y las características de los personajes. Usa el formato de Final Draft."},
@@ -108,9 +111,11 @@ def generate_dialogues1(chars,scene1):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues2(chars,scene1,dial1,scene2):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo de manera concisa, realista, irónico y sarcástico para una escena de un episodio. El usuario te da la escena que tienes que desarrollar. Es muy importante que tomes en cuenta las escenas anteriores. Usa el formato de Final Draft. Los personajes son los siguientes:"+chars},
@@ -122,9 +127,11 @@ def generate_dialogues2(chars,scene1,dial1,scene2):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues3(chars,scene1,dial1,scene2,dial2,scene3):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo de manera concisa, realista, irónico y sarcástico para una escena de un episodio. El usuario te da la escena que tienes que desarrollar. Es muy importante que tomes en cuenta las escenas anteriores. Usa el formato de Final Draft. Los personajes son los siguientes:"+chars},
@@ -138,9 +145,11 @@ def generate_dialogues3(chars,scene1,dial1,scene2,dial2,scene3):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues4(chars,scene1,dial1,scene2,dial2,scene3,dial3,scene4):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo de manera concisa, realista, irónico y sarcástico para una escena de un episodio. El usuario te da la escena que tienes que desarrollar. Es muy importante que tomes en cuenta las escenas anteriores. Usa el formato de Final Draft. Los personajes son los siguientes:"+chars},
@@ -156,9 +165,11 @@ def generate_dialogues4(chars,scene1,dial1,scene2,dial2,scene3,dial3,scene4):
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues5(chars,scene1,dial1,scene2,dial2,scene3,dial3,scene4,dial4,scene5):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo satírico, realista, irónico y sarcástico para una escena de un episodio. El usuario te da la escena que tienes que desarrollar y las instrucciones. Es muy importante que tomes en cuenta las escenas anteriores. Usa el formato de Final Draft. Los personajes son los siguientes:"+chars},
@@ -176,9 +187,11 @@ def generate_dialogues5(chars,scene1,dial1,scene2,dial2,scene3,dial3,scene4,dial
     )
     return response.choices[0].message.content
 
-@backoff.on_exception(backoff.expo, openai.error.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.RateLimitError)
+@backoff.on_exception(backoff.expo, openai.APIConnectionError)
+@backoff.on_exception(backoff.expo, openai.APIError)
 def generate_dialogues6(chars,scene1,dial1,scene2,dial2,scene3,dial3,scene4,dial4,scene5,dial5,scene6):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-4-1106-preview",
         messages=[
             {"role": "system", "content": "Eres un escritor sarcástico de comedia situacional con humor negro. Escribe el diálogo de manera concisa, realista, irónico y sarcástico para una escena de un episodio. El usuario te da la escena que tienes que desarrollar y las instrucciones. Es muy importante que tomes en cuenta las escenas anteriores. Usa el formato de Final Draft. Los personajes son los siguientes:"+chars},