Update app.py

app.py

@@ -7,19 +7,23 @@ from datetime import datetime
 # Put your OpenAI API key here
 openai.api_key = os.getenv('OpenAPI')
 
-def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5):
-    try: 
-        response = openai.ChatCompletion.create(
-            model=engine,
-            messages=[{"role": "system", "content": "You are a creative writer who specializes in crafting engaging stories to sell products or services"},
-                  {"role": "user", "content": prompt}],
-            max_tokens=max_tokens,
-            n=1,
-            temperature=temperature
-        )
-        return response.choices[0].message['content'].strip()
-    except Exception as e:
-        return f"Error in gpt3_query: {str(e)}", None
+def gpt3_query(prompt, engine='gpt-3.5-turbo-16k', max_tokens=15050, temperature=0.5, retries=3):
+    for i in range(retries):
+        try: 
+            response = openai.ChatCompletion.create(
+                model=engine,
+                messages=[{"role": "system", "content": "You are a creative writer who specializes in crafting engaging stories to sell products or services"},
+                      {"role": "user", "content": prompt}],
+                max_tokens=max_tokens,
+                n=1,
+                temperature=temperature
+            )
+            return response.choices[0].message['content'].strip()
+        except Exception as e:
+            if i < retries - 1:  # i is zero indexed
+                continue
+            else:
+                return f"Error in gpt3_query after {retries} attempts: {str(e)}", None
 
 def read_file(file_path):
     with open(file_path, 'r') as file: