Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-# Business-Focused Streamlit Application for AI Talent Screening (FIXED)
+# Modern Dark Mode Streamlit Application for AI Talent Screening
 
 import streamlit as st
 from transformers import BertTokenizer, BertForSequenceClassification, T5Tokenizer, T5ForConditionalGeneration
@@ -13,65 +13,71 @@ from docx import Document
 import time
 import pandas as pd
 
-# Set page config with professional theme and wide layout
+# Set page config with modern dark theme and wide layout
 st.set_page_config(
     page_title="AI Talent Screening Tool", 
-    page_icon="💼", 
+    page_icon="🚀", 
     layout="wide",
     initial_sidebar_state="expanded",
 )
 
-# --- CUSTOM PROFESSIONAL CSS OVERHAUL (UNCHANGED) ---
+# --- CUSTOM MODERN DARK MODE CSS OVERHAUL ---
 st.markdown("""
     <style>
-    /* 0. GLOBAL CONFIG & LIGHT THEME */
+    /* 0. GLOBAL CONFIG & DARK THEME */
     :root {
-        --primary-color: #007BFF; /* Corporate Blue */
-        --success-color: #28A745; /* Corporate Green */
-        --warning-color: #FFC107; /* Corporate Yellow */
-        --danger-color: #DC3545; /* Corporate Red */
-        --background-color: #F8F9FA; /* Light Gray Background */
-        --container-background: #FFFFFF; /* White Container */
-        --text-color: #212529; /* Dark Text */
+        --primary-color: #9C27B0; /* Vibrant Purple/Magenta (Accent) */
+        --success-color: #4CAF50; /* Green (Good Match) */
+        --warning-color: #FFC107; /* Amber/Yellow (Review) */
+        --danger-color: #F44336; /* Red (Irrelevant/Error) */
+        --background-color: #1E1E1E; /* Deep Dark Background */
+        --container-background: #2D2D2D; /* Slightly Lighter Container */
+        --text-color: #F8F8F8; /* Light Text */
+        --secondary-text-color: #B0B0B0; /* Muted Light Gray */
     }
     
     .main {
         background-color: var(--background-color);
         color: var(--text-color);
-        font-family: 'Arial', sans-serif;
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+    }
+    
+    /* Overall Streamlit container background (Dark) */
+    .stApp {
+        background-color: var(--background-color);
     }
 
     /* 1. HEADER & TITLES */
     h1 {
         text-align: center;
         color: var(--primary-color);
-        font-size: 2.5em;
-        font-weight: 700;
-        border-bottom: 3px solid rgba(0, 123, 255, 0.2);
-        padding-bottom: 10px;
+        font-size: 2.8em;
+        font-weight: 800;
+        border-bottom: 3px solid rgba(156, 39, 176, 0.3);
+        padding-bottom: 15px;
         margin-bottom: 30px;
     }
     h2, h3, h4 {
         color: var(--text-color);
-        border-left: 5px solid var(--success-color); /* Green marker for clarity */
-        padding-left: 10px;
-        margin-top: 25px;
+        border-left: 5px solid var(--primary-color); /* Purple marker for clarity */
+        padding-left: 15px;
+        margin-top: 30px;
         font-weight: 600;
     }
     
     /* 2. BUTTONS & HOVER EFFECTS */
     .stButton>button {
-        color: var(--primary-color) !important;
-        border: 1px solid var(--primary-color) !important;
+        color: var(--text-color) !important;
+        border: 1px solid var(--container-background) !important;
         background-color: var(--container-background) !important;
-        border-radius: 8px; /* Rounded corners for friendly look */
+        border-radius: 12px; /* More rounded for modern look */
         transition: all 0.3s ease;
-        box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
+        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.3);
         font-weight: 600;
     }
     .stButton>button:hover {
-        background-color: rgba(0, 123, 255, 0.1) !important;
-        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.15); /* Subtle shadow lift */
+        background-color: #404040 !important;
+        box-shadow: 0 6px 15px rgba(0, 0, 0, 0.5); 
         transform: translateY(-2px);
     }
     /* Primary Button (Run Screening) */
@@ -81,70 +87,95 @@ st.markdown("""
         border-color: var(--success-color) !important;
     }
     .stButton>button[kind="primary"]:hover {
-        background-color: #1e7e34 !important; /* Darker green on hover */
-        border-color: #1e7e34 !important;
+        background-color: #388E3C !important; /* Darker green on hover */
+        border-color: #388E3C !important;
     }
 
-    /* 3. INPUTS, CONTAINERS & TABS */
+    /* 3. INPUTS, CONTAINERS, TABS & SIDEBAR */
     .stTextArea, .stTextInput, .stFileUploader {
         border-radius: 8px;
-        border: 1px solid #CED4DA;
-        background-color: var(--container-background);
-        box-shadow: inset 0 1px 2px rgba(0, 0, 0, 0.075);
+        border: 1px solid #444444;
+        background-color: #333333; /* Darker input background */
+        color: var(--text-color); /* Ensure input text is light */
+        box-shadow: inset 0 1px 2px rgba(0, 0, 0, 0.2);
     }
     .stTabs [aria-selected="true"] {
-        color: var(--success-color) !important;
-        border-bottom: 3px solid var(--success-color) !important;
+        color: var(--primary-color) !important;
+        border-bottom: 3px solid var(--primary-color) !important;
         font-weight: bold;
     }
     .stSidebar {
-        background-color: #E9ECEF; /* Sidebar background color matching light theme */
-        border-right: 1px solid #DEE2E6;
+        background-color: #252525; /* Slightly lighter dark sidebar for contrast */
+        border-right: 1px solid #3A3A3A;
+        color: var(--text-color);
     }
     
-    /* Custom Scorecard Style (Clean, White Tiles) */
+    /* Fix: Ensure text in sidebar expanders is visible */
+    [data-testid="stSidebar"] p, 
+    [data-testid="stSidebar"] li,
+    [data-testid="stSidebar"] [data-testid="stExpander"] {
+        color: var(--secondary-text-color) !important; 
+    }
+    
+    /* Custom Scorecard Style (Dark Tiles) */
     .scorecard-block {
-        border: 1px solid #DEE2E6; 
-        border-radius: 10px; /* Highly rounded for friendly look */
-        padding: 15px;
+        border: 1px solid #3A3A3A; 
+        border-radius: 12px; 
+        padding: 20px;
         margin: 5px 0;
-        background-color: var(--container-background);
+        background-color: #333333;
         transition: all 0.3s;
-        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.05);
+        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.2);
     }
     .scorecard-block:hover {
-        box-shadow: 0 6px 15px rgba(0, 0, 0, 0.1);
+        box-shadow: 0 6px 15px rgba(0, 0, 0, 0.4);
     }
     .scorecard-value {
-        font-size: 36px;
-        font-weight: 700;
+        font-size: 38px;
+        font-weight: 800;
         color: var(--primary-color);
     }
     .scorecard-label {
         font-size: 14px;
-        color: #6C757D;
+        color: var(--secondary-text-color);
     }
     /* Color override for specific blocks */
     .block-relevant { border-left: 5px solid var(--success-color); }
     .block-uncertain { border-left: 5px solid var(--warning-color); }
     .block-irrelevant { border-left: 5px solid var(--danger-color); }
 
-    /* Streamlit's Info/Success/Warning blocks made friendlier */
+    /* Streamlit's Info/Success/Warning blocks made darker */
     [data-testid="stAlert"] {
         border-radius: 8px;
         padding: 15px;
         font-size: 16px;
+        background-color: #333333 !important; /* Darker alert backgrounds */
+        color: var(--text-color) !important;
     }
-
+    /* Override specific alert colors for dark theme contrast */
+    [data-testid="stAlert"] div[role="alert"] {
+        background-color: #333333 !important;
+    }
+    [data-testid="stAlert"] div[role="alert"].st-emotion-cache-1f81d5m { /* Info */
+        border-left: 5px solid #2196F3;
+    }
+    [data-testid="stAlert"] div[role="alert"].st-emotion-cache-1218yph { /* Warning */
+        border-left: 5px solid var(--warning-color);
+    }
+    [data-testid="stAlert"] div[role="alert"].st-emotion-cache-22lkyf { /* Error */
+        border-left: 5px solid var(--danger-color);
+    }
+    [data-testid="stAlert"] div[role="alert"].st-emotion-cache-5lq06g { /* Success */
+        border-left: 5px solid var(--success-color);
+    }
+    
     </style>
 """, unsafe_allow_html=True)
 
-# --- (Model and Helper Functions - Core logic remains the same) ---
-# ... (skills_list, skills_pattern, extract_text_from_pdf/docx/file, normalize_text, check_experience_mismatch, validate_input, load_models, tokenize_inputs, extract_skills, classify_and_summarize_batch, generate_skill_pie_chart functions remain unchanged) ...
-
-# NOTE: Since the file content is large, I'm only including the modified function `render_sidebar`
-# and the affected part of `main` for brevity. The full code block at the end contains the complete, fixed file.
 
+# --- (Model and Helper Functions - Core logic remains the same) ---
+# NOTE: The core ML logic and utility functions for PDF/DOCX parsing remain unchanged
+# as they are robust and purely functional.
 skills_list = [
     'python', 'sql', 'c++', 'java', 'tableau', 'machine learning', 'data analysis',
     'business intelligence', 'r', 'tensorflow', 'pandas', 'spark', 'scikit-learn', 'aws',
@@ -181,7 +212,7 @@ def extract_text_from_docx(file):
             text += paragraph.text + "\n"
         return text.strip()
     except: return ""
-
+    
 def extract_text_from_file(uploaded_file):
     if uploaded_file.name.endswith('.pdf'): return extract_text_from_pdf(uploaded_file)
     elif uploaded_file.name.endswith('.docx'): return extract_text_from_docx(uploaded_file)
@@ -296,13 +327,13 @@ def classify_and_summarize_batch(resume, job_description, _bert_tok, _t5_input,
         elif detected_skills: final_summary = f"Key Skills: {', '.join(detected_skills)}"
         else: final_summary = f"Experience: {exp_match.group(0) if exp_match else 'Unknown'}"
         
-        if suitability == "Relevant": color = "#28A745"
-        elif suitability == "Irrelevant": color = "#DC3545"
+        if suitability == "Relevant": color = "#4CAF50"
+        elif suitability == "Irrelevant": color = "#F44336"
         else: color = "#FFC107"
 
         return {"Suitability": suitability, "Key Skills & Experience Summary": final_summary, "Flagging Reason": warning, "Suitability_Color": color}
     except Exception as e:
-        return {"Suitability": "Error", "Key Skills & Experience Summary": "Failed to process profile", "Flagging Reason": str(e), "Suitability_Color": "#DC3545"}
+        return {"Suitability": "Error", "Key Skills & Experience Summary": "Failed to process profile", "Flagging Reason": str(e), "Suitability_Color": "#F44336"}
 
 @st.cache_data
 def generate_skill_pie_chart(resumes):
@@ -326,29 +357,30 @@ def generate_skill_pie_chart(resumes):
     labels = list(top_skills.keys())
     sizes = [(count / sum(top_skills.values())) * 100 for count in top_skills.values()]
     
-    plt.style.use('default')
+    # Use dark theme settings for the chart
+    plt.style.use('dark_background')
     fig, ax = plt.subplots(figsize=(6, 4))
-    colors = plt.cm.tab10(np.linspace(0, 1, len(labels))) 
-    plt.rcParams['text.color'] = 'black'
-    wedges, texts, autotexts = ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors, textprops={'fontsize': 10, 'color': 'black'})
+    colors = plt.cm.magma(np.linspace(0.3, 0.9, len(labels))) # Vibrant color map for dark mode
+    plt.rcParams['text.color'] = '#F8F8F8'
+    wedges, texts, autotexts = ax.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors, textprops={'fontsize': 10, 'color': '#F8F8F8'})
     ax.axis('equal') 
-    plt.title("Top Candidate Skill Frequency", fontsize=14, color='#007BFF', pad=10)
+    plt.title("Top Candidate Skill Frequency", fontsize=14, color='#9C27B0', pad=10)
     return fig
 
 def render_sidebar():
-    """Render sidebar content with professional HR language. FIXED: Replaced st.get_style_color with hex codes."""
-    # Define hex colors to replace st.get_style_color() calls
-    SUCCESS_COLOR = "#28A745"  # Corporate Green
-    WARNING_COLOR = "#FFC107"  # Corporate Yellow
-    DANGER_COLOR = "#DC3545"   # Corporate Red
-    PRIMARY_COLOR = "#007BFF"  # Corporate Blue
+    """Render sidebar content with professional HR language."""
+    # Define hex colors
+    SUCCESS_COLOR = "#4CAF50"  
+    WARNING_COLOR = "#FFC107"  
+    DANGER_COLOR = "#F44336"   
+    PRIMARY_COLOR = "#9C27B0"  
 
     with st.sidebar:
         st.markdown(f"""
             <h2 style='text-align: center; border-left: none; padding-left: 0; color: {PRIMARY_COLOR};'>
-                Talent Screening Assistant
+                TALENT SCREENING ASSISTANT
             </h2>
-            <p style='text-align: center; font-size: 14px; margin-top: 0; color: #6C757D;'>
+            <p style='text-align: center; font-size: 14px; margin-top: 0; color: #B0B0B0;'>
                 Powered by Advanced NLP (BERT + T5)
             </p>
         """, unsafe_allow_html=True)
@@ -388,7 +420,7 @@ def main():
     if 'valid_resumes' not in st.session_state: st.session_state.valid_resumes = []
     if 'models' not in st.session_state: st.session_state.models = None
 
-    st.markdown("<h1>AI TALENT SCREENING TOOL</h1>", unsafe_allow_html=True)
+    st.markdown("<h1>🚀 AI TALENT SCREENING TOOL</h1>", unsafe_allow_html=True)
     
     # HR-friendly Tab Names
     tab_setup, tab_resumes, tab_results = st.tabs(["1. Job Requirement Setup", "2. Candidate Profile Upload", "3. Screening Report & Analytics"])
@@ -503,17 +535,8 @@ def main():
                 
                 for i, resume in enumerate(valid_resumes):
                     status_text.text(f"Status: Analyzing Profile {i+1} of {total_steps}...")
-                    
-                    # Create single-batch tensors for BERT and T5
-                    bert_tok_single = {
-                        'input_ids': bert_tokenized['input_ids'][i].unsqueeze(0), 
-                        'attention_mask': bert_tokenized['attention_mask'][i].unsqueeze(0)
-                    }
-                    t5_tok_single = {
-                        'input_ids': t5_tokenized['input_ids'][i].unsqueeze(0), 
-                        'attention_mask': t5_tokenized['attention_mask'][i].unsqueeze(0)
-                    }
-                    
+                    bert_tok_single = {'input_ids': bert_tokenized['input_ids'][i].unsqueeze(0), 'attention_mask': bert_tokenized['attention_mask'][i].unsqueeze(0)}
+                    t5_tok_single = {'input_ids': t5_tokenized['input_ids'][i].unsqueeze(0), 'attention_mask': t5_tokenized['attention_mask'][i].unsqueeze(0)}
                     result = classify_and_summarize_batch(resume, job_description, bert_tok_single, t5_inputs[i], t5_tok_single, job_skills_set)
                     result["Profile ID"] = f"Candidate {i+1}"
                     results.append(result)
@@ -541,10 +564,10 @@ def main():
             st.markdown(f"#### Overview: {total} Candidate Profiles Processed")
             
             # Define hex colors again for the scorecard blocks
-            PRIMARY_COLOR = "#007BFF"  # Corporate Blue
-            SUCCESS_COLOR = "#28A745"  # Corporate Green
-            WARNING_COLOR = "#FFC107"  # Corporate Yellow
-            DANGER_COLOR = "#DC3545"   # Corporate Red
+            PRIMARY_COLOR = "#9C27B0"  
+            SUCCESS_COLOR = "#4CAF50"  
+            WARNING_COLOR = "#FFC107"  
+            DANGER_COLOR = "#F44336"   
 
             col1, col2, col3, col4 = st.columns(4)