Update app.py

app.py

@@ -7,6 +7,9 @@ from PIL import Image
 import io
 
 
+# evalstate -- set remove paddle_ocr option, default to true.
+
+
 import base64, os
 from util.utils import check_ocr_box, get_yolo_model, get_caption_model_processor, get_som_labeled_img
 import torch
@@ -51,6 +54,10 @@ def process(
     use_paddleocr,
     imgsz
 ) -> Optional[Image.Image]:
+"""
+Parse a GUI screen and return an array of structured elements, and a marked-up image showing the bounding boxes. 
+Array contains:  Element Type (e.g. Icon, Text, Label), Bounding Box, Interactivity and Element Content
+"""
 
     # image_save_path = 'imgs/saved_image_demo.png'
     # image_input.save(image_save_path)
@@ -64,7 +71,7 @@ def process(
     }
     # import pdb; pdb.set_trace()
 
-    ocr_bbox_rslt, is_goal_filtered = check_ocr_box(image_input, display_img = False, output_bb_format='xyxy', goal_filtering=None, easyocr_args={'paragraph': False, 'text_threshold':0.9}, use_paddleocr=use_paddleocr)
+    ocr_bbox_rslt, is_goal_filtered = check_ocr_box(image_input, display_img = False, output_bb_format='xyxy', goal_filtering=None, easyocr_args={'paragraph': False, 'text_threshold':0.9}, use_paddleocr=True)
     text, ocr_bbox = ocr_bbox_rslt
     dino_labled_img, label_coordinates, parsed_content_list = get_som_labeled_img(image_input, yolo_model, BOX_TRESHOLD = box_threshold, output_coord_in_ratio=True, ocr_bbox=ocr_bbox,draw_bbox_config=draw_bbox_config, caption_model_processor=caption_model_processor, ocr_text=text,iou_threshold=iou_threshold, imgsz=imgsz,)  
     image = Image.open(io.BytesIO(base64.b64decode(dino_labled_img)))
@@ -84,9 +91,9 @@ with gr.Blocks() as demo:
                 label='Box Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.05)
             # set the threshold for removing the bounding boxes with large overlap, default is 0.1
             iou_threshold_component = gr.Slider(
-                label='IOU Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.1)
-            use_paddleocr_component = gr.Checkbox(
-                label='Use PaddleOCR', value=True)
+                label='IOU (Intersection over Union) Threshold', minimum=0.01, maximum=1.0, step=0.01, value=0.1)
+#            use_paddleocr_component = gr.Checkbox(
+#                label='Use PaddleOCR', value=True)
             imgsz_component = gr.Slider(
                 label='Icon Detect Image Size', minimum=640, maximum=1920, step=32, value=640)
             submit_button_component = gr.Button(