Update main.py

main.py

@@ -12,6 +12,8 @@ ocr_ru = PaddleOCR(use_angle_cls=True, lang='cyrillic')
 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_fa = PaddleOCR(user_angle_cls=True, lang="fa")
+
 #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,8 +25,10 @@ def perform_ocr(img, lang):
         ocr = ocr_ch
     elif lang == "Russian" or lang == "Ukrainian":
         ocr = ocr_ru
-    elif lang == "Arabic" or lang == "Persian":
+    elif lang == "Arabic":
         ocr = ocr_arabic
+    elif lang == "Persian":
+        ocr = ocr_fa
     elif lang == "Only Arabic Lang":
         ocr = ocr_ar
     elif lang == "Korean":
@@ -39,7 +43,7 @@ def perform_ocr(img, lang):
     scores = [line[1][1] for line in result]
     # im_show = draw_ocr(image, boxes, txts, scores, font_path='fonts/simfang.ttf')
 
-    return [img, result]
+    return [img, txts, result]
 
 demo = gr.Blocks()
 
@@ -52,8 +56,9 @@ with demo:
             input_button = gr.Button("Run!")
         with gr.Column():
             output_image = gr.Image()
+            output_texts = gr.Textbox(label="Texts")
             output_text = gr.Textbox(label="Results")
 
-    input_button.click(fn=perform_ocr, inputs=[input_image, input_radio], outputs=[output_image, output_text])
+    input_button.click(fn=perform_ocr, inputs=[input_image, input_radio], outputs=[output_image, output_texts, output_text])
 
 demo.launch(server_name="0.0.0.0", server_port=7860)