From 2d046776f9605d353a937f86db87faf0c870a6d3 Mon Sep 17 00:00:00 2001 From: David Sapiro <115489098+Encord-davids@users.noreply.github.com> Date: Tue, 7 Mar 2023 18:01:36 +0000 Subject: [PATCH] fix(project-selection): use project name from meta instead of dirname (#233) --- src/encord_active/app/streamlit_entrypoint.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/encord_active/app/streamlit_entrypoint.py b/src/encord_active/app/streamlit_entrypoint.py index 3e760baf8..f7ed67d69 100644 --- a/src/encord_active/app/streamlit_entrypoint.py +++ b/src/encord_active/app/streamlit_entrypoint.py @@ -27,6 +27,7 @@ is_project, try_find_parent_project, ) +from encord_active.lib.common.utils import fetch_project_meta from encord_active.lib.db.connection import DBConnection from encord_active.lib.metrics.utils import MetricScope @@ -60,6 +61,14 @@ def to_items(d: dict, parent_key: Optional[str] = None): return [to_item(k, v, parent_key) for k, v in d.items()] +def get_project_name(path: Path): + try: + project_meta = fetch_project_meta(path) + return project_meta["project_title"] + except: + return path.name + + def project_selector(path: Path): if is_project(path): projects = [path] @@ -67,7 +76,7 @@ def project_selector(path: Path): parent_project = try_find_parent_project(path) projects = [parent_project] if parent_project else find_child_projects(path) - return st.selectbox("Choose Project", projects, format_func=lambda path: path.name) + return st.selectbox("Choose Project", projects, format_func=get_project_name) def main(target: str):