Fix the dfs in visualization

change_visualizer.py

@@ -13,12 +13,20 @@ STATISTICS = {"manual": statistics.get_statistics_for_df(df_manual),
               "synthetic": statistics.get_statistics_for_df(df_synthetic)}
 
 
-def update_dataset_view(diff_idx):
+def update_dataset_view(diff_idx, df):
     diff_idx -= 1
-    return df_manual.iloc[diff_idx]['annotated_diff'], df_manual.iloc[diff_idx]['commit_msg_start'], \
-        df_manual.iloc[diff_idx][
-            'commit_msg_end'], df_manual.iloc[diff_idx][
-        'session'], f"https://github.com/{df_manual.iloc[diff_idx]['repo']}/commit/{df_manual.iloc[diff_idx]['hash']}"
+    return df.iloc[diff_idx]['annotated_diff'], df.iloc[diff_idx]['commit_msg_start'], \
+        df.iloc[diff_idx][
+            'commit_msg_end'], df.iloc[diff_idx][
+        'session'], f"https://github.com/{df.iloc[diff_idx]['repo']}/commit/{df.iloc[diff_idx]['hash']}"
+
+
+def update_dataset_view_manual(diff_idx):
+    update_dataset_view(diff_idx, df_manual)
+
+
+def update_dataset_view_synthetic(diff_idx):
+    update_dataset_view(diff_idx, df_synthetic)
 
 
 force_light_theme_js_func = """
@@ -58,13 +66,13 @@ if __name__ == '__main__':
         with gr.Tab("Manual"):
             slider_manual, view_manual = dataset_view_tab(n_diffs_manual)
 
-            slider_manual.change(update_dataset_view, inputs=slider_manual,
+            slider_manual.change(update_dataset_view_manual, inputs=slider_manual,
                                  outputs=view_manual)
 
         with gr.Tab("Synthetic"):
             slider_synthetic, view_synthetic = dataset_view_tab(n_diffs_synthetic)
 
-            slider_synthetic.change(update_dataset_view, inputs=slider_synthetic,
+            slider_synthetic.change(update_dataset_view_synthetic, inputs=slider_synthetic,
                                     outputs=view_synthetic)
         with gr.Tab("Compare"):
             def layout_for_statistics(statistics_group_name):
@@ -85,10 +93,10 @@ if __name__ == '__main__':
                 with gr.Column(scale=1):
                     layout_for_statistics("synthetic")
 
-        application.load(update_dataset_view, inputs=slider_manual,
+        application.load(update_dataset_view_manual, inputs=slider_manual,
                          outputs=view_manual)
 
-        application.load(update_dataset_view, inputs=slider_synthetic,
+        application.load(update_dataset_view_synthetic, inputs=slider_synthetic,
                          outputs=view_synthetic)
 
     application.launch()