Merge pull request #4 from openbiodiversity/feat/project-dropdown

app.py

@@ -11,6 +11,9 @@ import plotly.graph_objects as go
 import yaml
 from google.oauth2 import service_account
 
+
+from utils.js import get_window_url_params
+
 # Logging
 logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
 
@@ -317,6 +320,14 @@ def push_to_md():
  logging.info("upsert records into motherduck")
 
 
+def motherduck_list_projects(author_id):
+ return con.sql(
+ f"""
+ SELECT DISTINCT name FROM project WHERE authorId = '{author_id}'
+ """
+ ).df()
+
+
 with gr.Blocks() as demo:
  # Environment setup
  authenticate_gee(GEE_SERVICE_ACCOUNT, GEE_SERVICE_ACCOUNT_CREDENTIALS_FILE)
@@ -341,7 +352,8 @@ with gr.Blocks() as demo:
  with gr.Row():
  start_year = gr.Number(value=2017, label="Start Year", precision=0)
  end_year = gr.Number(value=2022, label="End Year", precision=0)
- project_name = gr.Textbox(label="Project Name")
+ # project_name = gr.Textbox(label="Project Name")
+ project_name = gr.Dropdown([], label="Project", value="Select project")
  # boroughs = gr.CheckboxGroup(choices=["Queens", "Brooklyn", "Manhattan", "Bronx", "Staten Island"], value=["Queens", "Brooklyn"], label="Select Methodology:")
  # btn = gr.Button(value="Update Filter")
  with gr.Row():
@@ -363,4 +375,28 @@ with gr.Blocks() as demo:
  view_btn.click(view_all, outputs=results_df)
  save_btn.click(push_to_md)
 
+ def update_project_dropdown_list(url_params):
+ username = url_params.get("username", "default")
+ projects = motherduck_list_projects(author_id=username)
+ # to-do: filter projects based on user
+ return gr.Dropdown.update(choices=projects["name"].tolist())
+
+ # 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()

utils/js.py

@@ -0,0 +1,8 @@
+get_window_url_params = """
+ function() {
+ const params = new URLSearchParams(window.location.search);
+ const url_params = Object.fromEntries(params);
+ console.log('url_params', url_params)
+ return url_params;
+ }
+ """