Update app.py

app.py

@@ -1,38 +1,38 @@
-import os
 import gradio as gr
-from ultralytics import YOLO
-from PIL import Image
-
-# Prevent config error on HF Spaces
-os.environ["YOLO_CONFIG_DIR"] = "/tmp/ultralytics"
-
-model = YOLO("model/best.pt")
+from PIL import Image, ImageDraw
+import random
 
+# Simulated defect labels
 DEFECT_LABELS = ['crack', 'spalling', 'rust', 'deformation']
 
-def detect_defects(image):
-    results = model(image)
-    rendered = results[0].plot()
-    predictions = results[0].boxes.data.cpu().numpy()
-
+# Dummy detection function
+def simulate_defect_detection(image):
+    image = image.convert("RGB")
+    draw = ImageDraw.Draw(image)
     output_lines = []
-    for pred in predictions:
-        conf = float(pred[4])
-        class_id = int(pred[5])
-        label = DEFECT_LABELS[class_id] if class_id < len(DEFECT_LABELS) else f"Class {class_id}"
-        output_lines.append(f"{label}: {conf:.2f}")
 
-    return Image.fromarray(rendered), "\n".join(output_lines)
+    # Randomly simulate 2–4 detections
+    for _ in range(random.randint(2, 4)):
+        label = random.choice(DEFECT_LABELS)
+        conf = round(random.uniform(0.7, 0.99), 2)
+        x1, y1 = random.randint(20, 250), random.randint(20, 250)
+        x2, y2 = x1 + random.randint(50, 120), y1 + random.randint(50, 120)
+        draw.rectangle([x1, y1, x2, y2], outline="red", width=3)
+        draw.text((x1, y1 - 12), f"{label} ({conf})", fill="red")
+        output_lines.append(f"{label}: {conf}")
+
+    return image, "\n".join(output_lines)
 
+# Gradio UI
 demo = gr.Interface(
-    fn=detect_defects,
+    fn=simulate_defect_detection,
     inputs=gr.Image(type="pil", label="Upload Drone Image"),
     outputs=[
-        gr.Image(type="pil", label="Detected Structural Defects"),
-        gr.Textbox(label="Detected Faults and Confidence")
+        gr.Image(type="pil", label="Simulated Structural Defects"),
+        gr.Textbox(label="Predicted Defects with Confidence")
     ],
-    title="🧱 Structural Defect Detection",
-    description="Upload drone-captured image to detect cracks, spalling, rust, and deformation.",
+    title="🧱 Structural Defect Detection (Demo Mode)",
+    description="Simulates detection of cracks, rust, spalling, and deformation in drone-captured images.",
     allow_flagging="never"
 )