JSON and Kill switch

app.py

@@ -262,6 +262,19 @@ def save_json(text, filename):
         json.dump(text, outfile, ensure_ascii=False)
     return filename
 
+def combine_json_files(json_files, progress=gr.Progress()):
+    combined_data = []
+    progress(0, desc="Starting")
+    for file in progress.tqdm(json_files, desc="Combining JSON Files"):
+        with open(file.name, 'r') as json_file:
+            data = json.load(json_file)
+            combined_data.extend(data)
+    # Convert the combined_data dict back to a JSON string
+    # You might want to save this to a file and return the file,
+    # or return the JSON string directly
+    logging.info("Combined JSON File: ", combined_data)
+    return save_json(combined_data, "Combined Json")
+
 with gr.Blocks(title="Axon OCR", css=".markdown {text-align: center;}") as app:
     gr.Markdown("""# Axon OCR
     Attach Images or Files below and convert them to Text.""", elem_classes="markdown")
@@ -289,13 +302,24 @@ with gr.Blocks(title="Axon OCR", css=".markdown {text-align: center;}") as app:
                     gr.Textbox(label="Any additional information? (Optional)", placeholder="This is document is an Official Reciept")]
             batch_document_output = gr.File(label="Result")
         batch_document_button = gr.Button("Scan")
-        retry_button = gr.Button("Retry Unprocessed Documents", label="Retry")
-         
-    
+        with gr.Row():
+            with gr.Column():
+                retry_button = gr.Button("Retry Unprocessed Documents", label="Retry")
+            with gr.Column():
+                stop_button = gr.Button("Stop Processing Document", label="Stop")
+    with gr.Tab("Combine JSON"):
+        with gr.Row():
+            with gr.Column():
+                json_files_input = gr.File(file_types=[".json"], file_count="multiple", label='Upload JSON files')
+                combined_json_output = gr.File(label="Result")
+                combine_button = gr.Button('Combine JSON files')
+
     image_button.click(image, inputs=image_input, outputs=image_output)
     document_button.click(document, inputs=document_input, outputs=document_output)
-    batch_document_button.click(batch_document, inputs=batch_document_input, outputs=batch_document_output)
+    batch_document_event = batch_document_button.click(batch_document, inputs=batch_document_input, outputs=batch_document_output)
     retry_button.click(retry_unprocessed_documents, outputs=batch_document_output)
+    stop_button.click(fn=None, inputs=None, outputs=None, cancels=[batch_document_event])
+    combine_button.click(combine_json_files, inputs=json_files_input, outputs=combined_json_output)
 
 app.queue()
 app.launch(auth=("username", "password"))