Update main.py

main.py

@@ -427,7 +427,7 @@ def create_learning_situations(state: AgentState, pathname: str = "") -> AgentSt
         "status": "learning_situations_created"
     }
 
-def create_academic_competencies(state: AgentState, pathname: str = "") -> AgentState:
+def create_academic_competencies(state: AgentState, checked:bool, pathname: str = "") -> AgentState:
     """
     Crée 4 compétences académiques, une pour chaque catégorie thématique.
 
@@ -449,7 +449,10 @@ def create_academic_competencies(state: AgentState, pathname: str = "") -> Agent
         cat_name = category["nom"]
         cat_desc = category["description"]
         teachings = classified_teachings.get(cat_name, [])
-        situation = learning_situations.get(cat_name, "")
+        if checked == False:
+            situation = ""
+        else:
+            situation = learning_situations.get(cat_name, "")
 
         if not teachings:
             print(f"Aucun enseignement classé dans la catégorie '{cat_name}'. Passage à la suivante.")
@@ -546,7 +549,7 @@ def create_academic_competencies(state: AgentState, pathname: str = "") -> Agent
         "status": "academic_competencies_created"
     }
 
-def build_workflow(num_bcc,file,pathname) -> StateGraph:
+def build_workflow(num_bcc,file,checked,pathname) -> StateGraph:
     """Construit le graphe de workflow LangGraph."""
 
     # Initialiser le graphe
@@ -558,7 +561,7 @@ def build_workflow(num_bcc,file,pathname) -> StateGraph:
     workflow.add_node("classify_teachings", lambda state: classify_teachings(state, num_bcc))
     workflow.add_node("create_learning_situations", lambda state: create_learning_situations(state, pathname))
     #workflow.add_node("export_to_excel_1", lambda state: export_to_excel(state))
-    workflow.add_node("create_academic_competencies", lambda state: create_academic_competencies(state, pathname))
+    workflow.add_node("create_academic_competencies", lambda state: create_academic_competencies(state, checked, pathname))
     #workflow.add_node("export_to_excel_2", lambda state: export_to_excel(state, "_BCC"))
     #workflow.add_node("process_next_file", get_next_file)
 
@@ -588,7 +591,7 @@ def build_workflow(num_bcc,file,pathname) -> StateGraph:
 
     return workflow
 
-def init_agent_state(current_url, num, pathname) -> AgentState:
+def init_agent_state(current_url, num, checked, pathname) -> AgentState:
     initial_state: AgentState = {
         #"files_list": csv_files[1:],  # Tous les fichiers sauf le premier
         #"current_file": csv_files[0],  # Premier fichier à traiter
@@ -603,7 +606,7 @@ def init_agent_state(current_url, num, pathname) -> AgentState:
     }
 
     # Construire le workflow
-    workflow = build_workflow(num,current_url,pathname)
+    workflow = build_workflow(num,current_url,checked,pathname)
 
     # Compiler le graphe
     app = workflow.compile()
@@ -1251,6 +1254,7 @@ app_page = dmc.MantineProvider(
                         style={"color": "rgb(80,106,139)","font-size": "0.75rem","border":"border 1px solid rgb(67,167,255)!important","margin-bottom":"0.5rem"},
                     ),sm=6),
             ]),
+            dbc.Row(dbc.Col(dmc.Checkbox(id="is-situation", label="Prendre en compte les situations d'apprentissage dans la génération des BCC", checked=False, mb=10),)),
             dbc.Row(dbc.Col(dbc.Button('Créer les nouveaux BCC et les nouvelles situations', id='submit-button', n_clicks=0, style={"background":"linear-gradient(to right, #00F3FF 0%, #7f00b2 100%)","color":"white","font-size": "0.75rem","margin-bottom":"2rem"}),sm=12))
         ], width=12)
     ]),
@@ -2189,11 +2193,12 @@ def display_page(pathname):
     Output("output-response", "children"),
     Input("num-bcc", "value"),
     Input("maquette-dropdown", "value"),
+    Input("is-situation", "checked"),
     Input("submit-button", "n_clicks"),
     State("url", "pathname"),
     prevent_initial_call=True,
 )
-def display_status(num, current, n_clicks, pathname):
+def display_status(num, current, checked, n_clicks, pathname):
     if not n_clicks:
         return no_update
     
@@ -2207,7 +2212,7 @@ def display_status(num, current, n_clicks, pathname):
     current_url = current
     
     print(f"Fichier Maquette sélectionné: {current_url}")
-    agent = init_agent_state(current_url, num, pathname)
+    agent = init_agent_state(current_url, num, checked, pathname)
     try:
         df = agent[0]
         result = agent[1]