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;
+        }
+    """