Spaces:
Build error
Build error
| import os | |
| import gradio as gr | |
| import plotly.express as px | |
| from utils import duckdb_queries as dq | |
| from utils.gradio import get_window_url_params | |
| from utils.indicators import IndexGenerator | |
| # Instantiate outside gradio app to avoid re-initializing GEE, which is slow | |
| indexgenerator = IndexGenerator() | |
| metric_names = os.listdir('metrics') | |
| for i in range(len(metric_names)): | |
| metric_names[i] = metric_names[i].split('.yaml')[0].replace('_', ' ') | |
| def toggle_metric_definition_box(text_input): | |
| if text_input is None or text_input == '': | |
| return indexgenerator.get_metric_file() | |
| else: | |
| return None | |
| def make_timeseries_plot(df): | |
| return px.line(df, x='Year', y='Score (Area * Value)', markers=True) | |
| with gr.Blocks() as demo: | |
| with gr.Column(): | |
| with gr.Row(): | |
| m1 = gr.Plot(label='Project Map') | |
| ts_plot = gr.Plot(label='Metric scores over time') | |
| with gr.Row(): | |
| project_name = gr.Dropdown([], label="Project", value="Select project") | |
| metric = gr.Dropdown(metric_names, label='Metric', value='Select metric') | |
| start_year = gr.Number(value=2017, label="Start Year", precision=0) | |
| end_year = gr.Number(value=2022, label="End Year", precision=0) | |
| with gr.Row(): | |
| view_btn = gr.Button(value="Show project map") | |
| calc_btn = gr.Button(value="Calculate metric") | |
| metric_btn = gr.Button(value='Show/hide metric definition') | |
| metric_docs = gr.Textbox( | |
| label="The chosen metric is a linear combination of these components normalized to a range of 0 to 1 and with the given coefficients", | |
| interactive=False) | |
| results_df = gr.Dataframe( | |
| headers=["Year", "Project Name", "Score"], | |
| datatype=["number", "str", "number"], | |
| label="Scores by year", | |
| ) | |
| calc_btn.click( | |
| indexgenerator.calculate_score, | |
| inputs=[start_year, end_year], | |
| outputs=results_df, | |
| ).then( | |
| make_timeseries_plot, | |
| inputs=results_df, | |
| outputs=ts_plot | |
| ) | |
| view_btn.click( | |
| fn=indexgenerator.show_project_map, | |
| outputs=[m1], | |
| ) | |
| def update_project_dropdown_list(url_params): | |
| username = url_params.get("username", "default") | |
| projects = dq.list_projects_by_author(author_id=username) | |
| # Initialize the first project in the list | |
| project_names = projects['name'].tolist() | |
| return gr.Dropdown.update(choices=project_names) | |
| # Change the project name in the index generator object when the | |
| # user selects a new project | |
| project_name.change( | |
| indexgenerator.set_project, | |
| inputs=project_name | |
| ) | |
| # Set the metric to be calculated | |
| metric.change( | |
| indexgenerator.set_metric, | |
| inputs=metric | |
| ) | |
| # Toggle display of metric information | |
| metric_btn.click( | |
| toggle_metric_definition_box, | |
| inputs=metric_docs, | |
| outputs=metric_docs | |
| ) | |
| # Get url params | |
| url_params = gr.JSON({"username": "default"}, visible=False, label="URL Params") | |
| # Gradio has a bug | |
| # For dropdown to update by demo.load, dropdown value must be called downstream | |
| b1 = gr.Button("Hidden button that fixes bug.", visible=False) | |
| b1.click(lambda x: x, inputs=project_name, outputs=[]) | |
| # Update project dropdown list on page load | |
| demo.load( | |
| fn=update_project_dropdown_list, | |
| inputs=[url_params], | |
| outputs=[project_name], | |
| _js=get_window_url_params, | |
| queue=False, | |
| ) | |
| demo.launch() | |