Move hardcoded configuration in a config file

climateqa/engine/talk_to_data/config.py

@@ -30,4 +30,53 @@ INDICATOR_COLUMNS_PER_TABLE = {
     "number_of_days_with_tx_above_30": "number_of_days_with_tx_above_30",
     "number_of_days_with_tx_above_35": "number_of_days_with_tx_above_35",
     "number_of_days_with_a_dry_ground": "number_of_days_with_dry_ground"
-}
+}
+
+DRIAS_MODELS = [
+    'ALL',
+    'RegCM4-6_MPI-ESM-LR',
+    'RACMO22E_EC-EARTH',
+    'RegCM4-6_HadGEM2-ES',
+    'HadREM3-GA7_EC-EARTH',
+    'HadREM3-GA7_CNRM-CM5',
+    'REMO2015_NorESM1-M',
+    'SMHI-RCA4_EC-EARTH',
+    'WRF381P_NorESM1-M',
+    'ALADIN63_CNRM-CM5',
+    'CCLM4-8-17_MPI-ESM-LR',
+    'HIRHAM5_IPSL-CM5A-MR',
+    'HadREM3-GA7_HadGEM2-ES',
+    'SMHI-RCA4_IPSL-CM5A-MR',
+    'HIRHAM5_NorESM1-M',
+    'REMO2009_MPI-ESM-LR',
+    'CCLM4-8-17_HadGEM2-ES'
+]
+
+DRIAS_UI_TEXT = """
+Hi, I'm **Talk to Drias**, designed to answer your questions using [**DRIAS - TRACC 2023**](https://www.drias-climat.fr/accompagnement/sections/401) data.  
+I'll answer by displaying a list of SQL queries, graphs and data most relevant to your question.
+
+❓ **How to use?**  
+You can ask me anything about these climate indicators: **temperature**, **precipitation** or **drought**.  
+You can specify **location** and/or **year**.  
+You can choose from a list of climate models. By default, we take the **average of each model**.
+
+For example, you can ask:  
+- What will the temperature be like in Paris?  
+- What will be the total rainfall in France in 2030?  
+- How frequent will extreme events be in Lyon?  
+
+**Example of indicators in the data**:  
+- Mean temperature (annual, winter, summer)  
+- Total precipitation (annual, winter, summer)  
+- Number of days with remarkable precipitations, with dry ground, with temperature above 30°C  
+
+⚠️ **Limitations**:  
+- You can't ask anything that isn't related to **DRIAS - TRACC 2023** data.  
+- You can only ask about **locations in France**.  
+- If you specify a year, there may be **no data for that year for some models**.  
+- You **cannot compare two models**.  
+
+🛈 **Information**  
+Please note that we **log your questions for meta-analysis purposes**, so avoid sharing any sensitive or personal information.
+"""

climateqa/engine/talk_to_data/main.py

@@ -88,25 +88,6 @@ def ask_drias(query: str, index_state: int = 0) -> tuple:
 
     return sql_query, dataframe, figure, sql_queries, result_dataframes, figures, index_state, table_list, ""
 
-DRIAS_MODELS = [
-    'ALL',
-    'RegCM4-6_MPI-ESM-LR',
-    'RACMO22E_EC-EARTH',
-    'RegCM4-6_HadGEM2-ES',
-    'HadREM3-GA7_EC-EARTH',
-    'HadREM3-GA7_CNRM-CM5',
-    'REMO2015_NorESM1-M',
-    'SMHI-RCA4_EC-EARTH',
-    'WRF381P_NorESM1-M',
-    'ALADIN63_CNRM-CM5',
-    'CCLM4-8-17_MPI-ESM-LR',
-    'HIRHAM5_IPSL-CM5A-MR',
-    'HadREM3-GA7_HadGEM2-ES',
-    'SMHI-RCA4_IPSL-CM5A-MR',
-    'HIRHAM5_NorESM1-M',
-    'REMO2009_MPI-ESM-LR',
-    'CCLM4-8-17_HadGEM2-ES'
-]
 # def ask_vanna(vn,db_vanna_path, query):
 
 #     try :

climateqa/engine/talk_to_data/workflow.py

@@ -19,8 +19,6 @@ from climateqa.engine.talk_to_data.utils import (
 
 ROOT_PATH = os.path.dirname(os.path.dirname(os.getcwd()))
 
-DRIAS_DB_PATH = ROOT_PATH + "/data/drias/drias.db"
-
 class TableState(TypedDict):
     """Represents the state of a table in the DRIAS workflow.
     

front/tabs/tab_drias.py

@@ -1,6 +1,7 @@
 import gradio as gr
 
-from climateqa.engine.talk_to_data.main import ask_drias, DRIAS_MODELS
+from climateqa.engine.talk_to_data.main import ask_drias
+from climateqa.engine.talk_to_data.config import DRIAS_MODELS, DRIAS_UI_TEXT
 
 
 def ask_drias_query(query: str, index_state: int):
@@ -84,36 +85,6 @@ def on_table_click(evt: gr.SelectData, table_names, sql_queries, dataframes, plo
     )
 
 
-DETAILS_TEXT = """
-Hi, I'm **Talk to Drias**, designed to answer your questions using [**DRIAS - TRACC 2023**](https://www.drias-climat.fr/accompagnement/sections/401) data.  
-I'll answer by displaying a list of SQL queries, graphs and data most relevant to your question.
-
-❓ **How to use?**  
-You can ask me anything about these climate indicators: **temperature**, **precipitation** or **drought**.  
-You can specify **location** and/or **year**.  
-You can choose from a list of climate models. By default, we take the **average of each model**.
-
-For example, you can ask:  
-- What will the temperature be like in Paris?  
-- What will be the total rainfall in France in 2030?  
-- How frequent will extreme events be in Lyon?  
-
-**Example of indicators in the data**:  
-- Mean temperature (annual, winter, summer)  
-- Total precipitation (annual, winter, summer)  
-- Number of days with remarkable precipitations, with dry ground, with temperature above 30°C  
-
-⚠️ **Limitations**:  
-- You can't ask anything that isn't related to **DRIAS - TRACC 2023** data.  
-- You can only ask about **locations in France**.  
-- If you specify a year, there may be **no data for that year for some models**.  
-- You **cannot compare two models**.  
-
-🛈 **Information**  
-Please note that we **log your questions for meta-analysis purposes**, so avoid sharing any sensitive or personal information.
-"""
-
-
 def create_drias_tab():
     sql_queries_state = gr.State([])
     dataframes_state = gr.State([])
@@ -123,7 +94,7 @@ def create_drias_tab():
 
     with gr.Tab("Beta - Talk to DRIAS", elem_id="tab-vanna", id=6):
         with gr.Accordion(label="Details"):
-            gr.Markdown(DETAILS_TEXT)
+            gr.Markdown(DRIAS_UI_TEXT)
 
         with gr.Row():
             drias_direct_question = gr.Textbox(