Update main.py

main.py

@@ -9,7 +9,8 @@ from PIL import Image
 ocr_en = PaddleOCR(use_angle_cls=True, lang="en")
 ocr_ch = PaddleOCR(use_angle_cls=True, lang='ch')
 ocr_ru = PaddleOCR(use_angle_cls=True, lang='cyrillic')
-ocr_ar = PaddleOCR(use_angle_cls=True, lang='arabic')
+ocr_arabic = PaddleOCR(use_angle_cls=True, lang='arabic')
+ocr_ar = PaddleOCR(use_angle_cls=True, lang='ar')
 ocr_korean = PaddleOCR(user_angle_cls=True, lang="korean")
 #ocr_ch = PaddleOCR(det_model_dir="models/det/ch/ch_PP-OCRv4_det_infer", rec_model_dir="models/rec/ch/ch_PP-OCRv4_rec_infer", cls_model_dir="models/cls/ch_ppocr_mobile_v2.0_cls_infer", rec_char_dict_path="models/dict/ppocr_keys_v1.txt", lang="ch")
 
@@ -23,6 +24,8 @@ def perform_ocr(img, lang):
     elif lang == "Russian" or lang == "Ukrainian":
         ocr = ocr_ru
     elif lang == "Arabic" or lang == "Persian":
+        ocr = ocr_arabic
+    elif lang == "Only Arabic Lang":
         ocr = ocr_ar
     elif lang == "Korean":
         ocr = ocr_korean
@@ -31,12 +34,12 @@ def perform_ocr(img, lang):
     final_result = ""
  
     image = Image.open(img).convert('RGB')
-    # boxes = [line[0] for line in result]
-    # txts = [line[1][0] for line in result]
-    # scores = [line[1][1] for line in result]
-    # im_show = draw_ocr(image, boxes, txts, scores, font_path='fonts/simfang.ttf')
-    # im_show = Image.fromarray(im_show)
-    return [image, result]
+    boxes = [line[0] for line in result]
+    txts = [line[1][0] for line in result]
+    scores = [line[1][1] for line in result]
+    im_show = draw_ocr(image, boxes, txts, scores, font_path='fonts/simfang.ttf')
+
+    return [im_show, result]
 
 demo = gr.Blocks()
 
@@ -45,7 +48,7 @@ with demo:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(source="upload", type="filepath")
-            input_radio = gr.Radio(["English", "Chinese (Simplified)", "Russian", "Ukrainian", "Arabic", "Persian"], label="Languages")
+            input_radio = gr.Radio(["English", "Chinese (Simplified)", "Russian", "Ukrainian", "Arabic", "Persian", "Only Arabic Lang"], label="Languages")
             input_button = gr.Button("Run!")
         with gr.Column():
             output_image = gr.Image()