Update api_server.py

api_server.py

@@ -7,6 +7,7 @@ from pathlib import Path
 from ultralytics import YOLO
 import io
 import base64
+import uuid
 
 # Disable tensorflow warnings
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
@@ -65,6 +66,9 @@ def predict():
 
     user_id = request.args.get('user_id') 
     
+    # 生成一個唯一的 message_id
+    message_id = str(uuid.uuid4())
+    
     if 'image' not in request.files:
         # Handle if no file is selected
         return 'No file selected'
@@ -105,7 +109,7 @@ def predict():
     # 儲存辨識後的圖片到指定資料夾
     for result in results:
         # 保存圖片
-        result.save_crop(f"{YOLO_DIR}/{user_id}")
+        result.save_crop(f"{YOLO_DIR}/{message_id}")
         
         num_detections = len(result.boxes)  # Get the number of detections
         labels = result.boxes.cls  # Get predicted label IDs
@@ -114,12 +118,13 @@ def predict():
         encoded_images=[]
 
         for label_name in label_names:
-            output_file=f"{YOLO_DIR}/{user_id}/{label_name}/im.jpg.jpg"
+            output_file=f"{YOLO_DIR}/{message_id}/{label_name}/im.jpg.jpg"
             # 將圖片轉換為 base64 編碼
             encoded_images.append(image_to_base64(output_file))
             
         # 建立回應資料
         response_data = {
+            'message_id': message_id,
             'images': encoded_images,
             'description': label_names
         }