Update app.py

app.py

@@ -20,6 +20,15 @@ try:
 except ImportError:
     GSPREAD_AVAILABLE = False
 
+try:
+    import pandas as pd
+    import matplotlib
+    matplotlib.use("Agg")
+    import matplotlib.pyplot as plt
+    PANDAS_AVAILABLE = True
+except ImportError:
+    PANDAS_AVAILABLE = False
+
 try:
     import spaces  # доступно только внутри HuggingFace Spaces
 except ImportError:
@@ -48,10 +57,15 @@ SPREADSHEET_ID = os.environ.get(
 GOOGLE_SERVICE_ACCOUNT_ENV = "GOOGLE_SERVICE_ACCOUNT_JSON"
 
 SHEETS_SCOPES = [
-    "https://www.googleapis.com/auth/spreadsheets.readonly",
+    "https://www.googleapis.com/auth/spreadsheets",  # чтение и запись
     "https://www.googleapis.com/auth/drive.readonly",
 ]
 
+# Раскладка столбцов на листе класса: A — тема (уже используется),
+# B/C/D — сгенерированные задания по уровням, E — счётчик генераций по теме
+RESULT_COLUMNS = {"базовый": 2, "средний": 3, "продвинутый": 4}  # B, C, D
+GENERATION_COUNT_COLUMN = 5  # E
+
 # Максимальная длина раздела учебника, которая уходит в промпт.
 # Ограничивает расход токенов Groq API даже для очень длинных глав.
 MAX_TEXTBOOK_SECTION_CHARS = 6000
@@ -327,6 +341,113 @@ def load_class_data(class_name: str):
     return topics, textbook_sections, status
 
 
+def format_tasks_for_cell(tasks: list) -> str:
+    """Форматирует список заданий одного уровня в текст для одной ячейки таблицы."""
+    if not tasks:
+        return ""
+
+    blocks = []
+    for i, t in enumerate(tasks, start=1):
+        blocks.append(
+            f"Задание {i}: {t.get('задание', '—')}\n"
+            f"Ответ: {t.get('ответ', '—')}\n"
+            f"Комментарий: {t.get('комментарий', '—')}"
+        )
+    return "\n\n".join(blocks)
+
+
+def write_result_to_sheet(class_name: str, topic: str, by_level: dict) -> str:
+    """
+    Записывает (с перезаписью) сгенерированные задания в столбцы B/C/D строки темы
+    и увеличивает счётчик генераций в столбце E. Работает только если тема реально
+    взята из листа таблицы (найдена точным совпадением в столбце A) — для тем,
+    введённых вручную, запись молча пропускается.
+    """
+    if not class_name or not topic:
+        return ""
+
+    try:
+        gc = get_sheets_client()
+        sh = gc.open_by_key(SPREADSHEET_ID)
+        ws = sh.worksheet(class_name)
+        col_values = ws.col_values(1)
+    except Exception as e:
+        return f"⚠️ Не удалось записать результат в таблицу: {e}"
+
+    row_index = None
+    for i, value in enumerate(col_values, start=1):
+        if value.strip() == topic.strip():
+            row_index = i
+            break
+
+    if row_index is None:
+        # Тема введена вручную и её нет в листе — запись в таблицу не имеет смысла
+        return ""
+
+    try:
+        for level, col in RESULT_COLUMNS.items():
+            cell_text = format_tasks_for_cell(by_level.get(level, []))
+            ws.update_cell(row_index, col, cell_text)
+
+        current_count_values = ws.col_values(GENERATION_COUNT_COLUMN)
+        current_count = 0
+        if len(current_count_values) >= row_index:
+            try:
+                current_count = int(current_count_values[row_index - 1])
+            except (ValueError, IndexError):
+                current_count = 0
+        ws.update_cell(row_index, GENERATION_COUNT_COLUMN, current_count + 1)
+    except Exception as e:
+        return f"⚠️ Задания сгенерированы, но не удалось записать их в таблицу: {e}"
+
+    return f"💾 Результат записан в таблицу (строка {row_index}, лист «{class_name}»)."
+
+
+def build_usage_stats(class_name: str):
+    """
+    Читает столбцы A (темы) и E (счётчик генераций) выбранного листа через Pandas
+    и строит график частоты запросов по темам (Matplotlib).
+    Возвращает (figure, status_message); figure может быть None при ошибке/отсутствии данных.
+    """
+    if not PANDAS_AVAILABLE:
+        return None, "⚠️ Pandas/Matplotlib не установлены."
+
+    if not class_name:
+        return None, "Класс не выбран."
+
+    try:
+        gc = get_sheets_client()
+        sh = gc.open_by_key(SPREADSHEET_ID)
+        ws = sh.worksheet(class_name)
+        topics_col = ws.col_values(1)[1:]  # без 1-й строки (ссылка на учебник)
+        counts_col = ws.col_values(GENERATION_COUNT_COLUMN)[1:]
+    except Exception as e:
+        return None, f"⚠️ Не удалось прочитать данные для статистики: {e}"
+
+    # Выравниваем длины столбцов и приводим счётчик к числу через Pandas
+    n = len(topics_col)
+    counts_col = (counts_col + [""] * n)[:n]
+
+    df = pd.DataFrame({"тема": topics_col, "генераций": counts_col})
+    df["генераций"] = pd.to_numeric(df["генераций"], errors="coerce").fillna(0).astype(int)
+    df = df[df["тема"].str.strip() != ""]
+
+    if df["генераций"].sum() == 0:
+        return None, "Пока нет ни одной генерации по темам этого класса — статистика появится после первых запусков."
+
+    df_sorted = df.sort_values("генераций", ascending=False).head(10)
+
+    fig, ax = plt.subplots(figsize=(8, 4.5))
+    ax.barh(df_sorted["тема"], df_sorted["генераций"], color="#1E8F7A")
+    ax.set_xlabel("Количество генераций")
+    ax.set_title(f"Частота запросов по темам — {class_name}")
+    ax.invert_yaxis()
+    fig.tight_layout()
+
+    total_generations = int(df["генераций"].sum())
+    return fig, f"Всего генераций по классу «{class_name}»: {total_generations}"
+
+
 # ---------------------------------------------------------------------------
 # Логика генерации
 # ---------------------------------------------------------------------------
@@ -390,7 +511,7 @@ def render_level_html(level_key: str, tasks: list) -> str:
 
 
 @spaces.GPU
-def generate_tasks(dropdown_value, custom_topic, n_basic, n_medium, n_advanced, textbook_sections_state):
+def generate_tasks(dropdown_value, custom_topic, n_basic, n_medium, n_advanced, textbook_sections_state, class_state):
     topic = build_topic(dropdown_value, custom_topic)
 
     if not topic:
@@ -453,6 +574,13 @@ def generate_tasks(dropdown_value, custom_topic, n_basic, n_medium, n_advanced,
     if source_note:
         header += f"<p style='color:#2e7d32; font-size:0.9em;'>{source_note}</p>"
 
+    # Запись результата обратно в таблицу (если тема пришла из листа выбранного класса)
+    if tasks and class_state:
+        write_status = write_result_to_sheet(class_state, topic, by_level)
+        if write_status:
+            color = "#2e7d32" if write_status.startswith("💾") else "#b8860b"
+            header += f"<p style='color:{color}; font-size:0.9em;'>{write_status}</p>"
+
     html_parts = [header]
     for level_key in ["базовый", "средний", "продвинутый"]:
         html_parts.append(render_level_html(level_key, by_level[level_key]))
@@ -513,10 +641,16 @@ def on_class_selected(class_name):
 
     if not topics:
         choices = DEFAULT_TOPICS + [CUSTOM_TOPIC_LABEL]
-        return gr.update(choices=choices, value=choices[0]), {}, status
+        return gr.update(choices=choices, value=choices[0]), {}, status, None
 
     choices = topics + [CUSTOM_TOPIC_LABEL]
-    return gr.update(choices=choices, value=choices[0]), textbook_sections, status
+    return gr.update(choices=choices, value=choices[0]), textbook_sections, status, class_name
+
+
+def on_refresh_stats(class_name):
+    """Строит график статистики использования тем через Pandas/Matplotlib."""
+    fig, status = build_usage_stats(class_name)
+    return fig, status
 
 
 # ---------------------------------------------------------------------------
@@ -530,13 +664,15 @@ with gr.Blocks(title="Генератор разноуровневых задан
     Выберите тему (или введите свою), укажите количество заданий на каждый уровень сложности — и получите готовый комплект с ответами и методическими комментариями.
     """)
 
-    # Скрытое состояние — распарсенные разделы учебника {тема: текст}
+    # Скрытое состояние — распарсенные разделы учебника {тема: текст} и текущий класс
     textbook_sections_state = gr.State({})
+    class_state = gr.State(None)
 
     with gr.Accordion("📊 Источник: Google Таблица с темами по классам", open=True):
         gr.Markdown(
             "Таблица содержит листы по классам. В каждом листе: 1-я строка — ссылка "
-            "на .md учебника на Google Drive, остальные строки — темы."
+            "на .md учебника на Google Drive, остальные строки — темы. Столбцы B–D "
+            "��аполняются сгенерированными заданиями по уровням, столбец E — счётчик генераций."
         )
         with gr.Row():
             refresh_classes_btn = gr.Button("🔄 Загрузить список классов")
@@ -580,6 +716,15 @@ with gr.Blocks(title="Генератор разноуровневых задан
         with gr.Column(scale=2):
             output_html = gr.HTML(label="Результат")
 
+    with gr.Accordion("📈 Статистика использования по темам (Pandas + Matplotlib)", open=False):
+        gr.Markdown(
+            "Читает столбец с количеством генераций по каждой теме выбранного класса "
+            "и строит график самых часто запрашиваемых тем."
+        )
+        refresh_stats_btn = gr.Button("Обновить статистику")
+        stats_status = gr.Markdown("Выберите класс выше и нажмите «Обновить статистику».")
+        stats_plot = gr.Plot(label="Частота запросов по темам")
+
     refresh_classes_btn.click(
         fn=on_refresh_classes,
         inputs=[],
@@ -589,7 +734,7 @@ with gr.Blocks(title="Генератор разноуровневых задан
     class_dropdown.change(
         fn=on_class_selected,
         inputs=[class_dropdown],
-        outputs=[topic_dropdown, textbook_sections_state, sheets_status],
+        outputs=[topic_dropdown, textbook_sections_state, sheets_status, class_state],
     )
 
     topics_file.change(
@@ -606,9 +751,15 @@ with gr.Blocks(title="Генератор разноуровневых задан
 
     generate_btn.click(
         fn=generate_tasks,
-        inputs=[topic_dropdown, custom_topic_input, n_basic, n_medium, n_advanced, textbook_sections_state],
+        inputs=[topic_dropdown, custom_topic_input, n_basic, n_medium, n_advanced, textbook_sections_state, class_state],
         outputs=output_html,
     )
 
+    refresh_stats_btn.click(
+        fn=on_refresh_stats,
+        inputs=[class_state],
+        outputs=[stats_plot, stats_status],
+    )
+
 if __name__ == "__main__":
     demo.launch()