Update app.py

app.py

@@ -125,19 +125,33 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
 
     out_width, out_height = resize_width, resize_height
     output_path = "processed_output.mp4"
-    fourcc = cv2.VideoWriter_fourcc(*'H264')  # Use H264 codec
-    out = cv2.VideoWriter(output_path, fourcc, fps, (out_width, out_height))
-    if not out.isOpened():
-        log_entries.append("Error: Failed to initialize video writer")
-        logging.error("Failed to initialize video writer")
+    # Try codecs in order of preference
+    codecs = [('mp4v', '.mp4'), ('MJPG', '.avi'), ('XVID', '.avi')]
+    out = None
+    for codec, ext in codecs:
+        fourcc = cv2.VideoWriter_fourcc(*codec)
+        output_path = f"processed_output{ext}"
+        out = cv2.VideoWriter(output_path, fourcc, fps, (out_width, out_height))
+        if out.isOpened():
+            log_entries.append(f"Using codec: {codec}, output: {output_path}")
+            logging.info(f"Using codec: {codec}, output: {output_path}")
+            break
+        else:
+            log_entries.append(f"Failed to initialize codec: {codec}")
+            logging.warning(f"Failed to initialize codec: {codec}")
+
+    if not out or not out.isOpened():
+        log_entries.append("Error: All codecs failed to initialize video writer")
+        logging.error("All codecs failed to initialize video writer")
         cap.release()
-        return "processed_output.mp4", json.dumps({"error": "Failed to initialize video writer"}, indent=2), "\n".join(log_entries), [], None, None
+        return "processed_output.mp4", json.dumps({"error": "All codecs failed"}, indent=2), "\n".join(log_entries), [], None, None
 
     processed_frames = 0
     all_detections = []
     frame_times = []
     detection_frame_count = 0
     output_frame_count = 0
+    last_annotated_frame = None
 
     while True:
         ret, frame = cap.read()
@@ -159,9 +173,10 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
             conf = float(detection.conf)
             box = detection.xyxy[0].cpu().numpy().astype(int).tolist()
             label = model.names[cls]
-            frame_detections.append({"label": label, "box": box, "conf": conf})
-            log_entries.append(f"Frame {frame_count}: Detected {label} with confidence {conf:.2f}")
-            logging.info(f"Frame {frame_count}: Detected {label} with confidence {conf:.2f}")
+            if label != 'Crocodile':  # Ignore irrelevant class
+                frame_detections.append({"label": label, "box": box, "conf": conf})
+                log_entries.append(f"Frame {frame_count}: Detected {label} with confidence {conf:.2f}")
+                logging.info(f"Frame {frame_count}: Detected {label} with confidence {conf:.2f}")
 
         if frame_detections:
             detection_frame_count += 1
@@ -178,6 +193,7 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
         # Write frame and duplicates
         out.write(annotated_frame)
         output_frame_count += 1
+        last_annotated_frame = annotated_frame
         if frame_skip > 1:
             for _ in range(frame_skip - 1):
                 out.write(annotated_frame)
@@ -195,8 +211,9 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
         detection_summary = {
             "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
             "frame": frame_count,
-            "cracks": sum(1 for det in frame_detections if det["label"] == "crack"),
-            "potholes": sum(1 for det in frame_detections if det["label"] == "pothole"),
+            "longitudinal": sum(1 for det in frame_detections if det["label"] == "Longitudinal"),
+            "pothole": sum(1 for det in frame_detections if det["label"] == "Pothole"),
+            "transverse": sum(1 for det in frame_detections if det["label"] == "Transverse"),
             "gps": gps_coord,
             "processing_time_ms": frame_time
         }
@@ -204,9 +221,9 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
         if len(log_entries) > 50:
             log_entries.pop(0)
 
-    # Pad remaining frames if needed
-    while output_frame_count < total_frames and annotated_frame is not None:
-        out.write(annotated_frame)
+    # Pad remaining frames
+    while output_frame_count < total_frames and last_annotated_frame is not None:
+        out.write(last_annotated_frame)
         output_frame_count += 1
 
     last_metrics = update_metrics(all_detections)
@@ -250,7 +267,7 @@ def process_video(video, resize_width=320, resize_height=240, frame_skip=5):
 
 # Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange")) as iface:
-    gr.Markdown("# Crack and Pothole Detection Dashboard")
+    gr.Markdown("# Road Defect Detection Dashboard")
     with gr.Row():
         with gr.Column(scale=3):
             video_input = gr.Video(label="Upload Video")