Update yolov12/detector.py

yolov12/detector.py

@@ -1,5 +1,6 @@
 import subprocess
 import os
+from pathlib import Path
 
 def run_yolov12_inference(img_path):
     project_dir = "static/uploaded"
@@ -16,14 +17,20 @@ def run_yolov12_inference(img_path):
     "--name", "yolov12_output",
     "--exist-ok"
     ]
-
-    subprocess.run(command, check=True)
-
-    # Try reading the prediction txt file
-    file_name = os.path.basename(img_path)
-    label_file = os.path.join(result_dir, "labels", file_name.replace(".jpg", ".txt"))
-
-    is_damaged = False
+# TEMPORARY: simulate label output
+    filename = Path(args.source).name
+    label_path = Path(args.project) / args.name / "labels"
+    label_path.mkdir(parents=True, exist_ok=True)
+    
+    with open(label_path / filename.replace(".jpg", ".txt"), "w") as f:
+        f.write("0 0.5 0.5 1 1 0.90\n")  # fake class 0
+        subprocess.run(command, check=True)
+    
+        # Try reading the prediction txt file
+        file_name = os.path.basename(img_path)
+        label_file = os.path.join(result_dir, "labels", file_name.replace(".jpg", ".txt"))
+    
+        is_damaged = False
     try:
         with open(label_file, "r") as f:
             for line in f: