Update app.py

app.py

@@ -579,11 +579,11 @@ with gr.Blocks(css=".gradio-container {max-width: 1200px;}") as demo:
     # ======== TAB 2: Unified Distribution/Co-occurrence ========
     with gr.Tab("Distribution/Co-occurrence"):
         gr.Markdown("### Select 1 or 2 features + 1 label to see a bar chart.")
-
-        # Features can be selected from the dataset's columns
-        list_of_features = sorted(df.columns)
+    
+        # Instead of using df.columns, we now only use the keys from input_mapping
+        list_of_features = sorted(input_mapping.keys())
         list_of_labels = sorted(predictor.prediction_map.keys())
-
+    
         selected_features = gr.CheckboxGroup(
             choices=list_of_features,
             label="Select 1 or 2 features"
@@ -592,10 +592,10 @@ with gr.Blocks(css=".gradio-container {max-width: 1200px;}") as demo:
             choices=list_of_labels,
             label="Label Column (e.g., YOWRCONC, YOSEEDOC, etc.)"
         )
-
+    
         generate_combined_btn = gr.Button("Generate Plot")
         combined_output = gr.Plot()
-
+    
         generate_combined_btn.click(
             fn=combined_plot,
             inputs=[selected_features, label_dd],