Update app.py

app.py

@@ -202,18 +202,8 @@ with gr.Blocks() as demo:
         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'.\
-        
+            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.
@@ -224,15 +214,16 @@ with gr.Blocks() as demo:
         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)-> Tuple[Dict[str, str], str, Any]:
+    def update_reports(topic_content_name:str, content_dict:dict, topic_name:str='')-> Tuple[Dict[str, str], str, Any]:
         """
         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 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`).
+            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:
@@ -249,6 +240,10 @@ with gr.Blocks() as demo:
             - 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)
+            
         topic_content_code = content_dict[topic_content_name]
         titles_dict = get_titles(topic_content_code)
         return titles_dict, topic_content_code, gr.update(choices=list(titles_dict.keys()), visible=True)
@@ -319,12 +314,7 @@ with gr.Blocks() as demo:
     
     Args:
         lang (str): Language. Default value 'en'.
-        topic_params_str (str): string representation of a list that is needed for data extraction.
-            Arguments in the list should be in the following order:
-            - topic code,
-            - topic content code,
-            - topic sub-content code
-            - report ID.
+        topic_params_str (str): string representation of a list that is needed for data extraction. Arguments in the list should be in the following order: topic code, topic content code, topic sub-content code, report ID.\
             These codes you can get from the function get_titles.
         query_kwargs_str (str): string representation of a dictionary - keyword arguments for query configuration.
             Possible query argument names ('code') and their possible values ('values')