Update app.py

app.py

@@ -54,20 +54,33 @@ def process_video(video_path, mode):
  yield None, output_video_file
 
 
-input_video = gr.Video(source="webcam", label="Input Video")
-
-output_frames = gr.Image(label="Output Frames")
-output_video_file = gr.Video(label="Output video")
-
-interface = gr.Interface(
- fn=process_video,
- inputs=[input_video, gr.Radio(choices=["Beginner", "Pro"], label="Select Mode")],
- outputs=[output_frames, output_video_file],
- title=f"AI Fitness Trainer: Squats Analysis",
- allow_flagging="never"
-)
-
-interface.queue().launch()
+input_video = gr.Video(label="Input Video")
+webcam_video = gr.Video(source="webcam", label="Input Video")
+
+output_frames_up = gr.Image(label="Output Frames")
+output_video_file_up = gr.Video(label="Output video")
+
+output_frames_cam = gr.Image(label="Output Frames")
+output_video_file_cam = gr.Video(label="Output video")
+
+upload_interface = gr.Interface(
+ fn=process_video,
+ inputs=[input_video, gr.Radio(choices=["Beginner", "Pro"], label="Select Mode")],
+ outputs=[output_frames_up, output_video_file_up],
+ title=f"AI Fitness Trainer: Squats Analysis",
+ allow_flagging="never"
+ )
  
 
+webcam_interface = gr.Interface(
+ fn=process_video,
+ inputs=[webcam_video, gr.Radio(choices=["Beginner", "Pro"], label="Select Mode")],
+ outputs=[output_frames_cam, output_video_file_cam],
+ title=f"AI Fitness Trainer: Squats Analysis",
+ allow_flagging="never"
+ ) 
+
+app = gr.TabbedInterface([upload_interface, webcam_interface], 
+ tab_names=["⬆️ Upload Video", "📷️ Live Stream"])
 
+app.queue().launch()