updated files

app.py

@@ -82,19 +82,19 @@ def get_domain_state(dataset):
     empty_update = gr.update(visible=False, value=None, choices=[])
     
     if "data" not in DB:
-        return "", empty_update, empty_update, empty_update
+        return "", empty_update, empty_update, empty_update, empty_update, empty_update
     
     df = DB["data"]
     subset = df[df['dataset_name'] == dataset]
     
     if subset.empty:
-        return "No data for this domain.", empty_update, empty_update, empty_update
+        return "No data for this domain.", empty_update, empty_update, empty_update, empty_update, empty_update
 
     consts, _ = analyze_domain_configs(subset)
     const_text = "CONSTANTS (Fixed for this domain):\n" + "\n".join([f"{k}: {v}" for k,v in consts.items()])
     
     # Fixed filter columns across all domains
-    FILTER_COLUMNS = ['reranker_model', 'chunking_strategy', 'summarization']
+    FILTER_COLUMNS = ['reranker_model', 'chunking_strategy', 'summarization', 'repacking', 'gpt_label']
     
     updates = []
     for col_name in FILTER_COLUMNS:
@@ -111,9 +111,9 @@ def get_domain_state(dataset):
         else:
             updates.append(empty_update)
             
-    return const_text, updates[0], updates[1], updates[2]
+    return const_text, updates[0], updates[1], updates[2], updates[3], updates[4]
 
-def plot_metrics_on_x_axis(dataset, f1_val, f2_val, f3_val):
+def plot_metrics_on_x_axis(dataset, f1_val, f2_val, f3_val, f4_val, f5_val):
     """Generates RMSE and Performance metric plots for selected domain and filters."""
     if "data" not in DB or not dataset:
         return None, None
@@ -126,8 +126,8 @@ def plot_metrics_on_x_axis(dataset, f1_val, f2_val, f3_val):
         return None, None
     
     # Fixed filter columns across all domains
-    FILTER_COLUMNS = ['reranker_model', 'chunking_strategy', 'summarization']
-    filters = [f1_val, f2_val, f3_val]
+    FILTER_COLUMNS = ['reranker_model', 'chunking_strategy', 'summarization', 'repacking', 'gpt_label']
+    filters = [f1_val, f2_val, f3_val, f4_val, f5_val]
     
     for i, val in enumerate(filters):
         if i < len(FILTER_COLUMNS) and val != "All" and val is not None:
@@ -339,6 +339,8 @@ with gr.Blocks(title="RAG Analytics Pro") as demo:
                     filter_1 = gr.Dropdown(visible=False)
                     filter_2 = gr.Dropdown(visible=False)
                     filter_3 = gr.Dropdown(visible=False)
+                    filter_4 = gr.Dropdown(visible=False)
+                    filter_5 = gr.Dropdown(visible=False)
                     
                 with gr.Column(scale=3):
                     plot_r = gr.Plot(label="RMSE Comparison")
@@ -389,17 +391,17 @@ with gr.Blocks(title="RAG Analytics Pro") as demo:
     ds_dropdown.change(
         get_domain_state,
         inputs=[ds_dropdown],
-        outputs=[constants_box, filter_1, filter_2, filter_3]
+        outputs=[constants_box, filter_1, filter_2, filter_3, filter_4, filter_5]
     ).then(
         plot_metrics_on_x_axis,
-        inputs=[ds_dropdown, filter_1, filter_2, filter_3],
+        inputs=[ds_dropdown, filter_1, filter_2, filter_3, filter_4, filter_5],
         outputs=[plot_r, plot_p]
     )
 
     gr.on(
-        triggers=[filter_1.change, filter_2.change, filter_3.change],
+        triggers=[filter_1.change, filter_2.change, filter_3.change, filter_4.change, filter_5.change],
         fn=plot_metrics_on_x_axis,
-        inputs=[ds_dropdown, filter_1, filter_2, filter_3],
+        inputs=[ds_dropdown, filter_1, filter_2, filter_3, filter_4, filter_5],
         outputs=[plot_r, plot_p]
     )