Fix the visualization

change_visualizer.py

@@ -13,7 +13,7 @@ STATISTICS = {"manual": statistics.get_statistics_for_manual_df(df_manual),
               "synthetic": statistics.get_statistics_for_synthetic_df(df_synthetic)}
 
 
-def update_manual_view(diff_idx):
+def update_dataset_view(diff_idx):
     diff_idx -= 1
     return df_manual.iloc[diff_idx]['annotated_diff'], df_manual.iloc[diff_idx]['commit_msg_start'], \
         df_manual.iloc[diff_idx][
@@ -21,13 +21,6 @@ def update_manual_view(diff_idx):
         'session'], f"https://github.com/{df_manual.iloc[diff_idx]['repo']}/commit/{df_manual.iloc[diff_idx]['hash']}"
 
 
-def update_synthetic_view(diff_idx):
-    diff_idx -= 1
-    return (df_synthetic.iloc[diff_idx]['annotated_diff'], df_synthetic.iloc[diff_idx]['initial_msg_pred'],
-            df_synthetic.iloc[diff_idx]['reference'],
-            f"https://github.com/{df_synthetic.iloc[diff_idx]['repo']}/commit/{df_synthetic.iloc[diff_idx]['hash']}")
-
-
 force_light_theme_js_func = """
 function refresh() {
     const url = new URL(window.location);
@@ -41,42 +34,37 @@ function refresh() {
 
 if __name__ == '__main__':
     with gr.Blocks(theme=gr.themes.Soft(), js=force_light_theme_js_func) as application:
-        with gr.Tab("Manual"):
-            slider_manual = gr.Slider(minimum=1, maximum=n_diffs_manual, step=1, value=1,
-                                      label=f"Sample number (total: {n_diffs_manual})")
-
-            diff_view_manual = gr.Highlightedtext(combine_adjacent=True, color_map={'+': "green", '-': "red"})
-            start_view_manual = gr.Textbox(interactive=False, label="Start message", container=True)
-            end_view_manual = gr.Textbox(interactive=False, label="End message", container=True)
-            session_view_manual = gr.Textbox(interactive=False, label="Session", container=True)
-            link_view_manual = gr.Markdown()
-            view_manual = [
-                diff_view_manual,
-                start_view_manual,
-                end_view_manual,
-                session_view_manual,
-                link_view_manual
+        def dataset_view_tab(n_items):
+            slider = gr.Slider(minimum=1, maximum=n_items, step=1, value=1,
+                               label=f"Sample number (total: {n_items})")
+
+            diff_view = gr.Highlightedtext(combine_adjacent=True, color_map={'+': "green", '-': "red"})
+            start_view = gr.Textbox(interactive=False, label="Start message", container=True)
+            end_view = gr.Textbox(interactive=False, label="End message", container=True)
+            session_view = gr.Textbox(interactive=False, label="Session", container=True)
+            link_view = gr.Markdown()
+
+            view = [
+                diff_view,
+                start_view,
+                end_view,
+                session_view,
+                link_view
             ]
 
-            slider_manual.change(update_manual_view, inputs=slider_manual,
+            return slider, view
+
+
+        with gr.Tab("Manual"):
+            slider_manual, view_manual = dataset_view_tab(n_diffs_manual)
+
+            slider_manual.change(update_dataset_view, inputs=slider_manual,
                                  outputs=view_manual)
 
         with gr.Tab("Synthetic"):
-            slider_synthetic = gr.Slider(minimum=1, maximum=n_diffs_synthetic, step=1, value=1,
-                                         label=f"Sample number (total: {n_diffs_synthetic})")
-
-            diff_view_synthetic = gr.Highlightedtext(combine_adjacent=True, color_map={'+': "green", '-': "red"})
-            start_view_synthetic = gr.Textbox(interactive=False, label="Start message", container=True)
-            end_view_synthetic = gr.Textbox(interactive=False, label="End message", container=True)
-            link_view_synthetic = gr.Markdown()
-            view_synthetic = [
-                diff_view_synthetic,
-                start_view_synthetic,
-                end_view_synthetic,
-                link_view_synthetic
-            ]
+            slider_synthetic, view_synthetic = dataset_view_tab(n_diffs_synthetic)
 
-            slider_synthetic.change(update_synthetic_view, inputs=slider_synthetic,
+            slider_synthetic.change(update_dataset_view, inputs=slider_synthetic,
                                     outputs=view_synthetic)
         with gr.Tab("Compare"):
             def layout_for_statistics(statistics_group_name):
@@ -97,10 +85,10 @@ if __name__ == '__main__':
                 with gr.Column(scale=1):
                     layout_for_statistics("synthetic")
 
-        application.load(update_manual_view, inputs=slider_manual,
+        application.load(update_dataset_view, inputs=slider_manual,
                          outputs=view_manual)
 
-        application.load(update_synthetic_view, inputs=slider_synthetic,
+        application.load(update_dataset_view, inputs=slider_synthetic,
                          outputs=view_synthetic)
 
     application.launch()