Update flask/app.py

flask/app.py

@@ -7,15 +7,12 @@ import json
 import cv2
 import numpy as np
 from time import gmtime, strftime
-from flask import Flask, request, jsonify
+from pydantic import BaseModel
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from typing import Dict
 
-from engine.header import get_version
-from engine.header import get_deviceid
-from engine.header import init_sdk
-from engine.header import init_sdk_offline
-
-from engine.header import check_liveness
-from engine.header import print_log, print_error, print_info, print_warning
+from engine.header import *
 
 file_path = os.path.abspath(__file__)
 dir_path = os.path.dirname(file_path)
@@ -42,7 +39,6 @@ def activate_sdk():
     if online_key is None:
         print_warning("Liveness online license key not found!")
     else:
-        print_info(f"FL_LICENSE_KEY: {online_key}")
         ret = init_sdk(dict_path.encode('utf-8'), online_key.encode('utf-8'))
 
     if ret == 0:
@@ -89,16 +85,25 @@ def generate_response(result, face_rect, score, angles):
             "pitch": angles[2]
         }
 
-    response = jsonify(data)
-    response.status_code = 200
-    response.headers["Content-Type"] = "application/json; charset=utf-8"
-    return response
-    
-@app.route('/api/check_liveness', methods=['POST'])
-def check_liveness_api():
+    return JSONResponse(content=data, status_code=200)
+
+@app.get("/")
+def read_root():
+    return {"status": "API is running"}
+
+def read_image(file: UploadFile) -> np.ndarray:
+    # Read the image file and convert it to OpenCV format
+    image_bytes = file.file.read()
+    image_np = np.frombuffer(image_bytes, np.uint8)
+    image = cv2.imdecode(image_np, cv2.IMREAD_COLOR)
+    return image
+
+@app.post("/api/check_liveness")
+async def check_liveness_api(
+    image: UploadFile = File(...)
+) -> JSONResponse:
     try:
-        file = request.files['image']
-        image_mat = cv2.imdecode(np.frombuffer(file.read(), np.uint8), cv2.IMREAD_COLOR)
+        image_mat = read_image(image)
     except:
         response = generate_response("Failed to open file!", None, None, None)
         return response
@@ -107,12 +112,24 @@ def check_liveness_api():
     response = generate_response(result, face_rect, score, angles)
     return response
 
-@app.route('/api/check_liveness_base64', methods=['POST'])
-def check_liveness_base64_api():
+def decode_base64_image(base64_string: str) -> np.ndarray:
+    try:
+        image_data = base64.b64decode(base64_string)
+        image_np = np.frombuffer(image_data, np.uint8)
+        image = cv2.imdecode(image_np, cv2.IMREAD_COLOR)
+        if image is None:
+            raise ValueError("Decoded image is None")
+        return image
+    except Exception as e:
+        raise ValueError(f"Failed to decode base64 image: {str(e)}")
+
+class CheckLivenessRequest(BaseModel):
+    image: str
+
+@app.post("/api/check_liveness_base64")
+async def check_liveness_base64_api(request: CheckLivenessRequest) -> JSONResponse:
     try: 
-        content = request.get_json()
-        imageBase64 = content['image']
-        image_mat = cv2.imdecode(np.frombuffer(base64.b64decode(imageBase64), dtype=np.uint8), cv2.IMREAD_COLOR)
+        image_mat = decode_base64_image(request.image)
     except:
         response = generate_response("Failed to open file!", None, None, None)
         return response
@@ -125,5 +142,15 @@ if __name__ == '__main__':
     ret = activate_sdk()
     if ret != 0:
         exit(-1)
-    port = int(os.environ.get("PORT", 8000))
-    app.run(host='0.0.0.0', port=port)
+        
+    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=7860)