src/ui.py - master - Chaizee/ZenithCode_Incident-LLM-analytics

src/ui.py
129 строк5 KB
Chaizee
fix: delete unused code
11 июн 2026, 21:09
11 июн 2026, 21:09d952298
from __future__ import annotations

from typing import Any

import pandas as pd
import plotly.express as px
import streamlit as st

_REQUIRED_CHART_COLUMNS = {"municipality", "problem_count", "mean_severity"}

def _format_number(value: Any, digits: int = 1) -> str:
    try:
        number = float(value)
    except (TypeError, ValueError):
        return "—"
    if number.is_integer():
        return f"{int(number):,}".replace(",", "\u00a0")
    return f"{number:.{digits}f}".replace(".", ",")

def render_charts(df: pd.DataFrame | None, *, critical: pd.DataFrame | None = None) -> None:
    st.markdown(
        "<div class='sr-section-label'>Аналитика по муниципалитетам</div>",
        unsafe_allow_html=True,
    )

    if df is None or df.empty:
        st.warning("Нет данных для построения графиков.")
        return

    missing = _REQUIRED_CHART_COLUMNS.difference(df.columns)
    if missing:
        st.warning(f"Для графиков не хватает столбцов: {', '.join(sorted(missing))}.")
        return

    chart_df = df.copy()
    chart_df["problem_count"] = pd.to_numeric(chart_df["problem_count"], errors="coerce").fillna(0)
    chart_df["mean_severity"] = pd.to_numeric(chart_df["mean_severity"], errors="coerce").fillna(0)
    chart_df = chart_df.sort_values(["problem_count", "mean_severity"], ascending=False)

    total_problems = int(chart_df["problem_count"].sum())
    avg_severity = chart_df["mean_severity"].mean()
    leader = str(chart_df.iloc[0]["municipality"]) if not chart_df.empty else "—"

    col_total, col_avg, col_leader = st.columns(3)
    col_total.metric("Проблем в ТОП", _format_number(total_problems))
    col_avg.metric("Средняя тяжесть", _format_number(avg_severity, digits=2))
    col_leader.metric("Лидер по проблемам", leader)

    st.markdown("<div style='height:1.5rem'></div>", unsafe_allow_html=True)

    tab_count, tab_severity, tab_data = st.tabs([
        "Количество проблем",
        "Средняя тяжесть",
        "Данные",
    ])

    CHART_COLOR = "#1F7AFF"

    with tab_count:
        sorted_for_bar = chart_df.sort_values("problem_count", ascending=True)
        fig = px.bar(
            sorted_for_bar,
            x="problem_count",
            y="municipality",
            orientation="h",
            text="problem_count",
            color_discrete_sequence=[CHART_COLOR],
            labels={
                "problem_count": "Проблемных обращений",
                "municipality": "Муниципалитет",
                "mean_severity": "Средняя тяжесть",
            },
            hover_data={"mean_severity": ":.2f", "municipality": False},
        )
        fig.update_traces(textposition="outside", cliponaxis=False, marker_line_width=0)
        fig.update_layout(
            template="plotly_white",
            height=max(360, len(chart_df) * 52),
            margin=dict(l=0, r=48, t=10, b=0),
            xaxis_title="Проблемных обращений",
            yaxis_title=None,
            showlegend=False,
            plot_bgcolor="rgba(0,0,0,0)",
            paper_bgcolor="rgba(0,0,0,0)",
            font=dict(family="Avenir Next Cyr, system-ui, sans-serif", size=13),
        )
        fig.update_xaxes(showgrid=True, gridcolor="rgba(0,0,0,.06)", zeroline=False)
        fig.update_yaxes(showgrid=False)
        st.plotly_chart(fig, use_container_width=True)

    with tab_severity:
        sorted_for_severity = chart_df.sort_values("mean_severity", ascending=True)
        fig = px.bar(
            sorted_for_severity,
            x="mean_severity",
            y="municipality",
            orientation="h",
            text=sorted_for_severity["mean_severity"].round(2),
            color_discrete_sequence=["#55AAFF"],
            labels={
                "mean_severity": "Средняя тяжесть",
                "municipality": "Муниципалитет",
                "problem_count": "Проблемных обращений",
            },
            hover_data={"problem_count": True, "municipality": False},
        )
        fig.update_traces(textposition="outside", cliponaxis=False, marker_line_width=0)
        fig.update_layout(
            template="plotly_white",
            height=max(360, len(chart_df) * 52),
            margin=dict(l=0, r=48, t=10, b=0),
            xaxis_title="Средняя тяжесть",
            yaxis_title=None,
            showlegend=False,
            plot_bgcolor="rgba(0,0,0,0)",
            paper_bgcolor="rgba(0,0,0,0)",
            font=dict(family="Avenir Next Cyr, system-ui, sans-serif", size=13),
        )
        fig.update_xaxes(showgrid=True, gridcolor="rgba(0,0,0,.06)", zeroline=False)
        fig.update_yaxes(showgrid=False)
        st.plotly_chart(fig, use_container_width=True)

    with tab_data:
        display_df = chart_df.rename(columns={
            "municipality": "Муниципалитет",
            "problem_count": "Проблемных обращений",
            "mean_severity": "Средняя тяжесть",
        })
        st.dataframe(display_df, use_container_width=True, hide_index=True)