Update app.py

app.py

@@ -63,25 +63,25 @@ def activate_sdk():
 
 
 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)
+	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)
+	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
+	image: str
 
 @app.get("/")
 def read_root():
@@ -90,47 +90,47 @@ def read_root():
 @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_path1 = await save_upload_file(image)
 
-        file_path2 = ""
-        if image2 is not None:
-            file_path2 = await save_upload_file(image2)
+		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)
+		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)
+	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)
+	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)
+		# 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)
+		# Remove the temp file
+		os.remove(file_path)
 
-        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+		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)
+	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)
+		file_path = await save_upload_file(image)
 
 		ocrResult = ocr_credit_card(file_path.encode('utf-8'))
 		ocrResDict = json.loads(ocrResult)  
@@ -139,31 +139,31 @@ async def read_credit(image: UploadFile = File(...)):
 
 		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)
+	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)
+	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)
+		# 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)
+		# Remove the temp file
+		os.remove(file_path)
 
-        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+		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)
+	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)
+		file_path = await save_upload_file(image)
 
 		ocrResult = ocr_barcode(file_path.encode('utf-8'))
 		ocrResDict = json.loads(ocrResult)  
@@ -172,25 +172,25 @@ async def read_barcode(image: UploadFile = File(...)):
 
 		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)
+	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)
+	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)
+		# 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)
+		# Remove the temp file
+		os.remove(file_path)
 
-        return JSONResponse(content={"status": "ok", "data": ocr_res_dict}, status_code=200)
+		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)
+	except Exception as e:
+		return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
 
 
 if __name__ == '__main__':
@@ -198,14 +198,14 @@ if __name__ == '__main__':
 	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)
+	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)