import gradio as gr import generate_annotated_diffs import statistics df_manual = generate_annotated_diffs.manual_data_with_annotated_diffs() n_diffs_manual = len(df_manual) df_synthetic = generate_annotated_diffs.synthetic_data_with_annotated_diffs() n_diffs_synthetic = len(df_synthetic) STATISTICS = {"manual": statistics.get_statistics_for_df(df_manual), "synthetic": statistics.get_statistics_for_df(df_synthetic)} 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][ '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']}" force_light_theme_js_func = """ function refresh() { const url = new URL(window.location); if (url.searchParams.get('__theme') !== 'light') { url.searchParams.set('__theme', 'light'); window.location.href = url.href; } } """ if __name__ == '__main__': with gr.Blocks(theme=gr.themes.Soft(), js=force_light_theme_js_func) as application: 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 ] 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, view_synthetic = dataset_view_tab(n_diffs_synthetic) slider_synthetic.change(update_dataset_view, inputs=slider_synthetic, outputs=view_synthetic) with gr.Tab("Compare"): def layout_for_statistics(statistics_group_name): gr.Markdown(f"### {statistics_group_name}") stats = STATISTICS[statistics_group_name] gr.Number(label="Average deletions number (rel to the initial message length)", interactive=False, value=stats['deletions'].mean().item(), precision=3) gr.Number(label="Average insertions number (rel to the result length)", interactive=False, value=stats['insertions'].mean().item(), precision=3) gr.Number(label="Average changes number (rel to the result length)", interactive=False, value=stats['changes'].mean().item(), precision=3) with gr.Row(): with gr.Column(scale=1): layout_for_statistics("manual") with gr.Column(scale=1): layout_for_statistics("synthetic") application.load(update_dataset_view, inputs=slider_manual, outputs=view_manual) application.load(update_dataset_view, inputs=slider_synthetic, outputs=view_synthetic) application.launch()