Add error changes

app.py

@@ -48,13 +48,14 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
 
     if progress == "None":
         for x in content:
-            retries = 1
-            timeout = 1
+            retries = 3
+            timeout = 3
             i = 0
             while True: 
                 try: 
                     data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
                     combined_data.append(data)
+                    i = 0
                     break
                 except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
                     logging.error(f'Retry {i+1} failed: {e}')
@@ -76,6 +77,7 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
                 try: 
                     data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
                     combined_data.append(data)
+                    i = 0
                     break
                 except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
                     logging.error(f'Retry {i+1} failed: {e}')
@@ -92,10 +94,42 @@ def batch_document(content, document_type, context, progress = gr.Progress()):
     if document_type == "":
         document_type = "error"
     return save_json(combined_data, document_type)
+
 def image(content, context):
-    return chat_gpt_image(detect_image(content), context)
+    retries = 3
+    timeout = 3
+    i = 0
+    while True: 
+        try: 
+            data = chat_gpt_image(detect_image(content), context)
+            break
+        except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
+            logging.error(f'Retry {i+1} failed: {e}')
+            if i < retries - 1:
+                logging.error(f'Retrying in {timeout} seconds...')
+                time.sleep(timeout)
+                i += 1
+            else:
+                break
+    return data
+
 def document(content, document_type, context):
-    return chat_gpt_document(detect_document(content),document_type,context)
+    retries = 3
+    timeout = 3
+    i = 0
+    while True: 
+        try: 
+            data = chat_gpt_document(detect_document(content),document_type,context)
+            break
+        except (openai.error.APIConnectionError, openai.error.AuthenticationError, openai.error.RateLimitError, HttpResponseError, requests.exceptions.RequestException) as e:
+            logging.error(f'Retry {i+1} failed: {e}')
+            if i < retries - 1:
+                logging.error(f'Retrying in {timeout} seconds...')
+                time.sleep(timeout)
+                i += 1
+            else:
+                break
+    return data
 
 with gr.Blocks(title="Axon OCR", css=".markdown {text-align: center;}") as app:
     gr.Markdown("""# Axon OCR