Upload 21 files

.gitattributes

@@ -33,3 +33,13 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+dependency/libimutils.so filter=lfs diff=lfs merge=lfs -text
+dependency/libimutils.so_for_ubuntu22 filter=lfs diff=lfs merge=lfs -text
+dependency/libttvcore.so filter=lfs diff=lfs merge=lfs -text
+engine/libMetaChecker.so filter=lfs diff=lfs merge=lfs -text
+engine/libOCR.so filter=lfs diff=lfs merge=lfs -text
+examples/1_b.png filter=lfs diff=lfs merge=lfs -text
+examples/1_f.png filter=lfs diff=lfs merge=lfs -text
+examples/2_b.png filter=lfs diff=lfs merge=lfs -text
+examples/2_f.png filter=lfs diff=lfs merge=lfs -text
+examples/4.png filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,22 @@
+FROM ubuntu:22.04
+RUN ln -snf /usr/share/zoneinfo/$CONTAINER_TIMEZONE /etc/localtime && echo $CONTAINER_TIMEZONE > /etc/timezone
+RUN apt-get update && \
+    apt-get install -y binutils python3 python3-pip python3-opencv libcurl4-openssl-dev libssl-dev libpcsclite-dev psmisc && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+COPY --chown=user . .
+COPY --chown=user ./dependency/libimutils.so_for_ubuntu22 /usr/lib/libimutils.so
+COPY --chown=user ./dependency/libttvcore.so /usr/lib
+
+RUN pip3 install gradio==5.23.2 opencv-python numpy fastapi
+RUN chmod a+x run_demo.sh
+
+EXPOSE 9000 7860
+
+ENTRYPOINT ["./run_demo.sh"]
+

app.py

@@ -0,0 +1,212 @@
+import sys
+sys.path.append('../')
+
+import os
+import base64
+import json
+import uuid
+import cv2
+import numpy as np
+import gradio as gr
+from time import gmtime, strftime
+from pydantic import BaseModel
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from typing import Dict
+
+from engine.header import *
+
+file_path = os.path.abspath(__file__)
+dir_path = os.path.dirname(file_path)
+root_path = os.path.dirname(dir_path)
+dump_path = os.path.join(root_path, "dump2/")
+
+device_id = get_deviceid().decode('utf-8')
+print_info('\t <Hardware ID> \t\t {}'.format(device_id))
+
+def activate_sdk():
+	online_key = os.environ.get("LICENSE_KEY")
+	offline_key_path = os.path.join(root_path, "license.txt")
+	
+	dict_path = os.path.join(root_path, "engine/bin")
+
+	ret = -1
+	if online_key is None:
+		print_warning("Online license key not found!")
+	else:
+		activate_ret = set_activation(online_key.encode('utf-8')).decode('utf-8')
+		ret = json.loads(activate_ret).get("errorCode", None)
+
+	if ret == 0:
+		print_log("Successfully online activation SDK!")
+	else:
+		print_error(f"Failed to online activation SDK, Error code {ret}\n Trying offline activation SDK...");
+		if os.path.exists(offline_key_path) is False:
+			print_warning("Offline license key file not found!")
+			print_error(f"Falied to offline activation SDK, Error code {ret}")
+			return ret
+		else:
+			file=open(offline_key_path,"r")
+			offline_key = file.read()
+			file.close()
+			activate_ret = set_activation(offline_key.encode('utf-8')).decode('utf-8')
+			ret = json.loads(activate_ret).get("errorCode", None)
+			if ret == 0:
+				print_log("Successfully offline activation SDK!")
+			else:
+				print_error(f"Falied to offline activation SDK, Error code {ret}")
+				return ret
+	
+	init_ret = init_sdk(dict_path.encode('utf-8')).decode('utf-8')
+	ret = json.loads(activate_ret).get("errorCode", None)
+	print_log(f"Init SDK: {ret}")
+	return ret
+
+
+async def save_upload_file(upload_file: UploadFile) -> str:
+    file_name = uuid.uuid4().hex[:6] + "_" + upload_file.filename
+    save_path = os.path.join(dump_path, file_name)
+    with open(save_path, "wb") as buffer:
+        shutil.copyfileobj(upload_file.file, buffer)
+    return os.path.abspath(save_path)
+
+async def save_base64_file(base64_string: str) -> str:
+    file_name = uuid.uuid4().hex[:6] + ".jpg"  # or ".png" depending on your use
+    save_path = os.path.join(dump_path, file_name)
+    
+    with open(save_path, "wb") as buffer:
+        buffer.write(base64.b64decode(base64_string))
+    
+    return os.path.abspath(save_path)
+				
+app = FastAPI()
+
+class ImageBase64Request(BaseModel):
+    image: str
+
+@app.get("/")
+def read_root():
+	return {"status": "API is running"}
+
+@app.post("/api/read_idcard")
+async def read_idcard(image: UploadFile = File(...), image2: UploadFile = File(None)):
+	try:
+        file_path1 = await save_upload_file(image)
+
+        file_path2 = ""
+        if image2 is not None:
+            file_path2 = await save_upload_file(image2)
+
+
+		ocrResult = ocr_id_card(file_path1.encode('utf-8'), file_path2.encode('utf-8'))
+		ocrResDict = json.loads(ocrResult)  
+		
+		os.remove(file_path1)
+        if file_path2:
+            os.remove(file_path2)
+
+		return JSONResponse(content={"status": "ok", "data": ocrResDict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+
+@app.post("/api/read_idcard_base64")
+async def read_idcard_base64(request_data: ImageBase64Request):
+    try:
+        file_path = await save_base64_file(request_data.image)
+
+        # Your OCR function (pass bytes-encoded file path)
+        ocr_result = ocr_id_card(file_path.encode('utf-8'), ''.encode('utf-8'))
+        ocr_res_dict = json.loads(ocr_result)
+
+        # Remove the temp file
+        os.remove(file_path)
+
+        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+@app.post("/api/read_credit")
+async def read_credit(image: UploadFile = File(...)):
+	try:
+        file_path = await save_upload_file(image)
+
+		ocrResult = ocr_credit_card(file_path.encode('utf-8'))
+		ocrResDict = json.loads(ocrResult)  
+		
+		os.remove(file_path)
+
+		return JSONResponse(content={"status": "ok", "data": ocrResDict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+@app.post("/api/read_credit_base64")
+async def read_credit_base64(request_data: ImageBase64Request):
+    try:
+        file_path = await save_base64_file(request_data.image)
+
+        # Your OCR function (pass bytes-encoded file path)
+        ocr_result = ocr_credit_card(file_path.encode('utf-8'))
+        ocr_res_dict = json.loads(ocr_result)
+
+        # Remove the temp file
+        os.remove(file_path)
+
+        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+
+@app.post("/api/read_barcode")
+async def read_barcode(image: UploadFile = File(...)):
+	try:
+        file_path = await save_upload_file(image)
+
+		ocrResult = ocr_barcode(file_path.encode('utf-8'))
+		ocrResDict = json.loads(ocrResult)  
+		
+		os.remove(file_path)
+
+		return JSONResponse(content={"status": "ok", "data": ocrResDict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+@app.post("/api/read_barcode_base64")
+async def read_barcode_base64(request_data: ImageBase64Request):
+    try:
+        file_path = await save_base64_file(request_data.image)
+
+        # Your OCR function (pass bytes-encoded file path)
+        ocr_result = ocr_barcode(file_path.encode('utf-8'))
+        ocr_res_dict = json.loads(ocr_result)
+
+        # Remove the temp file
+        os.remove(file_path)
+
+        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+
+    except Exception as e:
+        return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
+
+
+if __name__ == '__main__':
+	ret = activate_sdk()
+	if ret != 0:
+		exit(-1)
+
+    dummy_interface = gr.Interface(
+        fn=lambda x: "API ready.",
+        inputs=gr.Textbox(label="Info"),
+        outputs=gr.Textbox(label="Response"),
+        allow_flagging="never"  # 🚫 disables writing to `flagged/`
+    )
+    
+    gr_app = gr.mount_gradio_app(app, dummy_interface, path="/gradio")
+    
+    import uvicorn
+    uvicorn.run(gr_app, host="0.0.0.0", port=9000)

engine/header.py

@@ -0,0 +1,47 @@
+import os
+import sys
+import numpy as np
+import ctypes, ctypes.util
+from enum import Enum
+from ctypes import *
+from numpy.ctypeslib import ndpointer
+
+def print_log(fmt): print("[LOG] \033[98m{}\033[00m" .format(fmt))
+def print_info(fmt): print("[INFO] \033[92m{}\033[00m" .format(fmt))
+def print_error(fmt): print("[ERR] \033[91m{}\033[00m" .format(fmt)) 
+def print_warning(fmt): print("[WARNING] \033[93m{}\033[00m" .format(fmt))
+
+
+ocr_path = os.path.abspath(os.path.dirname(__file__)) + '/libOCR.so'
+ocr_engine = cdll.LoadLibrary(ocr_path)
+
+get_deviceid = ocr_engine.TTVOcrGetHWID
+get_deviceid.argtypes = []
+get_deviceid.restype = ctypes.c_char_p
+
+set_activation = ocr_engine.TTVOcrSetActivation
+set_activation.argtypes = []
+set_activation.restype = ctypes.c_char_p
+
+init_sdk = ocr_engine.TTVOcrInit
+init_sdk.argtypes = [ctypes.c_char_p]
+init_sdk.restype = ctypes.c_char_p
+
+ocr_id_card = ocr_engine.TTVOcrProcess
+ocr_id_card.argtypes = [ctypes.c_char_p, ctypes.c_char_p]
+ocr_id_card.restype = ctypes.c_char_p
+
+ocr_credit_card = ocr_engine.TTVOcrCreditCard
+ocr_credit_card.argtypes = [ctypes.c_char_p]
+ocr_credit_card.restype = ctypes.c_char_p
+
+ocr_barcode = ocr_engine.TTVOcrBarCode
+ocr_barcode.argtypes = [ctypes.c_char_p]
+ocr_barcode.restype = ctypes.c_char_p
+
+metacheck_path = os.path.abspath(os.path.dirname(__file__)) + '/libMetaChecker.so'
+meta_engine = cdll.LoadLibrary(metacheck_path)
+
+check_meta = meta_engine.ttv_if_checker
+check_meta.argtypes = [ctypes.c_char_p]
+check_meta.restype = ctypes.c_int32

run_demo.sh

@@ -0,0 +1,10 @@
+#!/bin/sh
+
+# Add the current directory to PYTHONPATH
+export PYTHONPATH=$(dirname "$(pwd)"):$PYTHONPATH
+
+# Set LD_LIBRARY_PATH
+# export LD_LIBRARY_PATH="/usr/lib/openvino:$LD_LIBRARY_PATH"
+
+python3 app.py
+