Update app.py

app.py

@@ -59,14 +59,14 @@ def plot_3d_lungs(lungs_volume, threshold=0.5):
 # --- Gradio Interface ---
 
 def process_and_visualize(selected_folder):
-    if selected_folder not in ["tumor", "lung"]:
+    if selected_folder not in ["tumor", "lung", "tumor2"]:
         return "Invalid folder selection."  # Handle invalid input
 
     volume = segment_lungs_from_dicom(selected_folder)
     visualization = plot_3d_lungs(volume)
     return visualization
 
-inputs = gr.Dropdown(choices=["tumor", "lung"], label="Select DICOM Folder")
+inputs = gr.Dropdown(choices=["tumor", "lung", "tumor2"], label="Select DICOM Folder")
 output = gr.Plot(label="3D Segmented Lungs")
 
 iface = gr.Interface(