Update app.py

app.py

@@ -4,34 +4,31 @@ import matplotlib.pyplot as plt
 import io
 from PIL import Image
 
-# Данные
+# Оценки от 1 до 5
 metrics = {
     "Классификация обращений": {
-        "ChatGPT":   {"Точность": 8, "Интерпретируемость": 7, "Стабильность": 7, "Время отклика": 6, "Применимость": 8},
-        "GigaChat":  {"Точность": 7, "Интерпретируемость": 6, "Стабильность": 8, "Время отклика": 8, "Применимость": 7},
-        "DeepSeek":  {"Точность": 6, "Интерпретируемость": 8, "Стабильность": 6, "Время отклика": 9, "Применимость": 7},
+        "ChatGPT":   {"Точность": 5, "Интерпретируемость": 4, "Стабильность": 4, "Время отклика": 3, "Применимость": 4},
+        "GigaChat":  {"Точность": 4, "Интерпретируемость": 3, "Стабильность": 4, "Время отклика": 5, "Применимость": 4},
+        "DeepSeek":  {"Точность": 3, "Интерпретируемость": 4, "Стабильность": 3, "Время отклика": 4, "Применимость": 4},
     },
     "Антифрод": {
-        "ChatGPT":   {"Точность": 7, "Интерпретируемость": 7, "Стабильность": 9, "Время отклика": 5, "Применимость": 6},
-        "GigaChat":  {"Точность": 8, "Интерпретируемость": 6, "Стабильность": 8, "Время отклика": 6, "Применимость": 7},
-        "DeepSeek":  {"Точность": 6, "Интерпретируемость": 5, "Стабильность": 7, "Время отклика": 8, "Применимость": 8},
+        "ChatGPT":   {"Точность": 4, "Интерпретируемость": 3, "Стабильность": 5, "Время отклика": 2, "Применимость": 3},
+        "GigaChat":  {"Точность": 5, "Интерпретируемость": 3, "Стабильность": 4, "Время отклика": 3, "Применимость": 4},
+        "DeepSeek":  {"Точность": 3, "Интерпретируемость": 2, "Стабильность": 4, "Время отклика": 5, "Применимость": 4},
     },
     "Генерация SQL": {
-        "ChatGPT":   {"Точность": 9, "Интерпретируемость": 8, "Стабильность": 7, "Время отклика": 6, "Применимость": 7},
-        "GigaChat":  {"Точность": 6, "Интерпретируемость": 6, "Стабильность": 6, "Время отклика": 7, "Применимость": 6},
-        "DeepSeek":  {"Точность": 8, "Интерпретируемость": 5, "Стабильность": 8, "Время отклика": 9, "Применимость": 8},
+        "ChatGPT":   {"Точность": 5, "Интерпретируемость": 4, "Стабильность": 4, "Время отклика": 3, "Применимость": 4},
+        "GigaChat":  {"Точность": 3, "Интерпретируемость": 3, "Стабильность": 3, "Время отклика": 4, "Применимость": 3},
+        "DeepSeek":  {"Точность": 4, "Интерпретируемость": 2, "Стабильность": 5, "Время отклика": 5, "Применимость": 4},
     }
 }
 
-# Вспомогательная функция: сохраняем matplotlib-plot как изображение
 def fig_to_image(fig):
     buf = io.BytesIO()
     fig.savefig(buf, format="png", bbox_inches="tight", dpi=150)
     buf.seek(0)
-    img = Image.open(buf)
-    return img
+    return Image.open(buf)
 
-# Главная функция обновления
 def update(task, selected_criteria):
     models = ["ChatGPT", "GigaChat", "DeepSeek"]
     cards = []
@@ -42,8 +39,8 @@ def update(task, selected_criteria):
         items_html = ""
         for crit in criteria:
             score = vals[crit]
-            desc = "Высокая" if score >= 8 else "Средняя" if score >= 5 else "Низкая"
-            items_html += f"<li>{crit}: {score}/10 ({desc})</li>"
+            desc = "Высокая" if score == 5 else "Средняя" if score >= 3 else "Низкая"
+            items_html += f"<li>{crit}: {score}/5 ({desc})</li>"
         card_html = f"""
         <div class="card dark-card">
           <h3>{model}</h3>
@@ -56,21 +53,21 @@ def update(task, selected_criteria):
     x = np.arange(len(models))
     width = 0.8 / len(criteria)
 
-    # График 1: сравнение по критериям
+    # График 1: по критериям
     fig1, ax1 = plt.subplots(figsize=(5, 3))
     for i, crit in enumerate(criteria):
         values = [metrics[task][m][crit] for m in models]
         bars = ax1.bar(x + i * width, values, width, label=crit)
         for bar in bars:
             height = bar.get_height()
-            ax1.annotate(f'{height}', xy=(bar.get_x() + bar.get_width() / 2, height + 0.1),
-                         ha='center', fontsize=8)
+            ax1.annotate(f'{height}', xy=(bar.get_x() + bar.get_width() / 2, height + 0.15),
+                         ha='center', fontsize=9, color='white')
     ax1.set_xticks(x + width*(len(criteria)-1)/2)
     ax1.set_xticklabels(models)
-    ax1.set_ylim(0, 10.5)
-    ax1.set_ylabel("Оценка (0-10)")
+    ax1.set_ylim(0, 5.5)
+    ax1.set_ylabel("Оценка (1–5)")
     ax1.set_title(f"Сравнение моделей по критериям\nЗадача: {task}")
-    ax1.legend(fontsize=7)
+    ax1.legend(fontsize=7, loc='upper right')
     ax1.grid(axis='y', linestyle='--', alpha=0.3)
     fig1.patch.set_facecolor('#1e1e1e')
     ax1.set_facecolor('#1e1e1e')
@@ -81,7 +78,7 @@ def update(task, selected_criteria):
     ax1.legend().get_frame().set_facecolor('#2e2e2e')
     ax1.legend().get_frame().set_edgecolor('#444')
 
-    # График 2: средние оценки
+    # График 2: средние баллы
     avg_scores = []
     for model in models:
         vals = metrics[task][model]
@@ -92,10 +89,10 @@ def update(task, selected_criteria):
     bars = ax2.bar(models, avg_scores, color=['#4c78a8','#f58518','#54a24b'])
     for bar in bars:
         height = bar.get_height()
-        ax2.annotate(f'{height:.1f}', xy=(bar.get_x() + bar.get_width() / 2, height + 0.1),
-                     ha='center', fontsize=8)
-    ax2.set_ylim(0, 10.5)
-    ax2.set_ylabel("Средний балл")
+        ax2.annotate(f'{height:.2f}', xy=(bar.get_x() + bar.get_width() / 2, height + 0.15),
+                     ha='center', fontsize=9, color='white')
+    ax2.set_ylim(0, 5.5)
+    ax2.set_ylabel("Средний балл (1–5)")
     ax2.set_title("Средний балл моделей")
     ax2.grid(axis='y', linestyle='--', alpha=0.3)
     fig2.patch.set_facecolor('#1e1e1e')
@@ -105,7 +102,7 @@ def update(task, selected_criteria):
     ax2.xaxis.label.set_color('white')
     ax2.title.set_color('white')
 
-    # Лидер по сумме выбранных критериев
+    # Лидер
     sums = {model: sum(metrics[task][model][crit] for crit in criteria) for model in models}
     leader = max(sums, key=sums.get)
     result = f"📌 **Модель `{leader}` лидирует** по выбранным критериям в задаче **{task}**."
@@ -121,8 +118,8 @@ css = """
 .card.dark-card li { margin: 4px 0; }
 """
 
-# Интерфейс
-with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
+# Gradio интерфейс
+with gr.Blocks(css=css) as demo:
     gr.Markdown("## Сравнение моделей ChatGPT, GigaChat и DeepSeek", elem_id="title")
     with gr.Row():
         with gr.Column(scale=2):
@@ -136,7 +133,6 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
         chart2 = gr.Image(label="Средний балл", interactive=True)
     result_md = gr.Markdown()
 
-    # Обновление при выборе
     task_dropdown.change(update, [task_dropdown, criteria_check], [card1, card2, card3, chart1, chart2, result_md])
     criteria_check.change(update, [task_dropdown, criteria_check], [card1, card2, card3, chart1, chart2, result_md])