target_visualization/app.py at main · MGFPKU/target_visualization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import os
from shiny import App, ui, reactive
from shinywidgets import output_widget, render_widget

import polars as pl

from data import get_data
from i18n import i18n, set_language
import plotly.express as px


def apply_plot_interaction_defaults(fig):
   fig.update_layout(dragmode="pan")
   fig.update_layout(modebar_remove=['select2d', 'lasso2d'])
   return fig

# compile ui
app_ui = ui.page_fluid(
   ui.row(
      ui.tags.div(
         {"class": "col-12 col-xl-6"},
         output_widget("time_plot"),
      ),
      ui.tags.div(
         {"class": "col-12 col-xl-6"},
         output_widget("category_plot"),
      ),
   )
)


def server(input, output, session):

   @reactive.calc
   def lang():
       query = session.clientdata.url_search()
       params = {}
       if query.startswith("?"):
           query = query[1:]
       for pair in query.split("&"):
           if "=" in pair:
               k, v = pair.split("=", 1)
               params[k] = v
       # Query param takes precedence; fall back to env var; ultimate default CN
       return params.get("lang") or os.getenv("LANGUAGE", "CN")

   @reactive.calc
   def df():
       """Load the language-appropriate dataset (cached per language)."""
       return get_data(lang())

   @render_widget # pyrefly: ignore
   def time_plot():
      set_language(lang())
      data = df()
      freq = (
         data.group_by("Announcement_Year")
         .agg(pl.count().alias("frequency"))
         .sort("Announcement_Year")
      )
      fig = px.bar(
         freq,
         x="Announcement_Year",
         y="frequency",
         title=i18n("气候目标年度发布数量"),
         color_discrete_sequence=["#385E4B"]
      )
      fig.update_xaxes(title_text=i18n("发布年份"), tickangle=-45)
      fig.update_yaxes(title_text=i18n("数量"))
      fig.update_layout(yaxis_fixedrange=True)
      fig.update_layout(
         xaxis_tickangle=-45,
         plot_bgcolor="white",
         paper_bgcolor="white"
      )
      return apply_plot_interaction_defaults(fig)

   @render_widget # pyrefly: ignore
   def category_plot():
      set_language(lang())
      data = df()
      category_freq = (
         data.group_by("Target_Category")
         .agg(pl.count().alias("frequency"))
         .sort("frequency")
      )
      fig = px.bar(
         category_freq,
         x="frequency",
         y="Target_Category",
         title=i18n("气候目标类型发布数量"),
         color_discrete_sequence=["#385E4B"]
      )
      fig.update_yaxes(title_text=i18n("气候目标类型"))
      fig.update_xaxes(title_text=i18n("数量"))
      fig.update_layout(
         plot_bgcolor="white",
         paper_bgcolor="white"
      )
      return apply_plot_interaction_defaults(fig)

app = App(app_ui, server, debug=False)