Update app.py

app.py

@@ -251,6 +251,22 @@ def process_video(video_path, iou_threshold, confidence_threshold, show_detectio
     
     return frames, summary
 
+def process_webcam(frame, iou_threshold, confidence_threshold, show_detections):
+    # Convert from Gradio's webcam format to OpenCV
+    frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
+    
+    # Process the frame (using your existing process_frame function)
+    processed_frame, detected_objects, _ = process_frame(
+        frame, 
+        iou_threshold=iou_threshold,
+        confidence_threshold=confidence_threshold,
+        show_detections=show_detections
+    )
+    
+    # Convert back to RGB for Gradio
+    processed_frame = cv2.cvtColor(processed_frame, cv2.COLOR_BGR2RGB)
+    return processed_frame
+    
 # Gradio Interface
 with gr.Blocks(title="Screw Detection and Measurement") as demo:
     gr.Markdown("# 🔍 Screw Detection and Measurement (YOLOv11 OBB)")
@@ -299,15 +315,15 @@ with gr.Blocks(title="Screw Detection and Measurement") as demo:
             webcam_iou = gr.Slider(label="IoU Threshold (NMS)", minimum=0.0, maximum=1.0, value=0.7, step=0.05)
             webcam_conf = gr.Slider(label="Confidence Threshold", minimum=0.0, maximum=1.0, value=0.5, step=0.05)
             webcam_show_det = gr.Checkbox(label="Show Detections", value=True)
-            webcam_show_sum = gr.Checkbox(label="Show Summary", value=True)
-            settings_button = gr.Button("Update Settings")
         with gr.Column():
-            webrtc_ctx = WebRTC(
-                mode="sendonly",
-                video_processor_factory=VideoProcessor,
-                key="webcam-detection"
-            )
-            webcam_summary = gr.Textbox(label="Summary", interactive=False)
-            refresh_button = gr.Button("Refresh Summary")
+            webcam_input = gr.Image(source="webcam", streaming=True, label="Live Camera")
+            webcam_output = gr.Image(label="Processed Output", streaming=True)
+    
+    # Create live processing
+    webcam_input.change(
+        fn=process_webcam,
+        inputs=[webcam_input, webcam_iou, webcam_conf, webcam_show_det],
+        outputs=webcam_output
+    )
 
 demo.launch()