Update app.py

app.py

@@ -8,16 +8,16 @@ from typing import Any, Dict, List, Tuple
 def get_topics(name:str = '') -> dict:
     """Available topics from Official Statistics Portal of Latvia (CSP or Centrālā statistikas pārvalde).
     
-    Args:
-        name (str): name of the topic. If not defined, function will return all available topics.
-    Returns:
-        dict: The dictionary of topics, where key is topic name and value is topic code.
-    Examples:
-        >>> get_topics('vide')
-        {'Vide': 'ENV'}
-        >>> print(get_topics())
-        {'Iedzīvotāji': 'POP', 'Darbs': 'EMP', 'Sociālā aizsardzība un veselība': 'VES',...
-    """
+Args:
+    name (str): name of the topic. If not defined, function will return all available topics.
+Returns:
+    dict: The dictionary of topics, where key is topic name and value is topic code.
+Examples:
+    >>> get_topics('vide')
+    {'Vide': 'ENV'}
+    >>> print(get_topics())
+    {'Iedzīvotāji': 'POP', 'Darbs': 'EMP', 'Sociālā aizsardzība un veselība': 'VES',...
+"""
     name_capit = name.capitalize()
     base_url = 'https://data.stat.gov.lv/api/v1/lv/OSP_PUB/'
     content = requests.get(base_url)
@@ -32,22 +32,22 @@ def get_topics(name:str = '') -> dict:
 def get_topic_content(topic: str) -> dict:
     """Available contents of the topic from Official Statistics Portal of Latvia (CSP or Centrālā statistikas pārvalde).
     
-    Args:
-        topic (str): topic code. Use get_topics to get topic code.
-    Returns:
-        dict: The dictionary of the contents of the topic, where key is the topic content and value is the topic content code.
-    Examples:
-        >>> # First get topic code
-        ... get_topics('vide')
-        {'Vide': 'ENV'}
-        >>> # Then use this code to get content
-        ... print(get_topic_content('ENV'))
-        {'Vides konti': 'VI', 'Atkritumu apsaimniekošana': 'AK', 'Agro-vides rādītāji': 'AV',...
-        >>> get_topics('Iedzīvotāji')
-        {'Iedzīvotāji': 'POP'}
-        >>> print(get_topic_content('POP'))
-        {'Iedzīvotāju skaits un raksturojošie rādītāji': 'IR', 'Dzimstība': 'ID', 'Mirstība': 'IM', 'Nāves cēloņi': 'NC',...
-    """
+Args:
+    topic (str): topic code. Use get_topics to get topic code.
+Returns:
+    dict: The dictionary of the contents of the topic, where key is the topic content and value is the topic content code.
+Examples:
+    >>> # First get topic code
+    ... get_topics('vide')
+    {'Vide': 'ENV'}
+    >>> # Then use this code to get content
+    ... print(get_topic_content('ENV'))
+    {'Vides konti': 'VI', 'Atkritumu apsaimniekošana': 'AK', 'Agro-vides rādītāji': 'AV',...
+    >>> get_topics('Iedzīvotāji')
+    {'Iedzīvotāji': 'POP'}
+    >>> print(get_topic_content('POP'))
+    {'Iedzīvotāju skaits un raksturojošie rādītāji': 'IR', 'Dzimstība': 'ID', 'Mirstība': 'IM', 'Nāves cēloņi': 'NC',...
+"""
     base_url = 'https://data.stat.gov.lv/api/v1/lv/OSP_PUB/START/'
     content = requests.get(base_url+topic.upper())
     content_short = {i['text']: i['id'] for i in content.json()}
@@ -197,49 +197,43 @@ with gr.Blocks() as demo:
     run_button = gr.Button("Run Query")
 
     def update_topic_content(topic_name: str)-> Tuple[Any, Dict[str, str], str]:
-        """
-        Given a topic name, updates the UI dropdown choices with the corresponding content,
+        """Given a topic name, updates the UI dropdown choices with the corresponding content,
         and returns the content dictionary (what content is available under given topic_name) and internal topic code (ID of the topic_name).
 
-        Args:
-            topic_name (str): The name of the selected topic. Possible topic names: 'Darbs', 'Iedzīvotāji', 'Informācijas tehnoloģijas', 'Izglītība, kultūra un zinātne', 'Nozares',\
-            'Sociālā aizsardzība un veselība', 'Tirdzniecība un pakalpojumi', 'Uzņēmējdarbība', 'Valsts un ekonomika', 'Vide'.        
-        Returns:
-            tuple:
-                - gr.update: Gradio UI update object with new dropdown choices and visibility set to True.
-                - content_dict (dict): Dictionary containing content entries for the selected topic.
-                - topic_code (str): Internal code corresponding to the topic name.
-        """
+    Args:
+        topic_name (str): The name of the selected topic. Possible topic names: 'Darbs', 'Iedzīvotāji', 'Informācijas tehnoloģijas', 'Izglītība, kultūra un zinātne', 'Nozares',\
+        'Sociālā aizsardzība un veselība', 'Tirdzniecība un pakalpojumi', 'Uzņēmējdarbība', 'Valsts un ekonomika', 'Vide'.        
+    Returns:
+        tuple:
+            - gr.update: Gradio UI update object with new dropdown choices and visibility set to True.
+            - content_dict (dict): Dictionary containing content entries for the selected topic.
+            - topic_code (str): Internal code corresponding to the topic name.
+    """
         topic_code = topic_dict[topic_name]
         content_dict = get_topic_content(topic_code)
         return gr.update(choices=list(content_dict.keys()), visible=True), content_dict, topic_code
 
     def update_reports(topic_content_name: str, content_dict: dict = None,  topic_name: str = '')-> Tuple[Dict[str, str], str, Any]:
-        """
-        Updates the UI dropdown menu with available report titles for a selected topic content,
+        """Updates the UI dropdown menu with available report titles for a selected topic content,
         and returns the titles dictionary and the corresponding content code.
 
-        Args:
-            topic_content_name (str): The name of the selected topic content (e.g., "Darba samaksa (algas)"). Topic content names you can get from `update_topic_content` (`content_dict` object).
-            content_dict (dict): A dictionary mapping topic content names to their corresponding content codes. Obtained from `update_topic_content` (2nd returned object: `content_dict`).
-            topic_name (str): The name of the selected topic. Possible topic names: 'Darbs', 'Iedzīvotāji', 'Informācijas tehnoloģijas', 'Izglītība, kultūra un zinātne', 'Nozares',\
-            'Sociālā aizsardzība un veselība', 'Tirdzniecība un pakalpojumi', 'Uzņēmējdarbība', 'Valsts un ekonomika', 'Vide'.
-
-        Returns:
-            tuple:
-                - gr.update: A Gradio UI update object to populate a dropdown with the list of report titles
-                         and make it visible.
-                - titles_dict (dict): A dictionary of available report titles from the Official Statistics Portal
-                                      of Latvia (CSP). Each key is a human-readable report title, and each value
-                                      is a list of metadata: `[topic_code, topic_content_code, sub_content_code, report_id]`.
-                                      This list (as a string) of metadata later can be used in the `run_get_csp_data` function (into topic_params_str parameter)
-                - topic_content_code (str): The internal code associated with the selected topic content, 
-                                            used to retrieve titles via the `get_titles` function.
-
-        Dependencies:
-            - Relies on `get_titles(topic_content_code)` to fetch metadata from the CSP's API at:
-              https://data.stat.gov.lv/api/v1/lv/OSP_PUB?query=*&filter=*
-        """
+    Args:
+        topic_content_name (str): The name of the selected topic content (e.g., "Darba samaksa (algas)"). Topic content names you can get from `update_topic_content` (`content_dict` object).
+        content_dict (dict): A dictionary mapping topic content names to their corresponding content codes. Obtained from `update_topic_content` (2nd returned object: `content_dict`).
+        topic_name (str): The name of the selected topic. Possible topic names: 'Darbs', 'Iedzīvotāji', 'Informācijas tehnoloģijas', 'Izglītība, kultūra un zinātne', 'Nozares',\
+        'Sociālā aizsardzība un veselība', 'Tirdzniecība un pakalpojumi', 'Uzņēmējdarbība', 'Valsts un ekonomika', 'Vide'.
+    Returns:
+        tuple:
+            - gr.update: A Gradio UI update object to populate a dropdown with the list of report titles and make it visible.
+            - titles_dict (dict): A dictionary of available report titles from the Official Statistics Portal
+                                  of Latvia (CSP). Each key is a human-readable report title, and each value
+                                  is a list of metadata: `[topic_code, topic_content_code, sub_content_code, report_id]`.
+                                  This list (as a string) of metadata later can be used in the `run_get_csp_data` function (into topic_params_str parameter)
+            - topic_content_code (str): The internal code associated with the selected topic content, used to retrieve titles via the `get_titles` function.
+    Dependencies:
+        - Relies on `get_titles(topic_content_code)` to fetch metadata from the CSP's API at:
+          https://data.stat.gov.lv/api/v1/lv/OSP_PUB?query=*&filter=*
+    """
         if len(topic_name) > 0:
             topic_code = topic_dict[topic_name]
             content_dict = get_topic_content(topic_code)
@@ -249,38 +243,34 @@ with gr.Blocks() as demo:
         return titles_dict, topic_content_code, gr.update(choices=list(titles_dict.keys()), visible=True)
 
     def update_topic_params_and_link(report_title: str, titles_dict: str)-> Tuple[str, Any, Any]:
-        """
-        Prepares and returns metadata, a hyperlink, and query parameter preview for a selected report
+        """Prepares and returns metadata, a hyperlink, and query parameter preview for a selected report
         from the Official Statistics Portal of Latvia (CSP).
 
-        Args:
-            report_title (str): The title of the selected report, as shown in the dropdown.
-            titles_dict (dict): Dictionary mapping report titles to their metadata list:
-                                [topic_code, content_code, sub_content_code, report_id],
-                                typically retrieved using `get_titles(topic_content_code)`.
-
-        Returns:
-            tuple:
-                - topic_params_str (str): String representation of the internal report metadata (code list),
-                                          useful for debugging or internal reference.
-                                          This scring can be used in the `run_get_csp_data` (parameter `topic_params_str`).
-                - gr.update: Gradio component update with a Markdown-style hyperlink pointing to the
-                             CSP page for the selected report.
-                - gr.update: Gradio component update showing a sample query parameter, particularly
-                             for the `TIME` dimension if present, using the most recent 3 values.
-
-        Details:
-            - The function extracts the internal metadata for the selected report.
-            - It generates a URL using `construct_csp_link(...)` that links directly to the CSP report page.
-            - It attempts to fetch available query parameters using `get_query_values(...)`, then isolates
-              the `TIME` filter and selects the last 3 available values (e.g., most recent years).
-            - If fetching query parameters fails, an empty dictionary (`'{}'`) is returned as the fallback.
-
-        Example Output:
-            - topic_params_str: "['POP', 'ID', 'IDS', 'IDS010']"
-            - link (Markdown): "[Dzimušo skaits pēc dzimuma](https://data.stat.gov.lv/.../IDS010)"
-            - query_str: "{'TIME': ['2020', '2021', '2022']}"
-        """
+    Args:
+        report_title (str): The title of the selected report, as shown in the dropdown.
+        titles_dict (dict): Dictionary mapping report titles to their metadata list:
+                            [topic_code, content_code, sub_content_code, report_id],
+                            typically retrieved using `get_titles(topic_content_code)`.
+    Returns:
+        tuple:
+            - topic_params_str (str): String representation of the internal report metadata (code list),
+                                      useful for debugging or internal reference.
+                                      This scring can be used in the `run_get_csp_data` (parameter `topic_params_str`).
+            - gr.update: Gradio component update with a Markdown-style hyperlink pointing to the
+                         CSP page for the selected report.
+            - gr.update: Gradio component update showing a sample query parameter, particularly
+                         for the `TIME` dimension if present, using the most recent 3 values.
+    Details:
+        - The function extracts the internal metadata for the selected report.
+        - It generates a URL using `construct_csp_link(...)` that links directly to the CSP report page.
+        - It attempts to fetch available query parameters using `get_query_values(...)`, then isolates
+          the `TIME` filter and selects the last 3 available values (e.g., most recent years).
+        - If fetching query parameters fails, an empty dictionary (`'{}'`) is returned as the fallback.
+    Example Output:
+        - topic_params_str: "['POP', 'ID', 'IDS', 'IDS010']"
+        - link (Markdown): "[Dzimušo skaits pēc dzimuma](https://data.stat.gov.lv/.../IDS010)"
+        - query_str: "{'TIME': ['2020', '2021', '2022']}"
+    """
         title_value = titles_dict[report_title]
         topic_params_str = str(title_value)
         link = construct_csp_link(title_value)