Update app.py

app.py

@@ -306,7 +306,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                         "I dettagli sulla formula/metodologia appariranno qui.", visible=False
                     )
 
-            async def handle_panel_action( # This function is async
+            async def handle_panel_action( 
                 plot_id_clicked: str, 
                 action_type: str, 
                 current_active_action_from_state: dict, 
@@ -318,11 +318,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                 clicked_plot_config = next((p for p in plot_configs if p["id"] == plot_id_clicked), None)
                 if not clicked_plot_config:
                     logging.error(f"Configurazione non trovata per plot_id {plot_id_clicked}")
-                    # Ensure we return the correct number of updates even on error
                     num_button_updates = 2 * len(plot_configs)
-                    error_updates = [gr.update(visible=False)] * 10 # Updates for action column components
-                    error_updates.extend([current_active_action_from_state, current_chat_plot_id, current_chat_histories]) # States
-                    error_updates.extend([gr.update()] * num_button_updates) # Button icons
+                    error_updates = [gr.update(visible=False)] * 10 
+                    error_updates.extend([current_active_action_from_state, current_chat_plot_id, current_chat_histories]) 
+                    error_updates.extend([gr.update()] * num_button_updates) 
                     return error_updates
 
                 clicked_plot_label = clicked_plot_config["label"]
@@ -416,7 +415,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
                     updated_chat_histories         
                 ] + all_button_icon_updates
                 
-                return final_updates # This list should have 13 + (19*2) = 51 items
+                return final_updates
 
             async def handle_chat_message_submission(
                 user_message: str, 
@@ -541,21 +540,29 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
             ]
             explore_click_inputs = [explored_plot_id_state]
 
+            # --- Define async wrapper functions for click handlers ---
+            async def insights_click_wrapper(current_active_val, current_chats_val, current_chat_pid, p_id):
+                return await handle_panel_action(p_id, "insights", current_active_val, current_chats_val, current_chat_pid)
+
+            async def formula_click_wrapper(current_active_val, current_chats_val, current_chat_pid, p_id):
+                return await handle_panel_action(p_id, "formula", current_active_val, current_chats_val, current_chat_pid)
+            # --- End async wrapper functions ---
+
             for config_item in plot_configs:
                 plot_id = config_item["id"]
-                plot_label = config_item["label"] 
+                # plot_label = config_item["label"] # Not needed here anymore
                 if plot_id in plot_ui_objects:
                     ui_obj = plot_ui_objects[plot_id]
-                    # FIX: Make lambda async and await the call to handle_panel_action
+                    
+                    # Use a standard lambda to call the async wrapper, capturing p_id
                     ui_obj["bomb_button"].click(
-                        fn=async lambda current_active_val, current_chats_val, current_chat_pid, p_id=plot_id: await handle_panel_action(p_id, "insights", current_active_val, current_chats_val, current_chat_pid),
+                        fn=lambda cav, ccv, ccpid, p=plot_id: insights_click_wrapper(cav, ccv, ccpid, p),
                         inputs=action_click_inputs, 
                         outputs=action_panel_outputs_list,
                         api_name=f"action_insights_{plot_id}"
                     )
-                    # FIX: Make lambda async and await the call to handle_panel_action
                     ui_obj["formula_button"].click(
-                        fn=async lambda current_active_val, current_chats_val, current_chat_pid, p_id=plot_id: await handle_panel_action(p_id, "formula", current_active_val, current_chats_val, current_chat_pid),
+                        fn=lambda cav, ccv, ccpid, p=plot_id: formula_click_wrapper(cav, ccv, ccpid, p),
                         inputs=action_click_inputs,
                         outputs=action_panel_outputs_list,
                         api_name=f"action_formula_{plot_id}"