Update app.py

app.py

@@ -14,6 +14,7 @@ os.system("wget https://i.imgur.com/fSL1CGG.jpg")
 os.environ["PYTHONPATH"]=os.environ["PYTHONPATH"]+":"+f"{os.getcwd()}/src/text_recognition/deep-text-recognition-benchmark"
 
 import gradio as gr
+from PIL import Image
 from uuid import uuid4
 from pathlib import Path
 
@@ -23,8 +24,17 @@ def inference(im):
     Path(f'{dir_name}/img').mkdir(parents=True)
     im.save(f'{dir_name}/img/image.jpg')
     os.system(f'python main.py infer {dir_name}/img/image.jpg {dir_name}_output -s f -i')
+    image_path = f'{dir_name}_output/image/pred_img/image_L.jpg'
+    if Path(f'{dir_name}_output/image/pred_img/image_R.jpg').exists():
+        image_L = Image.open(f'{dir_name}_output/image/pred_img/image_L.jpg')
+        image_R = Image.open(f'{dir_name}_output/image/pred_img/image_R.jpg')
+        dst = Image.new('RGB', (image_L.width + image_R.width, image_L.height))
+        dst.paste(image_L, (0, 0))
+        dst.paste(image_R, (image_L.width, 0))
+        dst.save(f'{dir_name}_output/image/pred_img/image_LR.jpg')
+        image_path = f'{dir_name}_output/image/pred_img/image_LR.jpg'
     with open(f'{dir_name}_output/image/txt/image_main.txt') as f:
-        return f'{dir_name}_output/image/pred_img/image_L.jpg', f.read()
+        return image_path, f.read()
 
 title = "NDLOCR"
 description = "Gradio demo for NDLOCR. NDLOCR is a text recognition (OCR) Program."