Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,7 @@
 """
 Enhanced Telecom Customer Segmentation Dashboard
 ===============================================
+Features EVERYTHING Talha requested:
 ✅ Interactive visual insights
 ✅ Communication: Location, Intl calls, Frequency, Duration, Time (Morning/Noon/Night)
 ✅ Internet: Download, Upload, Overall
@@ -521,127 +522,134 @@ Provide response in this format:
 """
                     response = query_ai(context)
                     
-                    # Beautiful formatted output
-                    st.markdown("""
-                    <style>
-                    .recommendation-header {
-                        background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-                        color: white;
-                        padding: 20px;
-                        border-radius: 10px 10px 0 0;
-                        text-align: center;
-                        font-size: 24px;
-                        font-weight: bold;
-                        margin-bottom: 0;
-                    }
-                    .recommendation-body {
-                        background: #f8f9fa;
-                        padding: 25px;
-                        border-radius: 0 0 10px 10px;
-                        border: 2px solid #667eea;
-                        line-height: 1.8;
-                    }
-                    .usage-badge {
-                        display: inline-block;
-                        background: #e3f2fd;
-                        color: #1976d2;
-                        padding: 5px 12px;
-                        border-radius: 20px;
-                        font-size: 14px;
-                        font-weight: 600;
-                        margin: 5px 5px 5px 0;
-                    }
-                    .package-highlight {
-                        background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%);
-                        color: white;
-                        padding: 20px;
-                        border-radius: 10px;
-                        font-size: 18px;
-                        font-weight: bold;
-                        text-align: center;
-                        margin: 15px 0;
-                        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
-                    }
-                    </style>
-                    <div class="recommendation-header">
-                        💡 Personalized Package Recommendation
-                    </div>
-                    """, unsafe_allow_html=True)
-                    
-                    # Display formatted recommendation
-                    with st.container():
-                        st.markdown('<div class="recommendation-body">', unsafe_allow_html=True)
-                        
-                        # Usage summary badges
-                        col1, col2, col3, col4 = st.columns(4)
-                        with col1:
-                            usage_level = "High" if comm['voice_total_calls'] > 500 else "Moderate" if comm['voice_total_calls'] > 200 else "Low"
-                            st.markdown(f'<div class="usage-badge">📞 Voice: {usage_level}</div>', unsafe_allow_html=True)
-                        with col2:
-                            data_level = "High" if internet['total_mb'] > 1000 else "Moderate" if internet['total_mb'] > 500 else "Low"
-                            st.markdown(f'<div class="usage-badge">📊 Data: {data_level}</div>', unsafe_allow_html=True)
-                        with col3:
-                            sms_level = "High" if sms['total_messages'] > 200 else "Moderate" if sms['total_messages'] > 50 else "Low"
-                            st.markdown(f'<div class="usage-badge">💬 SMS: {sms_level}</div>', unsafe_allow_html=True)
-                        with col4:
-                            if intl['is_international_user']:
-                                st.markdown(f'<div class="usage-badge">🌍 International</div>', unsafe_allow_html=True)
-                            else:
-                                st.markdown(f'<div class="usage-badge">🏠 Domestic Only</div>', unsafe_allow_html=True)
-                        
-                        st.markdown("<br>", unsafe_allow_html=True)
-                        
-                        # AI Response with better formatting
-                        ai_text = response['answer']
-                        
-                        # Try to parse sections (if AI follows format)
-                        sections = {
-                            'profile': '',
-                            'package': '',
-                            'benefits': '',
-                            'pricing': ''
+                    # Check for errors first
+                    if response and 'error' in response:
+                        st.error(f"❌ {response['error']}")
+                        st.info("💡 Please check your backend connection and Gemini API key configuration.")
+                    elif response and 'answer' in response:
+                        # Beautiful formatted output
+                        st.markdown("""
+                        <style>
+                        .recommendation-header {
+                            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+                            color: white;
+                            padding: 20px;
+                            border-radius: 10px 10px 0 0;
+                            text-align: center;
+                            font-size: 24px;
+                            font-weight: bold;
+                            margin-bottom: 0;
                         }
+                        .recommendation-body {
+                            background: #f8f9fa;
+                            padding: 25px;
+                            border-radius: 0 0 10px 10px;
+                            border: 2px solid #667eea;
+                            line-height: 1.8;
+                        }
+                        .usage-badge {
+                            display: inline-block;
+                            background: #e3f2fd;
+                            color: #1976d2;
+                            padding: 5px 12px;
+                            border-radius: 20px;
+                            font-size: 14px;
+                            font-weight: 600;
+                            margin: 5px 5px 5px 0;
+                        }
+                        .package-highlight {
+                            background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%);
+                            color: white;
+                            padding: 20px;
+                            border-radius: 10px;
+                            font-size: 18px;
+                            font-weight: bold;
+                            text-align: center;
+                            margin: 15px 0;
+                            box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+                        }
+                        </style>
+                        <div class="recommendation-header">
+                            💡 Personalized Package Recommendation
+                        </div>
+                        """, unsafe_allow_html=True)
                         
-                        # Simple parsing (fallback to full text if not structured)
-                        if any(keyword in ai_text.lower() for keyword in ['usage profile', 'recommended package', 'benefits', 'pricing']):
-                            # Structured response
-                            lines = ai_text.split('\n')
-                            current_section = None
-                            for line in lines:
-                                line_lower = line.lower()
-                                if 'usage profile' in line_lower or 'profile' in line_lower and len(line) < 50:
-                                    current_section = 'profile'
-                                elif 'recommended package' in line_lower or 'package' in line_lower and len(line) < 50:
-                                    current_section = 'package'
-                                elif 'benefit' in line_lower and len(line) < 50:
-                                    current_section = 'benefits'
-                                elif 'pricing' in line_lower and len(line) < 50:
-                                    current_section = 'pricing'
-                                elif current_section and line.strip():
-                                    sections[current_section] += line + '\n'
+                        # Display formatted recommendation
+                        with st.container():
+                            st.markdown('<div class="recommendation-body">', unsafe_allow_html=True)
                             
-                            # Display structured
-                            if sections['profile']:
-                                st.markdown("**📋 Usage Profile Analysis**")
-                                st.info(sections['profile'].strip())
+                            # Usage summary badges
+                            col1, col2, col3, col4 = st.columns(4)
+                            with col1:
+                                usage_level = "High" if comm['voice_total_calls'] > 500 else "Moderate" if comm['voice_total_calls'] > 200 else "Low"
+                                st.markdown(f'<div class="usage-badge">📞 Voice: {usage_level}</div>', unsafe_allow_html=True)
+                            with col2:
+                                data_level = "High" if internet['total_mb'] > 1000 else "Moderate" if internet['total_mb'] > 500 else "Low"
+                                st.markdown(f'<div class="usage-badge">📊 Data: {data_level}</div>', unsafe_allow_html=True)
+                            with col3:
+                                sms_level = "High" if sms['total_messages'] > 200 else "Moderate" if sms['total_messages'] > 50 else "Low"
+                                st.markdown(f'<div class="usage-badge">💬 SMS: {sms_level}</div>', unsafe_allow_html=True)
+                            with col4:
+                                if intl['is_international_user']:
+                                    st.markdown(f'<div class="usage-badge">🌍 International</div>', unsafe_allow_html=True)
+                                else:
+                                    st.markdown(f'<div class="usage-badge">🏠 Domestic Only</div>', unsafe_allow_html=True)
                             
-                            if sections['package']:
-                                st.markdown("**🎁 Recommended Package**")
-                                st.markdown(f'<div class="package-highlight">{sections["package"].strip()}</div>', unsafe_allow_html=True)
+                            st.markdown("<br>", unsafe_allow_html=True)
                             
-                            if sections['benefits']:
-                                st.markdown("**✨ Key Benefits**")
-                                st.success(sections['benefits'].strip())
+                            # AI Response with better formatting
+                            ai_text = response['answer']
                             
-                            if sections['pricing']:
-                                st.markdown("**💰 Pricing Strategy**")
-                                st.warning(sections['pricing'].strip())
-                        else:
-                            # Fallback: display full text nicely
-                            st.markdown("**📋 Analysis & Recommendation**")
-                            st.markdown(ai_text)
-                        
-                        st.markdown('</div>', unsafe_allow_html=True)
+                            # Try to parse sections (if AI follows format)
+                            sections = {
+                                'profile': '',
+                                'package': '',
+                                'benefits': '',
+                                'pricing': ''
+                            }
+                            
+                            # Simple parsing (fallback to full text if not structured)
+                            if any(keyword in ai_text.lower() for keyword in ['usage profile', 'recommended package', 'benefits', 'pricing']):
+                                # Structured response
+                                lines = ai_text.split('\n')
+                                current_section = None
+                                for line in lines:
+                                    line_lower = line.lower()
+                                    if 'usage profile' in line_lower or 'profile' in line_lower and len(line) < 50:
+                                        current_section = 'profile'
+                                    elif 'recommended package' in line_lower or 'package' in line_lower and len(line) < 50:
+                                        current_section = 'package'
+                                    elif 'benefit' in line_lower and len(line) < 50:
+                                        current_section = 'benefits'
+                                    elif 'pricing' in line_lower and len(line) < 50:
+                                        current_section = 'pricing'
+                                    elif current_section and line.strip():
+                                        sections[current_section] += line + '\n'
+                                
+                                # Display structured
+                                if sections['profile']:
+                                    st.markdown("**📋 Usage Profile Analysis**")
+                                    st.info(sections['profile'].strip())
+                                
+                                if sections['package']:
+                                    st.markdown("**🎁 Recommended Package**")
+                                    st.markdown(f'<div class="package-highlight">{sections["package"].strip()}</div>', unsafe_allow_html=True)
+                                
+                                if sections['benefits']:
+                                    st.markdown("**✨ Key Benefits**")
+                                    st.success(sections['benefits'].strip())
+                                
+                                if sections['pricing']:
+                                    st.markdown("**💰 Pricing Strategy**")
+                                    st.warning(sections['pricing'].strip())
+                            else:
+                                # Fallback: display full text nicely
+                                st.markdown("**📋 Analysis & Recommendation**")
+                                st.markdown(ai_text)
+                            
+                            st.markdown('</div>', unsafe_allow_html=True)
+                    else:
+                        st.warning("⚠️ Unable to generate AI recommendation. Please try again later.")
             
             else:
                 st.error(f"❌ Customer {customer_id} not found")