Update app.py

app.py

@@ -239,9 +239,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
         show_progress="full"
     ).then(
         fn=run_agentic_pipeline_autonomously, # Generator function
-        inputs=[agentic_pipeline_inputs, 
-            orchestration_raw_results_st,
-            selected_key_result_ids_st],
+        inputs=[token_state, orchestration_raw_results_st, selected_key_result_ids_st, key_results_for_selection_st],
         outputs=agentic_pipeline_outputs_list,
         show_progress="minimal" # Use minimal for generators that yield status
     )
@@ -260,9 +258,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="sky"),
     )
     sync_event_part2.then(
         fn=run_agentic_pipeline_autonomously, # Generator function
-        inputs=[agentic_pipeline_inputs, 
-            orchestration_raw_results_st,
-            selected_key_result_ids_st],
+        inputs=[token_state, orchestration_raw_results_st, selected_key_result_ids_st, key_results_for_selection_st]
         outputs=agentic_pipeline_outputs_list,
         show_progress="minimal"
     )