updating function calls

README.md

@@ -1,5 +1,5 @@
 ---
-title: Mcp Census
+title: 🏘️ Mcp Census
 emoji: 🏢
 colorFrom: red
 colorTo: gray

app.py

@@ -1,43 +1,43 @@
 import gradio as gr
 from mcp_functions.census_api_calls import (
-    decennial_2020_demographic_profile,
-    decennial_2020_demographic_profile_fips_lookup,
+    dec2020_dp,
+    dec2020_dp_fips_lookup,
 )
 from mcp_functions.census_api_docs import (
-    import_decennial_2020_datasets_homepage,
-    import_decennial_2020_demographic_profile_geographies,
-    import_decennial_2020_demographic_profile_variables,
+    import_dec2020_datasets_homepage,
+    import_dec2020_dp_geographies,
+    import_dec2020_dp_variables,
 )
 from mcp_functions.census_utils import (
-    decennial_2020_dhc_semantic_search,
+    dec2020_dhc_semantic_search,
     required_geograpy_hierarchy_parents,
 )
 
-decennial_2020_demographic_and_housing_characteristics_semantic_search_interface = gr.Interface(
-    fn=decennial_2020_dhc_semantic_search,
+dec2020_dhc_semantic_search_interface = gr.Interface(
+    fn=dec2020_dhc_semantic_search,
     inputs=["textbox"],
     outputs=gr.JSON(),
     title="U.S. Census Bureau 2020 demographic and housing characteristics documentation",
     description="Fetches demographic and housing characteristics documentation for the 2020 U.S. Census Bureau decennial census.",
 )
 
-decennial_2020_demographic_profile_interface = gr.Interface(
-    fn=decennial_2020_demographic_profile,
+dec2020_dp_interface = gr.Interface(
+    fn=dec2020_dp,
     inputs=["textbox", "textbox", "textbox"],
     outputs=gr.JSON(),
     title="U.S. Census Bureau 2020 Demographic Profile data",
     description="Fetches demographic profile data from the 2020 U.S. Census Bureau decennial API.",
 )
 
-decennial_2020_demographic_profile_fips_lookup_interface = gr.Interface(
-    fn=decennial_2020_demographic_profile_fips_lookup,
+dec2020_dp_fips_lookup_interface = gr.Interface(
+    fn=dec2020_dp_fips_lookup,
     inputs=["textbox", "textbox", "textbox"],
     outputs=gr.JSON(),
     title="FIPS code lookup for the U.S. Census Bureau 2020 decennial census demographic profile dataset",
     description="Lookup FIPS codes for geography hierarchies provided by the U.S. Census Bureau 2020 decennial census demographic profile dataset",
 )
 
-decennial_2020_demographic_profile_geographies_required_parent_geographies_interface = gr.Interface(
+dec2020_dp_geographies_required_interface = gr.Interface(
     fn=required_geograpy_hierarchy_parents,
     inputs=["textbox"],
     outputs=gr.JSON(),
@@ -45,16 +45,16 @@ decennial_2020_demographic_profile_geographies_required_parent_geographies_inter
     description="Utility function that provides required parent geographies when requesting geography hierarchies during U.S. Census Bureau 2020 decennial census demographic profile API calls",
 )
 
-decennial_2020_demographic_profile_geographies_interface = gr.Interface(
-    fn=import_decennial_2020_demographic_profile_geographies,
+dec2020_dp_geographies_interface = gr.Interface(
+    fn=import_dec2020_dp_geographies,
     inputs=[],
     outputs=gr.TextArea(),
     title="U.S. Census Bureau 2020 decennial census demographic profile dataset geographies",
     description="Information on available geographies for the the U.S. Census Bureau 2020 decennial census demographic profile API.",
 )
 
-decennial_2020_demographic_profile_variables_interface = gr.Interface(
-    fn=import_decennial_2020_demographic_profile_variables,
+dec2020_dp_variables_interface = gr.Interface(
+    fn=import_dec2020_dp_variables,
     inputs=[],
     outputs=gr.TextArea(),
     title="U.S. Census Bureau 2020 decennial census demographic profile dataset variables",
@@ -62,8 +62,8 @@ decennial_2020_demographic_profile_variables_interface = gr.Interface(
 )
 
 
-decennial_2020_datasets_homepage_interface = gr.Interface(
-    fn=import_decennial_2020_datasets_homepage,
+dec2020_datasets_homepage_interface = gr.Interface(
+    fn=import_dec2020_datasets_homepage,
     inputs=[],
     outputs=gr.TextArea(),
     title="U.S. Census Bureau 2020 decennial census datasets",
@@ -71,15 +71,15 @@ decennial_2020_datasets_homepage_interface = gr.Interface(
 )
 
 
-demo = gr.TabbedInterface(
+mcp_server_demo = gr.TabbedInterface(
     [
-        decennial_2020_datasets_homepage_interface,
-        decennial_2020_demographic_profile_geographies_interface,
-        decennial_2020_demographic_profile_variables_interface,
-        decennial_2020_demographic_profile_geographies_required_parent_geographies_interface,
-        decennial_2020_demographic_profile_fips_lookup_interface,
-        decennial_2020_demographic_profile_interface,
-        decennial_2020_demographic_and_housing_characteristics_semantic_search_interface,
+        dec2020_datasets_homepage_interface,
+        dec2020_dp_geographies_interface,
+        dec2020_dp_variables_interface,
+        dec2020_dp_geographies_required_interface,
+        dec2020_dp_fips_lookup_interface,
+        dec2020_dp_interface,
+        dec2020_dhc_semantic_search_interface,
     ],
     [
         "2020 U.S. Census Bureau decennial census API Homepage",
@@ -94,4 +94,4 @@ demo = gr.TabbedInterface(
 
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    mcp_server_demo.launch(mcp_server=True)

mcp_functions/census_api_calls.py

@@ -41,7 +41,7 @@ if not API_KEY:
     raise ValueError("Set a CENSUS_API_KEY environment variable")
 
 
-def decennial_2020_demographic_profile_fips_lookup(
+def dec2020_dp_fips_lookup(
     geography_hierarchy: str,
     name: str,
     required_parent_geographies: str,
@@ -86,10 +86,14 @@ def decennial_2020_demographic_profile_fips_lookup(
         data = response.json()
     except requests.RequestException as e:
         if error_text == "error: unknown/unsupported geography hierarchy":
+            # TODO: This error gets raise both if a parent geography is missing AND if the requested
+            # geography is invalid. We need to impove this raised error to help parse the difference.
             raise ValueError(
                 "Invalid geography hierarchy provided.",
                 "Acceptable required_parent_geographies must be provided.",
-                f"{geography_hierarchy} requires the following parent geographies: {find_required_parent_geographies(geography_hierarchy)}",
+                f"{geography_hierarchy} requires the following parent geographies: {find_required_parent_geographies(
+                geography_hierarchy
+            )}",
             )
         raise RuntimeError(f"Failed to fetch data from the Census API: {e} ") from e
 
@@ -108,7 +112,7 @@ def decennial_2020_demographic_profile_fips_lookup(
         )
 
 
-def decennial_2020_demographic_profile(
+def dec2020_dp(
     get_variables: str, for_clauses: str, in_clauses: str
 ):
     """

mcp_functions/census_api_docs.py

@@ -2,7 +2,7 @@ import requests
 from markdownify import markdownify
 
 
-def import_decennial_2020_datasets_homepage() -> str:
+def import_dec2020_datasets_homepage() -> str:
     """
     Fetches the homepage for the the U.S. Census Bureau 2020 decennial census API housed at https://api.census.gov/data/2020/dec/dp.html.
     Contains descriptions of available datasets.
@@ -19,7 +19,7 @@ def import_decennial_2020_datasets_homepage() -> str:
     return markdownify(response.text.strip())
 
 
-def import_decennial_2020_demographic_profile_geographies() -> str:
+def import_dec2020_dp_geographies() -> str:
     """
     Fetches information on available geographies for the the U.S. Census Bureau 2020 decennial census demographic profile API housed at https://api.census.gov/data/2020/dec/dp/geography.html.
     Includes:
@@ -37,7 +37,7 @@ def import_decennial_2020_demographic_profile_geographies() -> str:
     return markdownify(response.text.strip())
 
 
-def import_decennial_2020_demographic_profile_variables() -> str:
+def import_dec2020_dp_variables() -> str:
     """
     Fetches information on available variables for the the U.S. Census Bureau 2020 decennial census demographic profile API housed at https://api.census.gov/data/2020/dec/dp/variables.html.
         * "Name" -- used to access variable during API calls

mcp_functions/census_utils.py

@@ -20,7 +20,7 @@ def required_geograpy_hierarchy_parents(geography_hierarchy: str) -> List[str |
     return find_required_parent_geographies(geography_hierarchy)
 
 
-def decennial_2020_dhc_semantic_search(
+def dec2020_dhc_semantic_search(
     query: str,
 ) -> List[Document]:
     """