Update main.py

main.py

@@ -94,7 +94,6 @@ async def detect_single_image(file: UploadFile = File(...)):
         # Generate unique filename
         filename = f"detection_{datetime.now().strftime('%Y%m%d_%H%M%S')}_{uuid.uuid4().hex[:8]}.jpg"
 
-        # ✅ FIX: Use consistent relative path
         output_path = UPLOADS_DIR / filename
 
         # Save visualization image
@@ -110,7 +109,6 @@ async def detect_single_image(file: UploadFile = File(...)):
                 "total_damages": len(detections['boxes']),
                 "damage_types": list(set(detections['classes']))
             },
-            # ✅ FIX: Correct paths
             "visualized_image_path": f"uploads/{filename}",
             "visualized_image_url": image_url,
             "filename": filename
@@ -203,7 +201,6 @@ async def compare_vehicle_damages(
                 comparison
             )
 
-            # ✅ FIX: Save visualization image with consistent path
             vis_filename = f"comparison_{timestamp_str}_{session_id}_pos{i + 1}.jpg"
             vis_path = UPLOADS_DIR / vis_filename  # Use consistent relative path
 
@@ -212,7 +209,7 @@ async def compare_vehicle_damages(
             if not success:
                 print(f"⚠️ Warning: Failed to save visualization for position {i + 1}")
             else:
-                print(f"✅ Saved: {vis_path}")
+                print(f" Saved: {vis_path}")
 
             # Generate URL for accessing the image
             vis_url = f"http://localhost:8000/uploads/{vis_filename}"
@@ -227,7 +224,6 @@ async def compare_vehicle_damages(
                     "new_damages": comparison['new_damages'],
                     "matched_damages": comparison['matched_damages'],
                     "repaired_damages": comparison['repaired_damages'],
-                    # ✅ FIX: Correct paths
                     "visualization_path": f"uploads/{vis_filename}",
                     "visualization_url": vis_url,
                     "filename": vis_filename
@@ -249,16 +245,15 @@ async def compare_vehicle_damages(
         grid_results = [res[f"position_{i + 1}"] for i, res in enumerate(position_results)]
         grid_img = visualizer.create_summary_grid(grid_results, image_pairs)
 
-        # ✅ FIX: Save grid summary image with consistent path
         grid_filename = f"summary_grid_{timestamp_str}_{session_id}.jpg"
         grid_path = UPLOADS_DIR / grid_filename  # Already correct in original
 
         # Save the grid image
         grid_success = cv2.imwrite(str(grid_path), grid_img)
         if not grid_success:
-            print(f"⚠️ Warning: Failed to save grid summary")
+            print(f" Warning: Failed to save grid summary")
         else:
-            print(f"✅ Saved grid: {grid_path}")
+            print(f" Saved grid: {grid_path}")
 
         grid_url = f"http://localhost:8000/uploads/{grid_filename}"
 
@@ -279,7 +274,6 @@ async def compare_vehicle_damages(
                 }
             },
             "position_results": position_results,
-            # ✅ FIX: Correct paths
             "summary_visualization_path": f"uploads/{grid_filename}",
             "summary_visualization_url": grid_url,
             "all_visualizations": all_visualizations,  # Added for convenience
@@ -299,7 +293,7 @@ if __name__ == "__main__":
     uvicorn.run(
         "main:app",
         host="0.0.0.0",
-        port=int(os.environ.get("PORT", 7860)),  # Lấy port do HF set
-        reload=False,  # Production không cần reload
+        port=int(os.environ.get("PORT", 7860)),  
+        reload=False,  
         log_level="info"
     )