new

app.py

@@ -32,7 +32,7 @@ def chat_gpt_image(content, context):
     completion = openai.ChatCompletion.create(
     model=gpt_model,
     user="1",
-    temperature="0.5",
+    temperature="0.1",
     messages=[
         {"role": "system", "content": prompt},
         {"role": "user", "content": final_content}
@@ -191,33 +191,34 @@ def batch_document(content, document_type, context, progress=gr.Progress()):
     combined_data = []
     for x in progress.tqdm(content, desc="Processing"):
         while True: 
+            k = i+j
             try: 
                 data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
                 combined_data.append(data)
                 break
-            except Exception as e:
-                logging.error(f'Retry {i+1} failed: ', e)
+            except openai.error.APIConnectionError:
+                logging.error(f'Retry {k+1} failed: openai.error.APIConnectionError')
                 if i < retries - 1:
                     logging.error(f'Retrying in {timeout} seconds...')
                     time.sleep(timeout)
                     i += 1
-            # except openai.error.RateLimitError:
-            #     logging.error(f'Retry {j+1} failed: openai.error.RateLimitError')
-            #     if j < retries - 1:
-            #         logging.error(f'Retrying in {timeout} seconds...')
-            #         time.sleep(timeout)
-            #         j += 1
+            except openai.error.RateLimitError:
+                logging.error(f'Retry {k+1} failed: openai.error.RateLimitError')
+                if j < retries - 1:
+                    logging.error(f'Retrying in {timeout} seconds...')
+                    time.sleep(timeout)
+                    j += 1
     logging.info(combined_data)
     return save_json(combined_data, document_type)
 
 def save_json(text, filename):
     filename = filename+".json"
     with open(filename, "w", encoding='utf-8') as outfile:
-        json.dump(text, outfile)
+        json.dump(text, outfile, ensure_ascii=False)
     return filename
 
-with gr.Blocks(title="Ottico OCR", css=".markdown {text-align: center;}", theme='freddyaboulton/dracula_revamped') as app:
-    gr.Markdown("""# Ottico OCR
+with gr.Blocks(title="Axon OCR", css=".markdown {text-align: center;}", theme='freddyaboulton/dracula_revamped') as app:
+    gr.Markdown("""# Axon OCR
     Attach Images or Files below and convert them to Text.""", elem_classes="markdown")
     with gr.Tab("Scan Image"):
         with gr.Row():