Update app.py

app.py

@@ -113,20 +113,33 @@ def process_image(image):
     
     # Convert OCR results to JSON
     json_results = convert_to_json(ocr_results[0])
+    json_results_path = os.path.join(UPLOAD_DIR, 'ocr_results.json')
+    with open(json_results_path, "w") as f:
+        json.dump(json_results, f, indent=4)
     
     # Extract field-value pairs from the text
     text = " ".join([result[1][0] for result in ocr_results[0]])
     field_value_pairs = extract_field_value_pairs(text)
+    field_value_pairs_path = os.path.join(UPLOAD_DIR, 'extracted_fields.json')
+    with open(field_value_pairs_path, "w") as f:
+        json.dump(field_value_pairs, f, indent=4)
     
-    return image_with_boxes, json_results, field_value_pairs
+    return image_with_boxes, json_results_path, field_value_pairs_path
+
+# Define Gradio interface
+import gradio as gr
 
 # Define Gradio interface
 iface = gr.Interface(
     fn=process_image,
     inputs=gr.Image(type="pil"),
-    outputs=[gr.Image(type="pil"), gr.JSON(), gr.JSON()],
+    outputs=[
+        gr.Image(type="pil"),
+        gr.File(label="Download OCR Results"),
+        gr.File(label="Download Extracted Fields")
+    ],
     live=True
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()