Update app.py

app.py

@@ -163,12 +163,13 @@ with gr.Blocks() as demo:
             response_box = gr.Markdown(label="Chatbot Response")
 
     # Define button actions
-    predict_button.click(fn=lambda air, proc, spd, tq, wear: predict_status(air, proc, spd, tq, wear), 
+    predict_button.click(fn=predict_status, 
                          inputs=[air_temp, proc_temp, speed, torque, tool_wear], 
-                         outputs=output_status)
+                         outputs=[output_status, gr.Variable("reasons")])
 
-    submit_btn.click(fn=lambda reason: handle_query(reason), 
-                     inputs=output_status, 
+    # Correcting the submit_btn action to extract reasons properly
+    submit_btn.click(fn=lambda reasons: handle_query(reasons), 
+                     inputs=gr.Variable("reasons"), 
                      outputs=response_box)
 
     # Display the PDF content