UI v30

app.py

@@ -12,14 +12,12 @@ try:
 except ImportError:
     print("Error: Could not import VectorDatabase from vector_db.py.")
     print("Please ensure vector_db.py exists in the same directory and is correctly defined.")
-    # Exit if critical dependency is missing at import time
     exit(1)
 
 try:
     from langchain_openai import ChatOpenAI
 except ImportError:
     print("Error: langchain-openai not found. Please install it: pip install langchain-openai")
-    # Exit if critical dependency is missing at import time
     exit(1)
 
 from langchain.prompts import PromptTemplate
@@ -184,7 +182,7 @@ Answer:"""
             error_message = "Error: AI answer generation failed."
             details = f"Details: {str(e)}"
             if "authentication" in str(e).lower():
-                error_message = "Error: OpenAI API Key is invalid or expired. Please check your key."
+                error_message = "Error: Authentication failed. Please double-check your OpenAI API key."
                 details = ""
             elif "rate limit" in str(e).lower():
                 error_message = "Error: You've exceeded your OpenAI API rate limit or quota. Please check your usage and plan limits, or wait and try again."
@@ -224,7 +222,6 @@ Answer:"""
             raise FileNotFoundError(f"PDF file not found: {pdf_path}")
         try:
             logging.info(f"Attempting to load/verify data from PDF: {pdf_path}")
-            # Assuming process_and_load_pdf is part of VectorDatabase and correctly implemented
             num_states_processed = self.vector_db.process_and_load_pdf(pdf_path)
             doc_count = self.vector_db.document_collection.count()
             state_count = self.vector_db.state_collection.count()
@@ -242,10 +239,9 @@ Answer:"""
             logging.error(f"Failed to load or process PDF '{pdf_path}': {str(e)}", exc_info=True)
             raise RuntimeError(f"Failed to process PDF '{pdf_path}': {e}") from e
 
-    # --- GRADIO INTERFACE (NEW UI DESIGN) ---
+    # --- GRADIO INTERFACE (NEW UI DESIGN BASED ON IMAGE) ---
     def gradio_interface(self):
         def query_interface_wrapper(api_key: str, query: str, state: str) -> str:
-            # Basic client-side validation for immediate feedback (redundant but good UX)
             if not api_key or not api_key.strip() or not api_key.startswith("sk-"):
                 return "<div class='error-message'><span class='error-icon'>⚠️</span>Please provide a valid OpenAI API key (starting with 'sk-'). <a href='https://platform.openai.com/api-keys' target='_blank'>Get one free from OpenAI</a>.</div>"
             if not state or state == "Select a state..." or "Error" in state:
@@ -253,29 +249,23 @@ Answer:"""
             if not query or not query.strip():
                 return "<div class='error-message'><span class='error-icon'>⚠️</span>Please enter your question in the text box.</div>"
 
-            # Call the core processing logic
             result = self.process_query(query=query, state=state, openai_api_key=api_key)
             answer = result.get("answer", "<div class='error-message'><span class='error-icon'>⚠️</span>An unexpected error occurred.</div>")
 
-            # Check if the answer already contains an error message (from deeper within process_query)
             if "<div class='error-message'>" in answer:
-                # Error messages are returned directly as they contain their own styling
                 return answer
             else:
-                # Wrap successful response in a div with an animation class
                 formatted_response_content = f"<div class='response-header'><span class='response-icon'>📜</span>Response for {state}</div><hr class='divider'>{answer}"
                 return f"<div class='animated-output-content'>{formatted_response_content}</div>"
 
         try:
             available_states_list = self.get_states()
-            # Ensure "Select a state..." is always the first option
             dropdown_choices = ["Select a state..."] + (available_states_list if available_states_list and "Error" not in available_states_list[0] else ["Error: States unavailable"])
-            initial_value = dropdown_choices[0] # Set initial value to the prompt
-        except Exception: # Catch-all for safety
+            initial_value = dropdown_choices[0]
+        except Exception:
             dropdown_choices = ["Error: Critical failure loading states"]
             initial_value = dropdown_choices[0]
 
-        # Define example queries, filtering based on available states
         example_queries_base = [
             ["What are the rules for security deposit returns?", "California"],
             ["Can a landlord enter my apartment without notice?", "New York"],
@@ -286,315 +276,294 @@ Answer:"""
         example_queries = []
         if available_states_list and "Error" not in available_states_list[0] and len(available_states_list) > 0:
             loaded_states_set = set(available_states_list)
-            # Filter for examples whose state is in the loaded states
             example_queries = [ex for ex in example_queries_base if ex[1] in loaded_states_set]
-            # Add a generic example if no specific state examples match or if list is empty
             if not example_queries:
-                # Add one example using the first available state, or a common one if no states
                 example_queries.append(["What basic rights do tenants have?", available_states_list[0] if available_states_list else "California"])
-        else: # Fallback if states list is problematic (e.g., empty or error)
+        else:
             example_queries.append(["What basic rights do tenants have?", "California"])
 
 
-        # Custom CSS for better UI design, clear boundaries, and text alignment for HuggingFace
+        # Custom CSS for the dark theme based on the provided image
         custom_css = """
         /* Import legible fonts from Google Fonts */
         @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&family=Poppins:wght@600;700;800&display=swap');
         
-        /* Root variables for consistent theming - adjusted for very light calm_seafoam feel */
+        /* Root variables for consistent theming - based on image's dark theme */
         :root {
-            --primary-color: #3cb371; /* Medium Sea Green (vibrant seafoam) */
-            --primary-hover: #339966;
-            --background-primary: hsl(180, 100%, 98%); /* Very light seafoam for main cards */
-            --background-secondary: hsl(180, 100%, 96%); /* Slightly darker for overall app background */
-            --text-primary: hsl(210, 20%, 20%); /* Dark blue-gray for main text */
-            --text-secondary: hsl(210, 10%, 45%); /* Muted blue-gray for secondary text */
-            --border-color: hsl(180, 30%, 85%); /* Subtle seafoam gray for borders */
-            --border-focus: #3cb371; /* Focus color matches primary */
-            --shadow-sm: 0 1px 3px rgba(0,0,0,0.08);
-            --shadow-md: 0 4px 10px rgba(0,0,0,0.1);
-            --shadow-lg: 0 10px 20px rgba(0,0,0,0.15);
-            --error-bg: #FFEBEB;
-            --error-border: #FFCACA;
-            --error-text: #D32F2F;
-            --header-bg-color: hsl(180, 70%, 92%); /* Distinct light seafoam for header */
+            --primary-color: hsl(150, 60%, 50%); /* Vibrant Seafoam Green for highlights */
+            --primary-hover: hsl(150, 60%, 45%);
+            
+            --background-app: hsl(210, 15%, 15%); /* Dark charcoal/navy for overall background */
+            --background-card: hsl(210, 30%, 20%); /* Deep blue for content cards */
+            --background-input-output: hsl(210, 25%, 28%); /* Slightly lighter blue for input/output elements */
+            
+            --text-light: hsl(0, 0%, 95%); /* Very light text for dark backgrounds */
+            --text-muted-light: hsl(210, 10%, 70%); /* Muted light text for secondary info */
+            
+            --border-light: hsl(210, 10%, 65%); /* Light gray for borders */
+            --border-input: hsl(210, 15%, 45%); /* Mid-tone blue-gray for input borders */
+            
+            --shadow-sm: 0 1px 3px rgba(0,0,0,0.15);
+            --shadow-md: 0 4px 10px rgba(0,0,0,0.3);
+            --shadow-lg: 0 10px 20px rgba(0,0,0,0.4);
+            
+            --error-bg: hsl(0, 70%, 20%); /* Dark red for error background */
+            --error-border: hsl(0, 70%, 35%);
+            --error-text: hsl(0, 90%, 80%); /* Light red for error text */
+            
+            --button-secondary-bg: hsl(0, 0%, 85%); /* Light gray for secondary button */
+            --button-secondary-text: hsl(210, 30%, 20%); /* Dark blue for secondary button text */
         }
         
-        /* Dark mode variables - for consistency if a dark mode toggle were present */
-        body.dark {
-            --background-primary: #1F303A; /* Dark blue-green */
-            --background-secondary: #2C404B;
-            --text-primary: #E0F2F1;
-            --text-secondary: #A7C5C8;
-            --border-color: #5F7C8A;
-            --primary-color: #66BB6A; /* Brighter green for dark mode */
-            --primary-hover: #5cb85f;
-            --error-bg: #3F1D1D;
-            --error-border: #5A1A1A;
-            --error-text: #FF7070;
-            --header-bg-color: #253842; /* Darker header for dark mode */
-        }
-
-        /* Base container improvements */
+        /* Body and Gradio Container */
+        body {
+            background-color: var(--background-app) !important;
+        }
         .gradio-container {
-            max-width: 900px !important; /* Slightly smaller for focused content */
-            margin: 0 auto !important; /* Center the whole app */
+            max-width: 900px !important;
+            margin: 0 auto !important;
             padding: 1.5rem !important;
             font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
-            background-color: var(--background-secondary) !important; /* Overall background */
-            box-shadow: none !important; /* Remove default gradio container shadow */
-        }
-        /* Ensure all main content sections have primary background */
-        .main-dashboard-container > * {
-            background-color: var(--background-primary) !important;
+            background-color: var(--background-app) !important; /* Overall background */
+            box-shadow: none !important;
+            border: 2px solid var(--border-light) !important; /* Outer border as per image */
+            border-radius: 16px !important;
         }
 
-        /* Header styling - centered and prominent */
+        /* Header Styling */
         .app-header-wrapper {
-            background: var(--header-bg-color) !important; /* Use new distinct header background color */
-            border: 2px solid var(--border-color) !important;
+            background: var(--background-card) !important; /* Consistent dark blue with other cards */
+            border: 2px solid var(--border-light) !important;
             border-radius: 16px !important;
-            padding: 2.5rem 1.5rem !important; /* More vertical padding */
+            padding: 2.5rem 1.5rem !important;
             margin-bottom: 1.5rem !important;
-            text-align: center !important; /* Center text within header */
+            text-align: center !important;
             box-shadow: var(--shadow-md) !important;
-            position: relative; /* For potential pseudo-element effects */
-            overflow: hidden; /* For any overflow animations */
+            position: relative;
+            overflow: hidden;
+            color: var(--text-light) !important; /* Ensure header text is light */
         }
-        .app-header-wrapper::before { /* Subtle background pattern for dynamism */
+        .app-header-wrapper::before { /* Subtle background pattern */
             content: '';
             position: absolute;
             top: 0;
             left: 0;
             width: 100%;
             height: 100%;
-            background: radial-gradient(circle at top left, rgba(60,179,113,0.05) 0%, transparent 40%),
-                        radial-gradient(circle at bottom right, rgba(60,179,113,0.05) 0%, transparent 40%);
+            background: radial-gradient(circle at top left, rgba(60,179,113,0.08) 0%, transparent 40%),
+                        radial-gradient(circle at bottom right, rgba(60,179,113,0.08) 0%, transparent 40%);
             z-index: 0;
             opacity: 0.8;
             pointer-events: none;
         }
-
         .app-header-logo {
             font-size: 4.5rem !important; /* Larger icon */
             margin-bottom: 0.75rem !important;
             display: block !important;
-            color: var(--primary-color) !important; /* Theme color */
+            color: var(--border-light) !important; /* Light gray for icon as per image */
             position: relative;
-            z-index: 1; /* Bring icon to front of pseudo-element */
-            /* Animation for dynamism */
+            z-index: 1;
             animation: float-icon 3s ease-in-out infinite alternate;
         }
-        /* Keyframes for floating icon */
         @keyframes float-icon {
             0% { transform: translateY(0px); }
             50% { transform: translateY(-5px); }
             100% { transform: translateY(0px); }
         }
-
         .app-header-title {
             font-family: 'Poppins', sans-serif !important;
-            font-size: 3rem !important; /* Even larger title */
-            font-weight: 800 !important; /* Bolder */
-            color: var(--text-primary) !important;
+            font-size: 3rem !important;
+            font-weight: 800 !important;
+            color: var(--text-light) !important; /* White for title */
             margin: 0 0 0.75rem 0 !important;
             line-height: 1.1 !important;
-            letter-spacing: -0.03em !important; /* Tighter spacing */
+            letter-spacing: -0.03em !important;
             position: relative;
             z-index: 1;
         }
         .app-header-tagline {
-            font-size: 1.25rem !important; /* Slightly larger tagline */
-            color: var(--text-secondary) !important;
+            font-size: 1.25rem !important;
+            color: var(--text-light) !important; /* White for tagline */
             font-weight: 400 !important;
             margin: 0 !important;
-            max-width: 700px; /* Constrain tagline width */
+            max-width: 700px;
             margin-left: auto;
             margin-right: auto;
             position: relative;
             z-index: 1;
         }
 
-        /* Main container with consistent spacing */
+        /* Main Dashboard Container & Card Sections */
         .main-dashboard-container {
             display: flex !important;
             flex-direction: column !important;
-            gap: 1.25rem !important; /* Consistent spacing between cards */
+            gap: 1.25rem !important;
         }
-        /* Card sections with clear boundaries (boundeyes) and subtle dynamic effects */
         .dashboard-card-section {
-            background: var(--background-primary) !important;
-            border: 2px solid var(--border-color) !important; /* Distinct border */
+            background: var(--background-card) !important; /* Deep blue for content cards */
+            border: 2px solid var(--border-light) !important;
             border-radius: 12px !important;
-            padding: 1.75rem !important; /* Consistent padding */
-            box-shadow: var(--shadow-sm) !important; /* Subtle shadow */
-            transition: all 0.3s ease-out !important; /* Smoother transition */
-            cursor: default; /* Indicate not directly clickable (unless examples) */
+            padding: 1.75rem !important;
+            box-shadow: var(--shadow-sm) !important;
+            transition: all 0.3s ease-out !important;
+            cursor: default;
+            color: var(--text-light) !important; /* Ensure card content text is light */
         }
         .dashboard-card-section:hover {
             box-shadow: var(--shadow-md) !important;
-            transform: translateY(-3px) !important; /* More pronounced lift */
+            transform: translateY(-3px) !important;
         }
 
-        /* Centered section titles with improved typography and consistent line */
+        /* Section Titles (e.g., "Welcome & Disclaimer") */
         .sub-section-title {
             font-family: 'Poppins', sans-serif !important;
-            font-size: 1.7rem !important; /* Slightly larger */
-            font-weight: 700 !important; /* Bolder */
-            color: var(--text-primary) !important;
-            text-align: center !important; /* Centered text */
-            margin: 0 0 1.25rem 0 !important; /* More space below title */
+            font-size: 1.7rem !important;
+            font-weight: 700 !important;
+            color: var(--text-light) !important; /* White for titles */
+            text-align: left !important; /* LEFT-ALIGNED as per image */
+            margin: 0 0 1.25rem 0 !important;
             padding-bottom: 0.75rem !important;
-            border-bottom: 2px solid var(--border-color) !important; /* Underline effect (the "line") */
+            border-bottom: 1px dashed rgba(255, 255, 255, 0.15) !important; /* Subtle light dashed line as separator */
             display: block !important;
             letter-spacing: -0.01em !important;
         }
-        
-        /* Specific styling for the markdown content within cards */
         .dashboard-card-section p {
             line-height: 1.7 !important;
-            color: var(--text-primary) !important;
+            color: var(--text-light) !important;
             font-size: 1rem !important;
         }
         .dashboard-card-section strong {
-            color: var(--primary-color) !important; /* Highlight strong text with primary color */
+            color: var(--primary-color) !important; /* Green for strong text */
         }
 
-        /* Improved input styling with clear boundaries and focus */
-        .gradio-textbox, .gradio-dropdown {
-            margin-bottom: 0.75rem !important;
-        }
+        /* Input Styling */
         .gradio-textbox textarea,
         .gradio-textbox input,
-        .gradio-dropdown > div > input[type="text"], /* Target dropdown input for custom values */
-        .gradio-dropdown .primary-wrap, /* Target dropdown wrapper */
-        .gradio-dropdown .scroll-hide /* Target dropdown list container for consistency */
-        {
-            background: var(--background-primary) !important;
-            border: 2px solid var(--border-color) !important; /* Clear border */
+        .gradio-dropdown > div > input[type="text"],
+        .gradio-dropdown .primary-wrap,
+        .gradio-dropdown .scroll-hide {
+            background: var(--background-input-output) !important; /* Lighter blue for inputs */
+            border: 2px solid var(--border-input) !important; /* Mid-tone blue-gray border */
             border-radius: 8px !important;
-            padding: 0.85rem 1rem !important; /* Slightly more padding */
+            padding: 0.85rem 1rem !important;
             font-size: 0.98rem !important;
             font-family: 'Inter', sans-serif !important;
-            color: var(--text-primary) !important;
-            transition: border-color 0.2s ease, box-shadow 0.2s ease !important; /* Smooth transitions */
+            color: var(--text-light) !important; /* White text for inputs */
+            transition: border-color 0.2s ease, box-shadow 0.2s ease !important;
             box-shadow: var(--shadow-sm) !important;
         }
+        .gradio-textbox textarea::placeholder,
+        .gradio-textbox input::placeholder {
+            color: var(--text-muted-light) !important; /* Muted light placeholder text */
+            opacity: 0.7;
+        }
         .gradio-textbox textarea:focus,
         .gradio-textbox input:focus,
         .gradio-dropdown > div > input[type="text"]:focus,
-        .gradio-dropdown .primary-wrap.focused { /* Apply focus style to dropdown wrap */
+        .gradio-dropdown .primary-wrap.focused {
             outline: none !important;
-            border-color: var(--border-focus) !important; /* Distinct border on focus (green) */
-            box-shadow: 0 0 0 4px rgba(60, 179, 113, 0.2) !important; /* Broader, softer glow on focus */
+            border-color: var(--primary-color) !important; /* Green border on focus */
+            box-shadow: 0 0 0 4px rgba(60, 179, 113, 0.3) !important; /* Green glow on focus */
         }
-
-        /* Label styling for better readability */
         .gradio-textbox label,
         .gradio-dropdown label {
-            font-weight: 600 !important; /* Bolder labels */
-            color: var(--text-primary) !important;
+            font-weight: 600 !important;
+            color: var(--text-light) !important; /* White labels */
             font-size: 1rem !important;
             margin-bottom: 0.6rem !important;
             display: block !important;
         }
-        /* Info text styling below inputs */
         .gradio-textbox .gr-form,
         .gradio-dropdown .gr-form {
             font-size: 0.9rem !important;
-            color: var(--text-secondary) !important;
-            margin-top: 0.4rem !important; /* More space for info text */
+            color: var(--text-muted-light) !important; /* Muted light info text */
+            margin-top: 0.4rem !important;
         }
-        /* Input row layout improvements */
         .input-row {
             display: flex !important;
-            gap: 1.25rem !important; /* Consistent gap between query and state */
+            gap: 1.25rem !important;
             margin-bottom: 0.5rem !important;
         }
         .input-field {
             flex: 1 !important;
         }
 
-        /* Button styling improvements with active state for dynamism */
+        /* Button Styling */
         .button-row {
             display: flex !important;
             gap: 1rem !important;
-            justify-content: flex-end !important; /* Align buttons to the right */
-            margin-top: 1.5rem !important; /* More space above buttons */
+            justify-content: flex-end !important;
+            margin-top: 1.5rem !important;
         }
         .gradio-button {
-            padding: 0.85rem 1.8rem !important; /* More padding for bigger buttons */
-            border-radius: 9px !important; /* Slightly more rounded */
-            font-weight: 600 !important; /* Bolder text */
+            padding: 0.85rem 1.8rem !important;
+            border-radius: 9px !important;
+            font-weight: 600 !important;
             font-size: 1rem !important;
-            transition: all 0.2s ease-out !important; /* Smooth transition for hover/active */
+            transition: all 0.2s ease-out !important;
             cursor: pointer !important;
             border: 2px solid transparent !important;
-            text-align: center !important; /* Ensure button text is centered */
+            text-align: center !important;
         }
         .gr-button-primary {
-            background: var(--primary-color) !important;
+            background: var(--primary-color) !important; /* Green button */
             color: white !important;
             box-shadow: var(--shadow-sm) !important;
         }
         .gr-button-primary:hover {
             background: var(--primary-hover) !important;
             box-shadow: var(--shadow-md) !important;
-            transform: translateY(-2px) !important; /* Subtle lift effect on hover */
+            transform: translateY(-2px) !important;
         }
-        .gr-button-primary:active { /* Press down effect on click */
+        .gr-button-primary:active {
             transform: translateY(1px) !important;
             box-shadow: none !important;
         }
         .gr-button-secondary {
-            background: transparent !important;
-            color: var(--text-primary) !important;
-            border-color: var(--border-color) !important;
+            background: var(--button-secondary-bg) !important; /* Light gray button */
+            color: var(--button-secondary-text) !important; /* Dark text */
+            border-color: var(--border-light) !important;
         }
         .gr-button-secondary:hover {
-            background: var(--background-secondary) !important;
+            background: hsl(0, 0%, 75%) !important; /* Darker gray on hover */
             border-color: var(--primary-color) !important;
             transform: translateY(-2px) !important;
         }
-        .gr-button-secondary:active { /* Press down effect on click */
+        .gr-button-secondary:active {
             transform: translateY(1px) !important;
             box-shadow: none !important;
         }
 
-        /* Output styling with clear boundaries (boundeyes are clear) and dynamic fade-in */
+        /* Output Styling */
         .output-content-wrapper {
-            background: var(--background-primary) !important;
-            border: 2px solid var(--border-color) !important; /* Clear border */
+            background: var(--background-input-output) !important; /* Lighter blue for output box */
+            border: 2px dashed var(--border-input) !important; /* Dashed border as per image */
             border-radius: 8px !important;
             padding: 1.5rem !important;
-            min-height: 150px !important; /* More space for output */
-            color: var(--text-primary) !important;
-            /* Ensure the inner animated content fits well */
+            min-height: 150px !important;
+            color: var(--text-light) !important; /* White text for output */
             display: flex;
             flex-direction: column;
-            justify-content: center; /* Center content vertically if small */
-            align-items: center; /* Center content horizontally if small */
+            justify-content: center;
+            align-items: center;
         }
-        /* The div holding the actual response content, enabling fade-in animation */
         .animated-output-content {
             opacity: 0;
-            animation: fadeInAndSlideUp 0.7s ease-out forwards; /* More pronounced animation */
-            width: 100%; /* Take full width of parent */
-            /* Preserve formatting within the animated content */
+            animation: fadeInAndSlideUp 0.7s ease-out forwards;
+            width: 100%;
             white-space: pre-wrap;
             overflow-wrap: break-word;
             word-break: break-word;
-            text-align: left !important; /* Ensure text is left-aligned within this div */
+            text-align: left !important;
         }
         @keyframes fadeInAndSlideUp {
             from { opacity: 0; transform: translateY(15px); }
             to { opacity: 1; transform: translateY(0); }
         }
-
         .response-header {
             font-size: 1.3rem !important;
             font-weight: 700 !important;
-            color: var(--primary-color) !important; /* Matches primary color */
+            color: var(--primary-color) !important;
             margin-bottom: 0.75rem !important;
             display: flex !important;
             align-items: center !important;
@@ -606,10 +575,9 @@ Answer:"""
         }
         .divider {
             border: none !important;
-            border-top: 1px dashed var(--border-color) !important; /* Dashed divider for visual separation */
+            border-top: 1px dashed rgba(255, 255, 255, 0.15) !important; /* Light dashed divider */
             margin: 1rem 0 !important;
         }
-        /* Error message styling */
         .error-message {
             background: var(--error-bg) !important;
             border: 2px solid var(--error-border) !important;
@@ -622,9 +590,9 @@ Answer:"""
             font-size: 0.95rem !important;
             font-weight: 500 !important;
             line-height: 1.6 !important;
-            text-align: left !important; /* Ensure error message text is left aligned */
-            width: 100%; /* Take full width of parent */
-            box-sizing: border-box; /* Include padding/border in width */
+            text-align: left !important;
+            width: 100%;
+            box-sizing: border-box;
         }
         .error-message a {
             color: var(--error-text) !important;
@@ -641,23 +609,22 @@ Answer:"""
             margin-top: 0.5rem !important;
             opacity: 0.8;
         }
-        /* Placeholder styling for empty output */
         .placeholder {
-            background: var(--background-secondary) !important;
-            border: 2px dashed var(--border-color) !important;
+            background: var(--background-input-output) !important; /* Lighter blue for placeholder */
+            border: 2px dashed var(--border-input) !important; /* Dashed border */
             border-radius: 8px !important;
             padding: 2.5rem 1.5rem !important;
             text-align: center !important;
-            color: var(--text-secondary) !important;
+            color: var(--text-muted-light) !important; /* Muted light text */
             font-style: italic !important;
             font-size: 1.1rem !important;
-            width: 100%; /* Ensure it takes full width of parent */
-            box-sizing: border-box; /* Include padding/border in width */
+            width: 100%;
+            box-sizing: border-box;
         }
 
-        /* Examples table styling with dynamic hover */
+        /* Examples Table Styling */
         .examples-section .gr-samples-table {
-            border: 2px solid var(--border-color) !important;
+            border: 2px solid var(--border-light) !important;
             border-radius: 8px !important;
             overflow: hidden !important;
             margin-top: 1rem !important;
@@ -667,44 +634,43 @@ Answer:"""
             padding: 0.9rem !important;
             border: none !important;
             font-size: 0.95rem !important;
-            text-align: left !important; /* Ensure example text is left-aligned */
+            text-align: left !important;
+            color: var(--text-light) !important; /* White text for examples */
         }
         .examples-section .gr-samples-table th {
-            background: var(--background-secondary) !important;
+            background: var(--background-card) !important; /* Deep blue for example header */
             font-weight: 700 !important;
-            color: var(--text-primary) !important;
         }
         .examples-section .gr-samples-table td {
-            background: var(--background-primary) !important;
-            color: var(--text-primary) !important;
-            border-top: 1px solid var(--border-color) !important;
+            background: var(--background-input-output) !important; /* Lighter blue for example rows */
+            border-top: 1px solid var(--border-input) !important; /* Mid-tone blue-gray border */
             cursor: pointer !important;
-            transition: background 0.2s ease, transform 0.1s ease !important; /* Smooth transitions */
+            transition: background 0.2s ease, transform 0.1s ease !important;
         }
         .examples-section .gr-samples-table tr:hover td {
-            background: var(--background-secondary) !important;
-            transform: translateX(5px); /* Subtle slide on hover */
+            background: hsl(210, 28%, 32%) !important; /* Slightly darker light blue on hover */
+            transform: translateX(5px);
         }
-        /* Hide Gradio default elements for examples for cleaner look */
         .gr-examples .gr-label,
         .gr-examples .label-wrap,
         .gr-examples .gr-accordion-header {
             display: none !important;
         }
         
-        /* Footer styling - centered text and green links */
+        /* Footer Styling */
         .app-footer-wrapper {
-            background: var(--background-secondary) !important;
-            border: 2px solid var(--border-color) !important;
+            background: var(--background-card) !important; /* Deep blue for footer */
+            border: 2px solid var(--border-light) !important;
             border-radius: 12px !important;
             padding: 1.75rem !important;
             margin-top: 1.5rem !important;
-            text-align: center !important; /* Centered footer text */
+            text-align: center !important;
+            color: var(--text-light) !important; /* White text for footer */
         }
         .app-footer p {
             margin: 0.6rem 0 !important;
             font-size: 0.95rem !important;
-            color: var(--text-secondary) !important;
+            color: var(--text-muted-light) !important; /* Muted light text */
             line-height: 1.6 !important;
         }
         .app-footer a {
@@ -718,7 +684,7 @@ Answer:"""
             text-decoration-color: var(--primary-color) !important; /* Green underline on hover */
         }
 
-        /* Responsive design for smaller screens */
+        /* Responsive Design */
         @media (max-width: 768px) {
             .gradio-container {
                 padding: 1rem !important;
@@ -733,13 +699,13 @@ Answer:"""
                 font-size: 1.4rem !important;
             }
             .input-row {
-                flex-direction: column !important; /* Stack inputs vertically */
+                flex-direction: column !important;
             }
             .button-row {
-                flex-direction: column !important; /* Stack buttons vertically */
+                flex-direction: column !important;
             }
             .gradio-button {
-                width: 100% !important; /* Full width buttons */
+                width: 100% !important;
             }
             .dashboard-card-section {
                 padding: 1.25rem !important;
@@ -754,11 +720,8 @@ Answer:"""
         }
         """
 
-        # Using gr.Blocks with the specified theme and custom CSS
         with gr.Blocks(theme="shivi/calm_seafoam", css=custom_css, title="Landlord-Tenant Rights Assistant") as demo:
-            # Header Section - uses gr.Group for distinct card-like styling
             with gr.Group(elem_classes="app-header-wrapper"):
-                # Markdown used for flexible styling and auto-centering via CSS
                 gr.Markdown(
                     """
                     <div class="app-header">
@@ -769,12 +732,10 @@ Answer:"""
                     """
                 )
 
-            # Main Dashboard Container - acts as a column to stack various sections
             with gr.Column(elem_classes="main-dashboard-container"):
 
-                # Introduction and Disclaimer Card
                 with gr.Group(elem_classes="dashboard-card-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Welcome & Disclaimer</h3>") # Centered by CSS
+                    gr.Markdown("<h3 class='sub-section-title'>Welcome & Disclaimer</h3>")
                     gr.Markdown(
                         """
                         Navigate landlord-tenant laws with ease. This assistant provides detailed, state-specific answers grounded in legal authority.
@@ -783,23 +744,21 @@ Answer:"""
                         """
                     )
 
-                # OpenAI API Key Input Card
                 with gr.Group(elem_classes="dashboard-card-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>OpenAI API Key</h3>") # Centered by CSS
+                    gr.Markdown("<h3 class='sub-section-title'>OpenAI API Key</h3>")
                     api_key_input = gr.Textbox(
                         label="API Key",
-                        type="password", # Hides the input for security
+                        type="password",
                         placeholder="Enter your OpenAI API key (e.g., sk-...)",
                         info="Required to process your query. Get one from OpenAI: platform.openai.com/api-keys",
                         lines=1,
-                        elem_classes=["input-field-group"] # Custom class for input styling
+                        elem_classes=["input-field-group"]
                     )
 
-                # Query Input and State Selection Card
                 with gr.Group(elem_classes="dashboard-card-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Ask Your Question</h3>") # Centered by CSS
-                    with gr.Row(elem_classes="input-row"): # Row for side-by-side query and state
-                        with gr.Column(elem_classes="input-field", scale=3): # Query text area takes more space
+                    gr.Markdown("<h3 class='sub-section-title'>Ask Your Question</h3>")
+                    with gr.Row(elem_classes="input-row"):
+                        with gr.Column(elem_classes="input-field", scale=3):
                             query_input = gr.Textbox(
                                 label="Your Question",
                                 placeholder="E.g., What are the rules for security deposit returns in my state?",
@@ -807,7 +766,7 @@ Answer:"""
                                 max_lines=8,
                                 elem_classes=["input-field-group"]
                             )
-                        with gr.Column(elem_classes="input-field", scale=1): # State dropdown takes less space
+                        with gr.Column(elem_classes="input-field", scale=1):
                             state_input = gr.Dropdown(
                                 label="Select State",
                                 choices=dropdown_choices,
@@ -815,32 +774,29 @@ Answer:"""
                                 allow_custom_value=False,
                                 elem_classes=["input-field-group"]
                             )
-                    with gr.Row(elem_classes="button-row"): # Row for action buttons
+                    with gr.Row(elem_classes="button-row"):
                         clear_button = gr.Button("Clear", variant="secondary", elem_classes=["gr-button-secondary"])
                         submit_button = gr.Button("Submit Query", variant="primary", elem_classes=["gr-button-primary"])
 
-                # Output Display Card - Using gr.HTML for better animation control
                 with gr.Group(elem_classes="dashboard-card-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Legal Assistant's Response</h3>") # Centered by CSS
-                    output = gr.HTML( # Changed to gr.HTML to wrap content with animation class
+                    gr.Markdown("<h3 class='sub-section-title'>Legal Assistant's Response</h3>")
+                    output = gr.HTML(
                         value="<div class='placeholder'>The answer will appear here after submitting your query.</div>",
-                        elem_classes="output-content-wrapper" # Custom class for output styling
+                        elem_classes="output-content-wrapper"
                     )
 
-                # Example Questions Section
                 with gr.Group(elem_classes="dashboard-card-section examples-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Example Questions</h3>") # Centered by CSS
+                    gr.Markdown("<h3 class='sub-section-title'>Example Questions</h3>")
                     if example_queries:
                         gr.Examples(
                             examples=example_queries,
                             inputs=[query_input, state_input],
                             examples_per_page=5,
-                            label="" # Hide default Gradio label for examples to use our custom title
+                            label=""
                         )
                     else:
                         gr.Markdown("<div class='placeholder'>Sample questions could not be loaded. Please ensure the vector database is populated.</div>")
 
-            # Footer Section - contains disclaimer and developer info
             with gr.Group(elem_classes="app-footer-wrapper"):
                 gr.Markdown(
                     """
@@ -849,20 +805,19 @@ Answer:"""
                     """
                 )
 
-            # Event Listeners for buttons
             submit_button.click(
                 fn=query_interface_wrapper,
                 inputs=[api_key_input, query_input, state_input],
                 outputs=output,
-                api_name="submit_query" # Useful for debugging / external calls
+                api_name="submit_query"
             )
             
             clear_button.click(
                 fn=lambda: (
-                    "", # Clear API key input
-                    "", # Clear query input
-                    initial_value, # Reset state dropdown to default prompt
-                    "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>" # Reset output message
+                    "",
+                    "",
+                    initial_value,
+                    "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>"
                 ),
                 inputs=[],
                 outputs=[api_key_input, query_input, state_input, output]
@@ -881,36 +836,32 @@ if __name__ == "__main__":
         PDF_PATH = os.getenv("PDF_PATH", DEFAULT_PDF_PATH)
         VECTOR_DB_PATH = os.getenv("VECTOR_DB_PATH", DEFAULT_DB_PATH)
 
-        # Ensure vector DB directory exists before initialization
         os.makedirs(os.path.dirname(VECTOR_DB_PATH), exist_ok=True)
 
         logging.info(f"Attempting to load PDF from: {PDF_PATH}")
         if not os.path.exists(PDF_PATH):
             logging.error(f"FATAL: PDF file not found at the specified path: {PDF_PATH}")
             print(f"\n--- CONFIGURATION ERROR ---\nPDF file ('{os.path.basename(PDF_PATH)}') not found at: {PDF_PATH}.\nPlease ensure it exists or set 'PDF_PATH' environment variable.\n---------------------------\n")
-            exit(1) # Correctly exits if PDF is not found
+            exit(1)
 
         if not os.access(PDF_PATH, os.R_OK):
             logging.error(f"FATAL: PDF file at '{PDF_PATH}' exists but is not readable. Check file permissions.")
             print(f"\n--- PERMISSION ERROR ---\nPDF file ('{os.path.basename(PDF_PATH)}') found but not readable at: {PDF_PATH}\nPlease check file permissions (e.g., using 'chmod +r' in terminal).\n---------------------------\n")
-            exit(1) # Correctly exits if PDF is unreadable
+            exit(1)
 
         logging.info(f"PDF file '{os.path.basename(PDF_PATH)}' found and is readable.")
 
-        # Initialize VectorDatabase and RAGSystem
         vector_db_instance = VectorDatabase(persist_directory=VECTOR_DB_PATH)
         rag = RAGSystem(vector_db=vector_db_instance)
 
-        # Load PDF data into the vector DB (or verify it's already loaded)
         rag.load_pdf(PDF_PATH)
 
-        # Get the Gradio interface object
         app_interface = rag.gradio_interface()
-        SERVER_PORT = int(os.getenv("PORT", 7860)) # Use PORT env var for Hugging Face Spaces
+        SERVER_PORT = int(os.getenv("PORT", 7860))
 
         logging.info(f"Launching Gradio app on http://0.0.0.0:{SERVER_PORT}")
         print(f"\n--- Gradio App Running ---\nAccess at: http://localhost:{SERVER_PORT} or your public Spaces URL\n--------------------------\n")
-        app_interface.launch(server_name="0.0.0.0", server_port=SERVER_PORT, share=False) # share=False is typical for Spaces
+        app_interface.launch(server_name="0.0.0.0", server_port=SERVER_PORT, share=False)
 
     except ModuleNotFoundError as e:
         if "vector_db" in str(e):