Fixed Gradio table select crash

app.py

@@ -129,7 +129,7 @@ def build_rag_chain():
         """
         PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
 
-        # Included the token-saving "k": 2 limit we discussed earlier!
+        # Included the token-saving "k": 2 limit
         retriever = main_vector_store.as_retriever(search_kwargs={"k": 2})
         QA_CHAIN = RetrievalQA.from_chain_type(
             llm=llm,
@@ -240,11 +240,18 @@ if build_rag_chain():
             submit_feedback_btn = gr.Button("Submit Rating")
             feedback_status = gr.Textbox(label="Feedback Status", interactive=False)
 
-        def update_selection(df, evt: gr.SelectData):
-            if df is None or df.empty: return ""
-            selected_row = df.iloc[evt.index[0]]
-            action = selected_row['action']
-            return action
+        # FIX 1: New safer update_selection function
+        def update_selection(table_df, evt: gr.SelectData):
+            # Safety check if the table is empty
+            if table_df is None or len(table_df) == 0: 
+                return ""
+            
+            # evt.index gives us [row_index, column_index] of the click
+            row_idx = evt.index[0]
+            
+            # "Recommended Action" is the 2nd column in your UI table (index 1)
+            selected_action = table_df.iloc[row_idx, 1]
+            return selected_action
 
         submit_btn.click(
             fn=fmea_rag_interface,
@@ -252,9 +259,10 @@ if build_rag_chain():
             outputs=[rpn_output, recommendations_output, df_state]
         )
 
+        # FIX 2: Trigger relies on the visible table (recommendations_output) instead of df_state
         recommendations_output.select(
             fn=update_selection,
-            inputs=[df_state],
+            inputs=[recommendations_output], # <-- Pass the visible table directly!
             outputs=[selected_action_text]
         )