img binaries

main.py

@@ -3,6 +3,7 @@ import io
 import os
 from ctypes import c_int, pointer, string_at
 from datetime import datetime
+from typing import List
 
 import cv2
 import numpy as np
@@ -73,7 +74,7 @@ def load_dependencies():
 
 
 @app.post("/analyze/")
-async def analyze_image(image_paths: UploadFile):
+async def analyze_image(image_binaries: List[UploadFile]):
     progress = print
     global trocr_pipeline
     # モデルが読み込まれているか確認
@@ -85,7 +86,7 @@ async def analyze_image(image_paths: UploadFile):
         return
 
     all_results = []
-    num_total_files = len(image_paths)
+    num_total_files = len(image_binaries)
 
     progress(0, desc="テンプレート画像読み込み中...")
     suit_templates = load_suit_templates(SUIT_TEMPLATE_PATH)
@@ -99,7 +100,7 @@ async def analyze_image(image_paths: UploadFile):
         processed_files_info = []
         # image_objects = {}
 
-        for i, image_path in enumerate(image_paths):
+        for i, image_path in enumerate(image_binaries):
             progress(
                 (i + 1) / num_total_files * 0.15,
                 desc="ステージ1/3: 文字候補を検出中...",
@@ -112,11 +113,12 @@ async def analyze_image(image_paths: UploadFile):
 
             try:
                 # ファイルをバイナリモードで安全に読み込む
-                with open(image_path, "rb") as f:
-                    # バイトデータをNumPy配列に変換
-                    file_bytes = np.asarray(
-                        bytearray(f.read()), dtype=np.uint8
-                    )
+                file_bytes = np.asarray(bytearray(image_path))
+                # with open(image_path, "rb") as f:
+                #     # バイトデータをNumPy配列に変換
+                #     file_bytes = np.asarray(
+                #         bytearray(f.read()), dtype=np.uint8
+                #     )
                 # NumPy配列（メモリ上のデータ）から画像をデコード
                 image = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
 
@@ -221,7 +223,7 @@ async def analyze_image(image_paths: UploadFile):
         # else:
         #     export_update = gr.update(interactive=False)
         final_result = all_results[0]["hands"]
-        filenames = [os.path.basename(p) for p in image_paths]
+        filenames = [os.path.basename(p) for p in image_binaries]
         # dropdown_update = gr.update(
         #     choices=filenames, value=filenames[0], interactive=True, open=True
         # )