Update main.py

main.py

@@ -1,6 +1,7 @@
 import os
 from uuid import uuid4
 
+import pytesseract
 import torch
 from PIL import Image
 from controlnet_aux import HEDdetector
@@ -162,15 +163,30 @@ def visual_qa():
     result = vqa.inference(filepath, question=question)
     return result
 
-@app.route('/instruct-pix2pix', methods=['POST'])
-def InstructPix2Pix():
+# @app.route('/instruct-pix2pix', methods=['POST'])
+# def InstructPix2Pix():
+#     file = request.files['file']  # 获取上传的文件
+#     filename = str(uuid4()) + '.png'
+#     filepath = os.path.join('data', 'upload', filename)
+#     file.save(filepath)
+#     output_filename = str(uuid4()) + '.png'
+#     question = request.args.get('t')
+#     result = ip2p.inference(filepath, question, output_filename)
+#     return result
+
+@app.route('/ocr', methods=['POST'])
+def ocr():
     file = request.files['file']  # 获取上传的文件
+    lang = request.args.get('lang')
+    if lang is None:
+        lang = 'chi_sim+eng'
     filename = str(uuid4()) + '.png'
     filepath = os.path.join('data', 'upload', filename)
     file.save(filepath)
-    output_filename = str(uuid4()) + '.png'
-    question = request.args.get('t')
-    result = ip2p.inference(filepath, question, output_filename)
+    # output_filename = str(uuid4()) + '.png'
+
+    # pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract'
+    result = pytesseract.image_to_string(Image.open(filepath), lang=lang)
     return result
 
 if __name__ == '__main__':