Update interface_pages/yoga_position_from_stream.py

interface_pages/yoga_position_from_stream.py

@@ -114,6 +114,7 @@ def yoga_position_from_stream():
             recorded_frames.append(processed_frame)
 
         return (
+            frame,
             processed_frame,
             current_position,
             f"Duration: {int(position_timer)} seconds",
@@ -153,14 +154,14 @@ def yoga_position_from_stream():
 
         return output_path, "Video saved successfully"
 
-    with gr.Blocks() as yoga_stream:
+    with gr.Column() as yoga_stream:
         gr.Markdown("# Yoga Position Classifier")
         gr.Markdown("Stream live yoga sessions and get real-time pose classification.")
 
         debug_toggle = gr.Checkbox(label="Debug Mode", value=False)
 
-        with gr.Row():
-            video_feed = gr.Video(label="Video Feed", source="webcam", streaming=True)
+        with gr.Column(visible=True) as normal_view:
+            video_feed = gr.Webcam(streaming=True, elem_classes="fullscreen")
             pose_output = gr.Textbox(label="Current Pose")
             timer_output = gr.Textbox(label="Pose Duration")
 
@@ -177,13 +178,13 @@ def yoga_position_from_stream():
         debug_toggle.change(
             toggle_debug,
             inputs=[debug_toggle],
-            outputs=[debug_view, video_feed, classified_video],
+            outputs=[debug_view, normal_view, classified_video],
         )
 
         video_feed.stream(
             classify_pose,
             inputs=[video_feed],
-            outputs=[video_feed, pose_output, timer_output],
+            outputs=[video_feed, classified_video, pose_output, timer_output],
             show_progress=False,
         )
 
@@ -196,4 +197,6 @@ def yoga_position_from_stream():
 
 
 if __name__ == "__main__":
-    yoga_position_from_stream().launch()
+    with gr.Blocks() as demo:
+        yoga_position_from_stream()
+    demo.launch()