Update app.py

app.py

@@ -182,45 +182,32 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
         
         result = response.json()
         
-        # ✅ FIX: Ensure result is a dict
         if not isinstance(result, dict):
-            yield f"❌ Invalid response type: {type(result)}\n\n```\n{str(result)[:500]}\n```", None
+            yield f"❌ Invalid response type: {type(result)}", None
             return
         
         print(f"🔍 KEYS: {list(result.keys())}")
         
-        # Extract with safe_get (handles None/non-dict)
-        recommendation = result.get('recommendation') if isinstance(result, dict) else {}
-        if not isinstance(recommendation, dict):
-            recommendation = {}
-        
-        if not recommendation:
-            rec_card = result.get('recommended_card', {})
-            if isinstance(rec_card, dict):
-                recommendation = {
-                    'card_name': rec_card.get('card_name', 'Unknown'),
-                    'rewards_earned': rec_card.get('reward_amount', 0),
-                    'rewards_rate': f"{rec_card.get('reward_rate', 0)}x",
-                    'confidence': 0.85
-                }
-        
-        reasoning_data = result.get('reasoning', {})
-        if not isinstance(reasoning_data, dict):
-            reasoning_data = {}
-        
-        if not reasoning_data.get('main'):
-            reasoning_data = {'main': result.get('final_recommendation', 'No reasoning provided')}
-        
-        card_name = recommendation.get('card_name', 'Unknown Card')
-        rewards_earned = float(recommendation.get('rewards_earned', 0))
-        rewards_rate = recommendation.get('rewards_rate', 'N/A')
-        confidence = float(recommendation.get('confidence', 0))
-        
-        print(f"✅ PARSED: {card_name}, ${rewards_earned}, {rewards_rate}")
-        
-        if card_name == 'Unknown Card' and rewards_earned == 0:
-            yield f"⚠️ **Debug:** Response received but empty\n\n```json\n{json.dumps(result, indent=2)[:1000]}\n```", None
-            return
+        # ✅ FIX: Data is at top level, not nested
+        card_id = result.get('recommended_card', 'Unknown')
+        rewards_earned = float(result.get('rewards_earned', 0))
+        rewards_rate = result.get('rewards_rate', 'N/A')
+        confidence = float(result.get('confidence', 0))
+        reasoning = result.get('reasoning', 'No reasoning provided')
+        alternatives = result.get('alternative_options', [])
+        warnings = result.get('warnings', [])
+        annual_impact = result.get('annual_impact', {})
+        
+        # Map card_id to card_name
+        card_name_map = {
+            'c_citi_custom_cash': 'Citi Custom Cash',
+            'c_amex_gold': 'American Express Gold',
+            'c_chase_sapphire_reserve': 'Chase Sapphire Reserve',
+            'c_chase_freedom_unlimited': 'Chase Freedom Unlimited'
+        }
+        card_name = card_name_map.get(card_id, card_id.replace('c_', '').replace('_', ' ').title())
+        
+        print(f"✅ PARSED: {card_name}, ${rewards_earned}, {rewards_rate}, {confidence}")
         
         output = f"""## 🤖 AI Agent Recommendation
 
@@ -233,9 +220,33 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
 
 ### 🧠 Agent's Reasoning:
 
-{reasoning_data.get('main', 'No reasoning')}
+{reasoning}
 
 ---
+"""
+        
+        if alternatives:
+            output += "\n### 🔄 Alternative Options:\n\n"
+            for alt in alternatives[:3]:
+                alt_card_id = alt.get('card', '')
+                alt_card_name = card_name_map.get(alt_card_id, alt_card_id.replace('c_', '').replace('_', ' ').title())
+                alt_reason = alt.get('reason', '')
+                output += f"**{alt_card_name}:**\n{alt_reason}\n\n"
+        
+        if warnings:
+            output += "\n### ⚠️ Important Warnings:\n\n"
+            for warning in warnings:
+                output += f"- {warning}\n"
+        
+        if annual_impact:
+            potential_savings = annual_impact.get('potential_savings', 0)
+            optimization_score = annual_impact.get('optimization_score', 0)
+            output += f"\n### 💰 Annual Impact:\n"
+            output += f"- **Potential Savings:** ${potential_savings:.2f}/year\n"
+            output += f"- **Optimization Score:** {optimization_score}/100\n"
+        
+        output += f"""
+---
 
 ### 📊 Transaction Details:
 - **Amount:** ${amount:.2f}
@@ -250,95 +261,63 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
         import traceback
         print(f"❌ ERROR: {traceback.format_exc()}")
         yield f"❌ **Error:** {str(e)}", None
-
+        
         
 def create_agent_recommendation_chart_enhanced(result: Dict) -> go.Figure:
-    """Create enhanced chart showing agent's recommendation with alternatives"""
     try:
-        # Extract recommendation data
-        recommendation = safe_get(result, 'recommendation', {})
-        reasoning_data = safe_get(result, 'reasoning', {})
-        
-        # Get recommended card
-        rec_name = safe_get(recommendation, 'card_name', 
-                          safe_get(recommendation, 'recommended_card', 'Recommended Card'))
-        rec_reward = float(safe_get(recommendation, 'rewards_earned', 0))
+        # ✅ FIX: Extract from top level
+        rec_name_map = {
+            'c_citi_custom_cash': 'Citi Custom Cash',
+            'c_amex_gold': 'Amex Gold',
+            'c_chase_sapphire_reserve': 'Sapphire Reserve',
+            'c_chase_freedom_unlimited': 'Freedom Unlimited'
+        }
         
-        # Get alternatives
-        alternatives = safe_get(reasoning_data, 'alternative_options', [])
+        rec_id = result.get('recommended_card', '')
+        rec_name = rec_name_map.get(rec_id, rec_id)
+        rec_reward = float(result.get('rewards_earned', 0))
         
         cards = [rec_name]
         rewards = [rec_reward]
-        colors = ['#667eea']  # Purple for recommended
-        
-        # Add alternatives
-        for alt in alternatives[:3]:  # Limit to top 3
-            if isinstance(alt, dict):
-                alt_card = safe_get(alt, 'card', 'Alternative')
-                # Try to extract reward amount from reason text if available
-                alt_reward = rec_reward * 0.8  # Estimate if not provided
-                cards.append(alt_card)
-                rewards.append(alt_reward)
-                colors.append('#cbd5e0')  # Gray for alternatives
-        
-        # Only create chart if we have valid data
-        if not cards or all(r == 0 for r in rewards):
-            fig = go.Figure()
-            fig.add_annotation(
-                text="No comparison data available",
-                xref="paper", yref="paper",
-                x=0.5, y=0.5, showarrow=False,
-                font=dict(size=14, color="#666")
-            )
-            fig.update_layout(height=400, template='plotly_white')
-            return fig
+        colors = ['#667eea']
+        
+        alternatives = result.get('alternative_options', [])
+        for alt in alternatives[:3]:
+            alt_id = alt.get('card', '')
+            alt_name = rec_name_map.get(alt_id, alt_id)
+            # Extract reward from reason text if possible
+            alt_reward = rec_reward * 0.8  # Estimate
+            cards.append(alt_name)
+            rewards.append(alt_reward)
+            colors.append('#cbd5e0')
         
-        # Create bar chart
         fig = go.Figure(data=[
             go.Bar(
                 x=cards,
                 y=rewards,
-                marker=dict(
-                    color=colors,
-                    line=dict(color='white', width=2)
-                ),
+                marker=dict(color=colors, line=dict(color='white', width=2)),
                 text=[f'${r:.2f}' for r in rewards],
-                textposition='outside',
-                hovertemplate='<b>%{x}</b><br>Rewards: $%{y:.2f}<extra></extra>'
+                textposition='outside'
             )
         ])
         
         fig.update_layout(
-            title={
-                'text': '🎯 Agent\'s Card Comparison',
-                'x': 0.5,
-                'xanchor': 'center'
-            },
+            title='🎯 Card Comparison',
             xaxis_title='Credit Card',
-            yaxis_title='Rewards Earned ($)',
+            yaxis_title='Rewards ($)',
             template='plotly_white',
             height=400,
-            showlegend=False,
-            margin=dict(t=60, b=50, l=50, r=50)
+            showlegend=False
         )
         
         return fig
         
     except Exception as e:
         print(f"Chart error: {e}")
-        import traceback
-        print(traceback.format_exc())
-        
         fig = go.Figure()
-        fig.add_annotation(
-            text="Chart unavailable",
-            xref="paper", yref="paper",
-            x=0.5, y=0.5, showarrow=False,
-            font=dict(size=14, color="#666")
-        )
+        fig.add_annotation(text="Chart unavailable", xref="paper", yref="paper", x=0.5, y=0.5, showarrow=False)
         fig.update_layout(height=400, template='plotly_white')
         return fig
-
         
 # Initialize clients
 client = RewardPilotClient(config.ORCHESTRATOR_URL)