Update app.py

app.py

@@ -2250,40 +2250,89 @@ with gr.Blocks(
             
             
             def respond(message, chat_history, user_id):
-                """Handle chat responses with error handling"""
+                """Handle chat responses with Gemini AI"""
                 if not message.strip():
                     return "", chat_history
                 
+                # ✅ Get user context for personalized responses
                 user_context = {}
                 try:
                     analytics = client.get_user_analytics(user_id)
                     if analytics.get('success'):
                         data = analytics.get('data', {})
                         user_context = {
+                            'user_id': user_id,
                             'cards': safe_get(data, 'cards', ['Amex Gold', 'Chase Sapphire Reserve']),
                             'monthly_spending': safe_get(data, 'total_spending', 0),
-                            'top_category': safe_get(data, 'top_category', 'Groceries')
+                            'top_category': safe_get(data, 'top_category', 'Groceries'),
+                            'total_rewards': safe_get(data, 'total_rewards', 0),
+                            'optimization_score': safe_get(data, 'optimization_score', 75)
                         }
                 except Exception as e:
                     print(f"Error getting user context: {e}")
                     user_context = {
+                        'user_id': user_id,
                         'cards': ['Amex Gold', 'Chase Sapphire Reserve'],
                         'monthly_spending': 3450.75,
-                        'top_category': 'Groceries'
+                        'top_category': 'Groceries',
+                        'total_rewards': 125.50,
+                        'optimization_score': 75
                     }
                 
+                # ✅ Use Gemini for chat responses
                 try:
-                    if config.LLM_ENABLED:
-                        bot_response = llm.chat_response(message, user_context, chat_history)
+                    if config.USE_GEMINI and gemini.enabled:
+                        bot_response = gemini.chat_response(message, user_context, chat_history)
+                        print(f"✅ Gemini chat response generated")
                     else:
-                        bot_response = "I'm currently in fallback mode. Ask me about specific cards or categories!"
+                        # Fallback response
+                        bot_response = generate_fallback_response(message, user_context)
+                        print(f"⚠️ Using fallback response (Gemini not available)")
                 except Exception as e:
-                    print(f"Chat error: {e}")
-                    bot_response = f"I encountered an error. Please try asking your question differently."
+                    print(f"❌ Chat error: {e}")
+                    print(traceback.format_exc())
+                    bot_response = generate_fallback_response(message, user_context)
                 
                 chat_history.append((message, bot_response))
                 return "", chat_history
             
+            
+            def generate_fallback_response(message: str, user_context: dict) -> str:
+                """Generate a simple fallback response when Gemini is unavailable"""
+                message_lower = message.lower()
+                
+                # Card-specific questions
+                if "amex" in message_lower or "gold" in message_lower:
+                    return "The Amex Gold Card offers 4x points on dining and groceries at U.S. supermarkets (up to $25,000/year). It's excellent for food spending!"
+                
+                elif "chase" in message_lower and "sapphire" in message_lower:
+                    return "Chase Sapphire Reserve offers 3x points on travel and dining, plus premium travel benefits like airport lounge access and travel credits."
+                
+                elif "citi" in message_lower and "custom" in message_lower:
+                    return "Citi Custom Cash gives you 5% cashback on your top spending category each month (up to $500), then 1% on everything else."
+                
+                # Category questions
+                elif "grocery" in message_lower or "groceries" in message_lower:
+                    return f"For groceries, I recommend using a card with high grocery rewards. Your top spending category is {user_context.get('top_category', 'Groceries')}."
+                
+                elif "restaurant" in message_lower or "dining" in message_lower:
+                    return "For dining, cards like Amex Gold (4x points) or Chase Sapphire Reserve (3x points) offer excellent rewards."
+                
+                elif "travel" in message_lower:
+                    return "For travel, consider Chase Sapphire Reserve (3x points) or cards with travel-specific bonuses and protections."
+                
+                # Spending questions
+                elif "spending" in message_lower or "how much" in message_lower:
+                    return f"Based on your profile, you've spent ${user_context.get('monthly_spending', 0):.2f} this month, earning ${user_context.get('total_rewards', 0):.2f} in rewards."
+                
+                elif "optimize" in message_lower or "maximize" in message_lower:
+                    return f"Your optimization score is {user_context.get('optimization_score', 0)}/100. To improve, use the 'Get Recommendation' tab for each purchase!"
+                
+                # Default
+                else:
+                    return "I'm here to help with credit card recommendations! Try asking about specific cards, categories, or your spending patterns. For personalized recommendations, use the 'Get Recommendation' tab."
+                    
+            
             msg.submit(respond, [msg, chatbot, chat_user], [msg, chatbot])
             send_btn.click(respond, [msg, chatbot, chat_user], [msg, chatbot])