Update app.py

app.py

@@ -148,13 +148,9 @@ import config
 
 
 def get_recommendation_with_agent(user_id, merchant, category, amount):
-    """
-    Get recommendation using autonomous agent
-    Shows loading state, then agent's reasoning
-    """
     import httpx
+    import json
     
-    # Show loading first
     yield "⏳ **Agent is thinking...** Analyzing your transaction and cards...", None
     
     try:
@@ -166,136 +162,82 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
             "amount_usd": float(amount)
         }
         
-        # Call orchestrator (which now uses agent)
+        print("=" * 80)
+        print(f"🚀 REQUEST: {config.ORCHESTRATOR_URL}/recommend")
+        print(f"PAYLOAD: {json.dumps(transaction, indent=2)}")
+        
         response = httpx.post(
             f"{config.ORCHESTRATOR_URL}/recommend",
             json=transaction,
-            timeout=60.0  # Agent needs more time
+            timeout=60.0
         )
         
+        print(f"📥 STATUS: {response.status_code}")
+        print(f"📦 RESPONSE: {response.text[:2000]}")
+        print("=" * 80)
+        
         if response.status_code != 200:
             yield f"❌ Error: API returned status {response.status_code}", None
             return
         
         result = response.json()
+        print(f"🔍 KEYS: {list(result.keys())}")
         
-        # ✅ FIXED: Extract data from new response format
+        # Extract with fallbacks
         recommendation = safe_get(result, 'recommendation', {})
-        reasoning_data = safe_get(result, 'reasoning', {})
-        services_data = safe_get(result, 'services_used', {})
-        metadata = safe_get(result, 'agent_metadata', {})
-        transaction_info = safe_get(result, 'transaction', {})
-        plan = safe_get(result, 'plan', {})
-        
-        # Extract card info
-        card_name = safe_get(recommendation, 'card_name', 
-                           safe_get(recommendation, 'recommended_card', 'Unknown Card'))
-        rewards_earned = safe_get(recommendation, 'rewards_earned', 0)
-        rewards_rate = safe_get(recommendation, 'rewards_rate', 'N/A')
-        confidence = safe_get(recommendation, 'confidence', 0)
-        confidence_label = safe_get(recommendation, 'confidence_label', 'Unknown')
+        if not recommendation:
+            rec_card = safe_get(result, 'recommended_card', {})
+            if isinstance(rec_card, dict):
+                recommendation = {
+                    'card_name': safe_get(rec_card, 'card_name', 'Unknown'),
+                    'rewards_earned': safe_get(rec_card, 'reward_amount', 0),
+                    'rewards_rate': f"{safe_get(rec_card, 'reward_rate', 0)}x",
+                    'confidence': 0.85
+                }
         
-        # Extract reasoning
-        main_reasoning = safe_get(reasoning_data, 'main', 'No reasoning provided')
-        why_points = safe_get(reasoning_data, 'why_this_card', [])
-        alternatives = safe_get(reasoning_data, 'alternative_options', [])
-        warnings = safe_get(reasoning_data, 'warnings', [])
-        
-        # Extract insights
-        insights = safe_get(result, 'insights', {})
-        annual_impact = safe_get(insights, 'annual_impact', {})
-        optimization_score = safe_get(insights, 'optimization_score', 0)
+        reasoning_data = safe_get(result, 'reasoning', {})
+        if not reasoning_data.get('main'):
+            reasoning_data = {'main': safe_get(result, 'final_recommendation', 'No reasoning provided')}
         
-        # Extract transaction details
-        tx_amount = safe_get(transaction_info, 'amount', amount)
-        tx_merchant = safe_get(transaction_info, 'merchant', merchant)
-        tx_category = safe_get(transaction_info, 'category', category)
+        card_name = safe_get(recommendation, 'card_name', 'Unknown Card')
+        rewards_earned = float(safe_get(recommendation, 'rewards_earned', 0))
+        rewards_rate = safe_get(recommendation, 'rewards_rate', 'N/A')
+        confidence = float(safe_get(recommendation, 'confidence', 0))
         
-        # Extract service statuses
-        services_used = []
-        for service_name, service_data in services_data.items():
-            if isinstance(service_data, dict) and safe_get(service_data, 'status') == 'success':
-                services_used.append(service_name)
+        print(f"✅ PARSED: {card_name}, ${rewards_earned}, {rewards_rate}")
         
-        # Processing time
-        processing_time = safe_get(metadata, 'processing_time_ms', 0)
+        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
         
-        # ✅ Format comprehensive output
         output = f"""## 🤖 AI Agent Recommendation
 
 ### 💳 Recommended Card: **{card_name}**
 
 **Rewards Earned:** ${rewards_earned:.2f} ({rewards_rate})  
-**Confidence:** {confidence_label} ({confidence*100:.0f}%)
+**Confidence:** {confidence*100:.0f}%
 
 ---
 
 ### 🧠 Agent's Reasoning:
 
-{main_reasoning}
-"""
-        
-        # Add key points
-        if why_points:
-            output += "\n\n**Why This Card:**\n"
-            for point in why_points[:3]:
-                output += f"- {point}\n"
-        
-        output += f"""
+{safe_get(reasoning_data, 'main', 'No reasoning')}
+
 ---
 
 ### 📊 Transaction Details:
-- **Amount:** ${tx_amount:.2f}
-- **Merchant:** {tx_merchant}
-- **Category:** {tx_category}
-
-### ⚙️ Agent Performance:
-- **Services Used:** {', '.join(services_used) if services_used else 'N/A'}
-- **Processing Time:** {processing_time:.0f}ms
-- **Planning Confidence:** {safe_get(metadata, 'planning_confidence', 0)*100:.0f}%
+- **Amount:** ${amount:.2f}
+- **Merchant:** {merchant}
+- **Category:** {category}
 """
         
-        # Add annual impact if available
-        if annual_impact:
-            potential_savings = safe_get(annual_impact, 'potential_savings', 0)
-            if potential_savings > 0:
-                output += f"\n### 💰 Annual Impact:\n- **Potential Savings:** ${potential_savings:.2f}/year\n"
-                output += f"- **Optimization Score:** {optimization_score}/100\n"
-        
-        # Add warnings
-        if warnings:
-            output += "\n\n### ⚠️ Important Warnings:\n"
-            for warning in warnings:
-                output += f"- {warning}\n"
-        
-        # Add alternatives
-        if alternatives:
-            output += "\n\n### 🔄 Alternative Options:\n"
-            for alt in alternatives[:3]:
-                alt_card = safe_get(alt, 'card', 'Unknown')
-                alt_reason = safe_get(alt, 'reason', 'Alternative option')
-                output += f"- **{alt_card}:** {alt_reason}\n"
-        
-        # Add plan strategy
-        strategy = safe_get(plan, 'strategy', '')
-        if strategy:
-            output += f"\n\n### 🎯 Agent Strategy:\n{strategy}\n"
-        
-        output += "\n---\n\n💡 **The AI agent analyzed your transaction context, card portfolio, and spending patterns to determine this is your optimal choice.**"
-        
-        # ✅ Create enhanced comparison chart
         chart = create_agent_recommendation_chart_enhanced(result)
-        
         yield output, chart
         
-    except httpx.TimeoutException:
-        yield "⏱️ **Request timed out.** The agent is taking longer than expected. Please try again.", None
     except Exception as e:
         import traceback
-        error_details = traceback.format_exc()
-        print(f"Agent recommendation error: {error_details}")
-        yield f"❌ **Error:** {str(e)}\n\nPlease check the orchestrator service or try again.", None
-        
+        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"""