Update app.py

app.py

@@ -24,6 +24,8 @@ from utils.formatters import (
     create_trend_line_chart,
     create_card_performance_chart)
 from utils.llm_explainer import get_llm_explainer
+from utils.gemini_explainer import get_gemini_explainer
+
 import config
 
 CARDS_FILE = os.path.join(os.path.dirname(__file__), "data", "cards.json")
@@ -704,6 +706,7 @@ def create_agent_recommendation_chart_enhanced(result: Dict) -> go.Figure:
 
 client = RewardPilotClient(config.ORCHESTRATOR_URL)
 llm = get_llm_explainer()
+gemini = get_gemini_explainer()
 
 def get_recommendation(
     user_id: str,
@@ -785,11 +788,11 @@ def get_recommendation_with_ai(user_id, merchant, category, amount):
             return
         
         data = normalize_recommendation_data(result.get('data', {}))
-        
+
         ai_explanation = ""
-        if config.LLM_ENABLED:
+        if config.USE_GEMINI and gemini.enabled:
             try:
-                ai_explanation = llm.explain_recommendation(
+                ai_explanation = gemini.explain_recommendation(
                     card=data['recommended_card'],
                     rewards=data['rewards_earned'],
                     rewards_rate=data['rewards_rate'],
@@ -800,10 +803,11 @@ def get_recommendation_with_ai(user_id, merchant, category, amount):
                     annual_potential=data['annual_potential'],
                     alternatives=data['alternatives']
                 )
+                ai_explanation = f"🤖 **Powered by Google Gemini**\n\n{ai_explanation}"
             except Exception as e:
-                print(f"LLM explanation failed: {e}")
+                print(f"Gemini explanation failed: {e}")
                 ai_explanation = ""
-        
+
         output = f"""
 ## 🎯 Recommendation for ${amount:.2f} at {merchant}
 
@@ -1850,6 +1854,13 @@ with gr.Blocks(
                         minimum=0.01,
                         step=0.01
                     )
+
+                    with gr.Accordion("🤖 AI Settings", open=False):
+                        use_gemini = gr.Checkbox(
+                            label="Use Google Gemini for explanations",
+                            value=False,
+                            info="Switch to Gemini 1.5 Pro for AI insights"
+                        )
                     
                     date_input = gr.Textbox(
                         label="📅 Transaction Date (Optional)",