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"]
 
@@ -10,12 +11,16 @@ to generate topic summary ideas for social media videos. Follow these steps in t
 2. Generate 100 ideas for videos a real estate agent should make, and analyze them to choose the 10 most compelling. Do not return all 100 ideas.
 3. Return a list of these 10 most compelling ideas."""}]
 
+@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
@@ -36,4 +41,3 @@ demo = gradio.Interface(fn=wrapped_chat_gpt, inputs="text", outputs="text", titl
 demo.launch(inline=False)
 
 
-