Update app.py

app.py

@@ -1,6 +1,7 @@
 import openai
 import gradio
 import os
+from tenacity import retry, wait_fixed, stop_after_attempt
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
@@ -18,12 +19,16 @@ of the viewer and summarize what they can expect to hear in the rest of the scri
 4. Choose the best hook and replace the beginning of the script draft with that hook.
 5. Return just the words the agent should read into the camera."""}]
 
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
+def call_openai_api(messages):
+    return openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages
+    )
+
 def CustomChatGPT(user_input, messages):
     messages.append({"role": "user", "content": user_input})
-    response = openai.ChatCompletion.create(
-        model = "gpt-3.5-turbo",
-        messages = messages
-    )
+    response = call_openai_api(messages)
     ChatGPT_reply = response["choices"][0]["message"]["content"]
     messages.append({"role": "assistant", "content": ChatGPT_reply})
     return ChatGPT_reply, messages