Update services/agentic_handlers.py

services/agentic_handlers.py

@@ -40,16 +40,17 @@ class AgenticHandlers:
         self.agentic_modules_really_loaded = AGENTIC_MODULES_LOADED
         logging.info(f"AgenticHandlers initialized. Modules loaded: {self.agentic_modules_really_loaded}")
 
-    def _safe_checkbox_update(self, choices=None, value=None, interactive=None):
+    def _update_checkbox_component_directly(self, component, choices=None, value=None):
         """
-        Create a safe update for CheckboxGroup that handles version compatibility.
+        Update CheckboxGroup component directly instead of using gr.update()
+        since gr.update() doesn't support choices parameter in some Gradio versions.
         """
-        update_dict = {}
-        
-        # Only include parameters that are supported and valid
-        if choices is not None:
-            # Ensure choices is a list of tuples (display_value, return_value)
-            if isinstance(choices, list):
+        try:
+            if component is None:
+                return gr.update()
+                
+            # Format choices properly
+            if choices is not None:
                 formatted_choices = []
                 for choice in choices:
                     if isinstance(choice, tuple) and len(choice) == 2:
@@ -57,32 +58,22 @@ class AgenticHandlers:
                     else:
                         # Convert single values to (display, value) tuples
                         formatted_choices.append((str(choice), choice))
-                update_dict['choices'] = formatted_choices
-            else:
-                update_dict['choices'] = []
-        
-        if value is not None:
-            # Ensure value is a list and only contains valid choices
-            if isinstance(value, list):
-                if choices is not None:
-                    # Filter value to only include items that exist in choices
-                    valid_values = [choice[1] if isinstance(choice, tuple) else choice for choice in (choices or [])]
-                    filtered_value = [v for v in value if v in valid_values]
-                    update_dict['value'] = filtered_value
+                
+                # Update component properties directly
+                component.choices = formatted_choices
+                
+            if value is not None:
+                if isinstance(value, list):
+                    component.value = value
                 else:
-                    update_dict['value'] = []
-            else:
-                update_dict['value'] = []
-        
-        # Try to include interactive parameter, but handle gracefully if not supported
-        if interactive is not None:
-            try:
-                update_dict['interactive'] = interactive
-            except:
-                # If interactive parameter is not supported, continue without it
-                pass
-        
-        return gr.update(**update_dict)
+                    component.value = []
+                    
+            # Return a simple update that just refreshes the component
+            return gr.update()
+            
+        except Exception as e:
+            logging.error(f"Error updating checkbox component: {e}")
+            return gr.update()
 
     async def run_agentic_pipeline_autonomously_on_update(self, current_token_state_val):
         """
@@ -108,12 +99,8 @@ class AgenticHandlers:
         report_status_md_update = gr.update(value=initial_report_status) if self.report_components.get("agentic_pipeline_status_md") else gr.update()
         report_display_md_update = gr.update() 
         
-        # Use safe checkbox update method
-        okrs_cbg_update = self._safe_checkbox_update(
-            choices=initial_okr_cbg_choices, 
-            value=initial_okr_cbg_value, 
-            interactive=initial_okr_cbg_interactive
-        ) if self.okrs_components.get("key_results_cbg") else gr.update()
+        # Initialize checkbox component - no update needed initially
+        okrs_cbg_update = gr.update() if self.okrs_components.get("key_results_cbg") else gr.update()
         
         okrs_detail_md_update = gr.update(value=initial_okr_details) if self.okrs_components.get("okr_detail_display_md") else gr.update()
 
@@ -154,7 +141,7 @@ class AgenticHandlers:
             
             # Update for key_results_cbg (CheckboxGroup) in error case
             if self.okrs_components.get("key_results_cbg"):
-                okrs_cbg_update = self._safe_checkbox_update(choices=[], value=[], interactive=False)
+                okrs_cbg_update = gr.update()
             
             if self.okrs_components.get("okr_detail_display_md"):
                  okrs_detail_md_update = gr.update(value=error_status) 
@@ -209,11 +196,14 @@ class AgenticHandlers:
                             kr_choices_for_cbg.append((kr_desc, kr['unique_kr_id']))
                 
                 if self.okrs_components.get("key_results_cbg"):
-                    okrs_cbg_update = self._safe_checkbox_update(
+                    # Update the checkbox component directly
+                    checkbox_component = self.okrs_components["key_results_cbg"]
+                    self._update_checkbox_component_directly(
+                        checkbox_component, 
                         choices=kr_choices_for_cbg, 
-                        value=[], 
-                        interactive=True
+                        value=[]
                     )
+                    okrs_cbg_update = gr.update()
                 
                 all_okrs_md_parts = []
                 if actionable_okrs and isinstance(actionable_okrs.get("okrs"), list):
@@ -235,7 +225,13 @@ class AgenticHandlers:
                 
                 # Update for key_results_cbg (CheckboxGroup) if no output
                 if self.okrs_components.get("key_results_cbg"):
-                    okrs_cbg_update = self._safe_checkbox_update(choices=[], value=[], interactive=False)
+                    checkbox_component = self.okrs_components["key_results_cbg"]
+                    self._update_checkbox_component_directly(
+                        checkbox_component, 
+                        choices=[], 
+                        value=[]
+                    )
+                    okrs_cbg_update = gr.update()
                 
                 if self.okrs_components.get("okr_detail_display_md"):
                     okrs_detail_md_update = gr.update(value="Nessun OKR generato o errore nella pipeline AI (Sempre).")