Update app.py

app.py

@@ -1925,12 +1925,17 @@ with gr.Blocks(
                 with gr.Column():
                     gr.Markdown("### 🔄 Card Comparison")
                     comparison_output = gr.Markdown()
-            
+
+            def get_recommendation_wrapper(user_id, merchant, category, mcc, amount, use_gemini):
+                global USE_GEMINI
+                USE_GEMINI = use_gemini
+                return get_recommendation_with_agent(user_id, merchant, category, mcc, amount)
+
             recommend_btn.click(
-                fn=get_recommendation_with_agent,
-                inputs=[user_dropdown, merchant_dropdown, category_dropdown, amount_input],
-                outputs=[recommendation_output, recommendation_chart]
-            )
+                                fn=get_recommendation_wrapper,
+                                inputs=[user_id_input, merchant_input, category_input, mcc_input, amount_input, use_gemini_toggle],
+                                outputs=[output, chart]
+                            )
             
             gr.Markdown("### 📝 Example Transactions")
             gr.Examples(