Refactor build_demo function to enhance the DeathMath leaderboard interface; update labels to Russian, improve layout, and add visualization options for model comparisons

app.py

@@ -121,87 +121,71 @@ def update_plot(selected_models):
     return create_plot(selected_models)
 
 def build_demo():
+    """
+    Строит интерфейс лидерборда DeathMath
+    """
+    # Загружаем данные для лидерборда
     download_openbench()
-    demo = gr.Blocks(title="Small Shlepa", css=custom_css)
+    
+    # Создаем интерфейс с настройками темы
+    demo = gr.Blocks(
+        title="DeathMath Leaderboard", 
+        css=custom_css,
+        theme=gr.themes.Default(
+            primary_hue="indigo",
+            secondary_hue="purple",
+        )
+    )
+    
+    # Получаем данные для лидерборда
     leaderboard_df = build_leadearboard_df()
+    
+    # Строим интерфейс
     with demo:
+        # Заголовок и введение
         gr.HTML(TITLE)
         gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
+        # Основные вкладки
         with gr.Tabs(elem_classes="tab-buttons"):
-            with gr.TabItem("🏅 LLM Benchmark", elem_id="llm-benchmark-tab-table", id=0):
+            # Вкладка лидерборда
+            with gr.TabItem("🏅 Лидерборд", elem_id="llm-benchmark-tab-table", id=0):
+                gr.Markdown("### Таблица результатов моделей DeathMath")
+                
+                # Таблица с результатами
                 Leaderboard(
                     value=leaderboard_df,
                     datatype=[c.type for c in fields(AutoEvalColumn)],
                     select_columns=SelectColumns(
                         default_selection=[c.name for c in fields(AutoEvalColumn) if c.displayed_by_default],
                         cant_deselect=[c.name for c in fields(AutoEvalColumn) if c.never_hidden or c.dummy],
-                        label="Select Columns to Display:",
+                        label="Выберите колонки для отображения:",
                     ),
                     search_columns=[
                         AutoEvalColumn.model.name,
-                        # AutoEvalColumn.fullname.name,
-                        # AutoEvalColumn.license.name
                     ],
                 )
 
-            # with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=1):
-            #    gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
-            # with gr.TabItem("❗FAQ", elem_id="llm-benchmark-tab-table", id=2):
-            #    gr.Markdown(FAQ_TEXT, elem_classes="markdown-text")
-
-            with gr.TabItem("🚀 Submit ", elem_id="llm-benchmark-tab-table", id=3):
+            # Вкладка для отправки результатов
+            with gr.TabItem("🚀 Отправить результаты", elem_id="submit-tab", id=1):
                 with gr.Row():
                     gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
-                with gr.Row():
-                    gr.Markdown("# ✨ Submit your model here!", elem_classes="markdown-text")
-
+                
                 with gr.Column():
-
-                    # def upload_file(file,su,mn):
-                    #     file_path = file.name.split("/")[-1] if "/" in file.name else file.name
-                    #     logging.info("New submition: file saved to %s", file_path)
-                    #     with open(file.name, "r") as f:
-                    #         v=json.load(f)
-                    #         new_file = v['results']
-                    #         new_file['model'] = mn+"/"+su
-                    #         new_file['moviesmc']=new_file['moviemc']["acc,none"]
-                    #         new_file['musicmc']=new_file['musicmc']["acc,none"]
-                    #         new_file['booksmc']=new_file['bookmc']["acc,none"]
-                    #         new_file['lawmc']=new_file['lawmc']["acc,none"]
-                    #         # name = v['config']["model_args"].split('=')[1].split(',')[0]
-                    #         new_file['model_dtype'] = v['config']["model_dtype"]
-                    #         new_file['ppl'] = 0
-                    #         new_file.pop('moviemc')
-                    #         new_file.pop('bookmc')
-                    #     buf = BytesIO()
-                    #     buf.write(json.dumps(new_file).encode('utf-8'))
-                    #     API.upload_file(
-                    #         path_or_fileobj=buf,
-                    #         path_in_repo="model_data/external/" + su+mn + ".json",
-                    #         repo_id="Vikhrmodels/s-openbench-eval",
-                    #         repo_type="dataset",
-                    #     )
-                    #     os.environ[RESET_JUDGEMENT_ENV] = "1"
-                    #     return file.name
-                    # gr.LoginButton()
-                    model_name_textbox = gr.Textbox(label="Model name")
-                    # submitter_username = gr.Textbox(label="Username")
-
-                    # def toggle_upload_button(model_name, username):
-                    #     return bool(model_name) and bool(username)
-                    file_output = gr.File(label="Drag and drop JSON file judgment here", type="filepath")
-                    # upload_button = gr.Button("Click to Upload & Submit Answers", elem_id="upload_button",variant='primary')
+                    model_name_textbox = gr.Textbox(label="Название модели")
+                    file_output = gr.File(label="Перетащите JSON файл с результатами сюда", type="filepath")
                     uploaded_file = gr.State()
                     file_path = gr.State()
+                    
                     with gr.Row():
                         with gr.Column():
-                            out = gr.Textbox("Статус отправки")
+                            out = gr.Textbox("Здесь будет показан статус отправки")
                         with gr.Column():
-                             login_button = gr.LoginButton(elem_id="oauth-button")
-
-                    submit_button = gr.Button("Submit File", elem_id="submit_button", variant='primary')
-
+                            login_button = gr.LoginButton(elem_id="oauth-button")
+                    
+                    submit_button = gr.Button("Отправить результаты", elem_id="submit_button", variant='primary')
+                    
+                    # Обработчики событий
                     file_output.upload(
                         handle_file_upload,
                         file_output,
@@ -214,24 +198,49 @@ def build_demo():
                         [out]
                     )
 
-            with gr.TabItem("📊 Analytics", elem_id="llm-benchmark-tab-table", id=4):
-                    with gr.Column():
+            # Вкладка с аналитикой
+            with gr.TabItem("📊 Аналитика", elem_id="analytics-tab", id=2):
+                with gr.Column():
+                    if len(leaderboard_df) > 0:
                         model_dropdown = gr.Dropdown(
                             choices=leaderboard_df["model"].tolist(),
-                            label="Models",
-                            value=leaderboard_df["model"].tolist(),
+                            label="Модели",
+                            value=leaderboard_df["model"].tolist()[:5] if len(leaderboard_df) >= 5 else leaderboard_df["model"].tolist(),
+                            multiselect=True,
+                            info="Выберите модели для сравнения"
+                        )
+                    else:
+                        model_dropdown = gr.Dropdown(
+                            choices=["example/model-1", "example/model-2"],
+                            label="Модели",
+                            value=["example/model-1", "example/model-2"],
                             multiselect=True,
-                            info="Select models"
+                            info="Выберите модели для сравнения"
                         )
-                    with gr.Column():
-                        plot = gr.Plot(update_plot(model_dropdown.value))
-                        # plot = gr.Plot()
+                    
+                    # Вкладки для разных типов визуализации
+                    with gr.Tabs():
+                        with gr.TabItem("Столбчатая диаграмма"):
+                            bar_plot = gr.Plot(create_plot(model_dropdown.value))
+                            
+                        with gr.TabItem("Радарная диаграмма"):
+                            from src.radial.radial import create_radar_plot
+                            radar_plot = gr.Plot(create_radar_plot(model_dropdown.value))
+                    
+                    # Обновление графиков при изменении выбранных моделей
+                    model_dropdown.change(
+                        fn=create_plot,
+                        inputs=[model_dropdown],
+                        outputs=[bar_plot]
+                    )
+                    
                     model_dropdown.change(
-                            fn=update_plot,
-                            inputs=[model_dropdown],
-                            outputs=[plot]
+                        fn=create_radar_plot,
+                        inputs=[model_dropdown],
+                        outputs=[radar_plot]
                     )
-                    return demo
+    
+    return demo
 
 
 # print(os.system('cd src/gen && ../../.venv/bin/python gen_judgment.py'))