Update app.py

Add source URL to the run_get_csp_data output

app.py

@@ -121,9 +121,14 @@ def get_query_values(topic_params: list[str] = []) -> List[Dict]:
     except ValueError as ve:
         raise RuntimeError(f"Parsing failed: {ve}")
 
+def construct_csp_link(params: list[str]) -> str:
+    base_url = 'https://data.stat.gov.lv/pxweb/lv/OSP_PUB/START__'
+    mid_path = '__'.join(params[:3])
+    last_part = params[3]
+    return f"{base_url}{mid_path}/{last_part}/"
 
 def get_csp_data(lang: str = 'en', topic_params: list[str] = [], **kwargs) -> List[Dict]:
-    """Get statistics from Official Statistics Portal of Latvia (CSP or Centrālā statistikas pārvalde).
+    """Get statistics from Official Statistics Portal of Latvia (CSP or Centrālā statistikas pārvalde). Always cite 'Source URL' in the output.
     
     Args:
         lang (str): Language. Default value 'en'.
@@ -139,7 +144,7 @@ def get_csp_data(lang: str = 'en', topic_params: list[str] = [], **kwargs) -> Li
             can be obtained using the function get_query_values.
             'valueTexts' from get_query_values could help if 'values' is unclear.
     Returns:
-        list: The list of the dictionaries, where dictionary's key 'key' contains query parameters and key 'values' contains values.
+        list: The list of the dictionaries, where dictionary's key 'key' contains query parameters and key 'values' contains values. First list element is data source (URL) from CSB.
     Examples:
         >>> topics = ['POP', 'IR', 'IRE', 'IRE010']
         >>> query_args = get_query_values(topics)
@@ -156,16 +161,11 @@ def get_csp_data(lang: str = 'en', topic_params: list[str] = [], **kwargs) -> Li
     """
     csp2 = LVStat(lang, *topic_params)
     csp2.set_query(**kwargs)
+
+    link = construct_csp_link(topic_params)
     
     cspdata2 = csp2.get_data()
-    return cspdata2['data']
-
-
-def construct_csp_link(params: list[str]) -> str:
-    base_url = 'https://data.stat.gov.lv/pxweb/lv/OSP_PUB/START__'
-    mid_path = '__'.join(params[:3])
-    last_part = params[3]
-    return f"{base_url}{mid_path}/{last_part}/"
+    return [{"Source URL": link}] + cspdata2['data']
 
 
 with gr.Blocks() as demo: