Update app.py

app.py

@@ -44,7 +44,7 @@ def segment_webcam():
 
 def process_inputs(start_button, stop_button, mode_selection, uploaded_video):
     global process
-    if start_button:
+    if start_button and mode_selection:
         process = True
         if mode_selection == "Video" and uploaded_video:
             segment_video(uploaded_video)
@@ -53,16 +53,28 @@ def process_inputs(start_button, stop_button, mode_selection, uploaded_video):
     elif stop_button:
         process = False
 
+start_button = gr.components.Button(label="Start", color="green", disabled=True)
+stop_button = gr.components.Button(label="Stop", color="red", disabled=True)
+mode_selection = gr.components.Radio(["Video", "Webcam"], label="Mode Selection")
+uploaded_video = gr.components.File(label="Upload Video")
+
+def on_mode_change(_, __, mode):
+    if mode == "Webcam":
+        uploaded_video.disable()
+    else:
+        uploaded_video.enable()
+    start_button.enable()
+
+mode_selection.on_change(on_mode_change)
+
 iface = gr.Interface(
     fn=process_inputs,
-    inputs=[
-        gr.components.Button(label="Start"),
-        gr.components.Button(label="Stop"),
-        gr.components.Radio(["Video", "Webcam"], label="Mode Selection"),
-        gr.components.File(label="Upload Video")
-    ],
+    inputs=[start_button, stop_button, mode_selection, uploaded_video],
     outputs=None,
-    live=True
+    live=True,
+    title="YOLO Image Segmentation",
+    description="This application uses the YOLO model to perform image segmentation on a video or webcam feed. Select a mode, upload a video (if applicable), and click 'Start' to begin. Click 'Stop' to end the process.",
+    theme="huggingface"
 )
 
 iface.launch()