Update app.py

app.py

@@ -329,6 +329,14 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
             return
         
         result = response.json()
+        # ✅ CRITICAL DEBUG: Check response structure
+        print("=" * 80)
+        print("🔍 DEBUG: Response Structure Analysis")
+        print(f"Response type: {type(result)}")
+        print(f"Top-level keys: {list(result.keys()) if isinstance(result, dict) else 'Not a dict'}")
+        
+        # ✅ FLEXIBLE EXTRACTION: Handle multiple response formats
+        recommendation = None
         
         # Stage 4
         yield """
@@ -360,12 +368,35 @@ def get_recommendation_with_agent(user_id, merchant, category, amount):
         
         print(f"🔍 KEYS: {list(result.keys())}")
         
-        recommendation = result.get('recommendation', {})
+        # Format 1: Nested recommendation (expected from your agent_core.py)
+        if isinstance(result, dict) and 'recommendation' in result:
+            recommendation = result['recommendation']
+            print("✅ Found 'recommendation' key (Format 1: Nested)")
+        
+        # Format 2: Direct recommendation (flat structure)
+        elif isinstance(result, dict) and 'recommended_card' in result:
+            recommendation = result
+            print("✅ Using direct response (Format 2: Flat)")
+        
+        # Format 3: Check if it's wrapped in 'data' key
+        elif isinstance(result, dict) and 'data' in result:
+            data = result['data']
+            if isinstance(data, dict) and 'recommendation' in data:
+                recommendation = data['recommendation']
+                print("✅ Found in 'data.recommendation' (Format 3)")
+            elif isinstance(data, dict) and 'recommended_card' in data:
+                recommendation = data
+                print("✅ Found in 'data' directly (Format 3b)")
         
         if not recommendation:
+            print(f"❌ ERROR: Could not find recommendation in response")
+            print(f"Available keys: {list(result.keys())}")
+            print(f"Full response (first 1000 chars): {str(result)[:1000]}")
+            
             yield f"❌ Invalid response: No recommendation found", None
             return
         
+        
         # ✅ Extract values correctly
         card_id = recommendation.get('recommended_card', 'Unknown')
         card_name = recommendation.get('card_name', card_id)