Update app.py

app.py

@@ -3,6 +3,12 @@ import pulp
 import pandas as pd
 import openpyxl
 import requests
+import os
+os.system('pip install paddlepaddle')
+os.system('pip install paddleocr')
+from paddleocr import PaddleOCR, draw_ocr
+from PIL import Image
+import torch
 
 w_style="""
 <style>
@@ -44,6 +50,20 @@ def fx4(x):
 def fx5():
     return html_code
     
+def fx6(img, lang):
+    ocr = PaddleOCR(use_angle_cls=True, lang=lang,use_gpu=False)
+    img_path = img.name
+    result = ocr.ocr(img_path, cls=True)
+    image = Image.open(img_path).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='simfang.ttf')
+    im_show = Image.fromarray(im_show)
+    im_show.save('result.jpg')
+    return 'result.jpg'
+    
     
 def fx7(input_path,input_path2):
   input_img = load_img(input_path)
@@ -85,6 +105,10 @@ with demo:
             qy_input=[gr.Image(label="待转换图片",type="filepath"),gr.Image(label="风格图片",type="filepath")]
             qy_output=gr.Image(type="auto", label="输出")
             qy_button = gr.Button("开始迁移>>")
+        with gr.TabItem("OCR识别"):
+            ocr_input=[gr.Image(type='file', label='请提供你需要识别的照片'),gr.Dropdown(choices=['ch', 'en', 'fr', 'german', 'korean', 'japan'], type="value", default='ch', label='请选择需要识别的语言')]
+            ocr_output=gr.Image(type='file', label='识别结果')
+            ocr_button = gr.Button("开始ocr识别")
             
     tj_button.click(fx1, inputs=text_input, outputs=text_output)
     cs_button.click(fx2, inputs=val_input, outputs=json_output,api_name="ghqj")
@@ -92,5 +116,6 @@ with demo:
     tts_button.click(fx4, inputs=TS_input, outputs=audio_output)
     pva_button.click(fx5, inputs=[],outputs=pva_output)
     qy_button.click(fx7,inputs=qy_input,outputs=qy_output)
+    ocr_button.click(fx6,inputs=ocr_input,outputs=ocr_output,enable_queue=True)
 
 demo.launch()