Update app.py

app.py

@@ -1,5 +1,5 @@
-import cv2
 import os
+import cv2
 import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 import json
@@ -114,22 +114,19 @@ 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')
+    json_results_path = os.path.join("/app", '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')
+    field_value_pairs_path = os.path.join("/app", '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_path, field_value_pairs_path
 
-# Define Gradio interface
-import gradio as gr
-
 # Define Gradio interface
 iface = gr.Interface(
     fn=process_image,
@@ -143,4 +140,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()