UI v19

app.py

@@ -5,7 +5,7 @@ from functools import lru_cache
 import re
 
 import gradio as gr
-import gradio.themes as themes # Import gradio.themes
+# import gradio.themes as themes # Not directly used for theme dropdown, but kept if you had other theme-related utilities.
 
 try:
     # Assuming vector_db.py exists in the same directory or is installed
@@ -13,30 +13,32 @@ 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
 from langchain.chains import LLMChain
 
-# Suppress warnings
+# Suppress warnings for cleaner console output
 import warnings
 warnings.filterwarnings("ignore", category=SyntaxWarning)
 warnings.filterwarnings("ignore", category=UserWarning, message=".*You are using gradio version.*")
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 
-# Enhanced logging
+# Enhanced logging configuration
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s'
 )
 
-# --- RAGSystem Class (Processing Logic - kept intact as requested) ---
+# --- RAGSystem Class (Processing Logic - KEPT INTACT AS REQUESTED) ---
 class RAGSystem:
     def __init__(self, vector_db: Optional[VectorDatabase] = None):
         logging.info("Initializing RAGSystem")
@@ -241,7 +243,7 @@ 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 (Refactored to use earneleh/paris theme) ---
+    # --- GRADIO INTERFACE (NEW UI DESIGN) ---
     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)
@@ -266,8 +268,9 @@ Answer:"""
 
         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]
+            initial_value = dropdown_choices[0] # Set initial value to the prompt
         except Exception: # Catch-all for safety
             dropdown_choices = ["Error: Critical failure loading states"]
             initial_value = dropdown_choices[0]
@@ -287,124 +290,158 @@ Answer:"""
             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
+        else: # Fallback if states list is problematic (e.g., empty or error)
             example_queries.append(["What basic rights do tenants have?", "California"])
 
-        # Improved Custom CSS for better UI design and HuggingFace compatibility
+
+        # Custom CSS for better UI design, clear boundaries, and text alignment for HuggingFace
+        # This CSS overrides Gradio defaults to create the desired look and feel.
         custom_css = """
-        /* Import legible fonts */
+        /* 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 */
+        /* Root variables for consistent theming - adjusted for calm_seafoam feel */
         :root {
-            --primary-color: #2563eb;
-            --primary-hover: #1d4ed8;
-            --background-primary: #ffffff;
-            --background-secondary: #f8fafc;
-            --text-primary: #1e293b;
-            --text-secondary: #64748b;
-            --border-color: #e2e8f0;
-            --border-focus: #2563eb;
-            --shadow-sm: 0 1px 2px 0 rgb(0 0 0 / 0.05);
-            --shadow-md: 0 4px 6px -1px rgb(0 0 0 / 0.1);
-            --shadow-lg: 0 10px 15px -3px rgb(0 0 0 / 0.1);
-            --error-bg: #fef2f2;
-            --error-border: #fecaca;
-            --error-text: #dc2626;
+            --primary-color: #4CAF50; /* A pleasant green */
+            --primary-hover: #45a049;
+            --background-primary: #F0F8F8; /* Light seafoam background */
+            --background-secondary: #E8F0F0; /* Slightly darker for contrast */
+            --text-primary: #2C3E50; /* Darker text for readability */
+            --text-secondary: #5F7C8A; /* Muted text */
+            --border-color: #B2D8D8; /* Seafoam border */
+            --border-focus: #4CAF50; /* 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;
         }
         
-        /* Dark mode variables */
-        @media (prefers-color-scheme: dark) {
-            :root {
-                --background-primary: #0f172a;
-                --background-secondary: #1e293b;
-                --text-primary: #f1f5f9;
-                --text-secondary: #94a3b8;
-                --border-color: #334155;
-                --error-bg: #1e1b1b;
-                --error-border: #451a1a;
-                --error-text: #f87171;
-            }
+        /* Dark mode variables - for consistency if a dark mode toggle were present */
+        /* calm_seafoam theme itself is light, but if a `dark` class is applied to body: */
+        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;
         }
+
         /* Base container improvements */
         .gradio-container {
-            max-width: 1000px !important;
-            margin: 0 auto !important;
-            padding: 1rem !important;
+            max-width: 900px !important; /* Slightly smaller for focused content */
+            margin: 0 auto !important; /* Center the whole app */
+            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;
         }
-        /* Header styling */
+
+        /* Header styling - centered and prominent */
         .app-header-wrapper {
             background: linear-gradient(135deg, var(--background-primary) 0%, var(--background-secondary) 100%) !important;
             border: 2px solid var(--border-color) !important;
             border-radius: 16px !important;
-            padding: 2rem !important;
+            padding: 2.5rem 1.5rem !important; /* More vertical padding */
             margin-bottom: 1.5rem !important;
-            text-align: center !important;
+            text-align: center !important; /* Center text within header */
             box-shadow: var(--shadow-md) !important;
         }
         .app-header-logo {
-            font-size: 3rem !important;
-            margin-bottom: 0.5rem !important;
+            font-size: 3.5rem !important; /* Larger icon */
+            margin-bottom: 0.75rem !important;
             display: block !important;
+            color: var(--primary-color) !important; /* Theme color */
         }
         .app-header-title {
             font-family: 'Poppins', sans-serif !important;
-            font-size: 2.5rem !important;
-            font-weight: 700 !important;
+            font-size: 2.8rem !important; /* Larger title */
+            font-weight: 800 !important; /* Bolder */
             color: var(--text-primary) !important;
-            margin: 0 0 0.5rem 0 !important;
-            line-height: 1.2 !important;
+            margin: 0 0 0.75rem 0 !important;
+            line-height: 1.1 !important;
+            letter-spacing: -0.02em !important; /* Slightly tighter spacing */
         }
         .app-header-tagline {
-            font-size: 1.1rem !important;
+            font-size: 1.2rem !important;
             color: var(--text-secondary) !important;
             font-weight: 400 !important;
             margin: 0 !important;
+            max-width: 700px; /* Constrain tagline width */
+            margin-left: auto;
+            margin-right: auto;
         }
-        /* Main container with compact spacing */
+
+        /* Main container with consistent spacing */
         .main-dashboard-container {
             display: flex !important;
             flex-direction: column !important;
-            gap: 1rem !important; /* Reduced gap for more compact look */
+            gap: 1.25rem !important; /* Consistent spacing between cards */
         }
-        /* Card sections with better boundaries */
+        /* Card sections with clear boundaries (boundeyes) */
         .dashboard-card-section {
             background: var(--background-primary) !important;
-            border: 2px solid var(--border-color) !important;
+            border: 2px solid var(--border-color) !important; /* Distinct border */
             border-radius: 12px !important;
-            padding: 1.5rem !important; /* Reduced padding for compactness */
-            box-shadow: var(--shadow-sm) !important;
+            padding: 1.75rem !important; /* Consistent padding */
+            box-shadow: var(--shadow-sm) !important; /* Subtle shadow */
             transition: all 0.2s ease !important;
         }
         .dashboard-card-section:hover {
-            box-shadow: var(--shadow-md) !important;
+            box-shadow: var(--shadow-md) !important; /* Enhanced shadow on hover */
         }
-        /* Centered section titles with better typography */
+
+        /* Centered section titles with improved typography */
         .sub-section-title {
             font-family: 'Poppins', sans-serif !important;
-            font-size: 1.5rem !important;
-            font-weight: 600 !important;
+            font-size: 1.6rem !important;
+            font-weight: 700 !important;
             color: var(--text-primary) !important;
-            text-align: center !important;
-            margin: 0 0 1rem 0 !important;
-            padding-bottom: 0.5rem !important;
-            border-bottom: 2px solid var(--border-color) !important;
+            text-align: center !important; /* Centered text */
+            margin: 0 0 1.25rem 0 !important; /* More space below title */
+            padding-bottom: 0.75rem !important;
+            border-bottom: 2px solid var(--border-color) !important; /* Underline effect */
             display: block !important;
+            letter-spacing: -0.01em !important;
+        }
+        
+        /* Specific styling for the welcome/disclaimer markdown content */
+        .dashboard-card-section p {
+            line-height: 1.7 !important;
+            color: var(--text-primary) !important;
+            font-size: 1rem !important;
+        }
+        .dashboard-card-section strong {
+            color: var(--primary-color) !important; /* Highlight strong text with primary color */
         }
-        /* Improved input styling with clear boundaries */
+
+        /* Improved input styling with clear boundaries and focus */
         .gradio-textbox, .gradio-dropdown {
-            margin-bottom: 0.75rem !important; /* Reduced margin for compactness */
+            margin-bottom: 0.75rem !important;
         }
         .gradio-textbox textarea,
         .gradio-textbox input,
-        .gradio-dropdown select {
+        .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;
+            border: 2px solid var(--border-color) !important; /* Clear border */
             border-radius: 8px !important;
-            padding: 0.75rem !important;
-            font-size: 0.95rem !important;
+            padding: 0.85rem 1rem !important; /* Slightly more padding */
+            font-size: 0.98rem !important;
             font-family: 'Inter', sans-serif !important;
             color: var(--text-primary) !important;
             transition: all 0.2s ease !important;
@@ -412,57 +449,55 @@ Answer:"""
         }
         .gradio-textbox textarea:focus,
         .gradio-textbox input:focus,
-        .gradio-dropdown select:focus {
+        .gradio-dropdown > div > input[type="text"]:focus,
+        .gradio-dropdown .primary-wrap.focused { /* Apply focus style to dropdown wrap */
             outline: none !important;
-            border-color: var(--border-focus) !important;
-            box-shadow: 0 0 0 3px rgba(37, 99, 235, 0.1) !important;
-        }
-        /* Placeholder text styling */
-        .gradio-textbox textarea::placeholder,
-        .gradio-textbox input::placeholder {
-            color: var(--text-secondary) !important;
-            opacity: 0.7 !important;
+            border-color: var(--border-focus) !important; /* Distinct border on focus */
+            box-shadow: 0 0 0 3px rgba(76, 175, 80, 0.15) !important; /* Soft glow on focus */
         }
+
         /* Label styling for better readability */
         .gradio-textbox label,
         .gradio-dropdown label {
-            font-weight: 500 !important;
+            font-weight: 600 !important; /* Bolder labels */
             color: var(--text-primary) !important;
-            font-size: 0.9rem !important;
-            margin-bottom: 0.5rem !important;
+            font-size: 1rem !important;
+            margin-bottom: 0.6rem !important;
             display: block !important;
         }
-        /* Info text styling */
+        /* Info text styling below inputs */
         .gradio-textbox .gr-form,
         .gradio-dropdown .gr-form {
-            font-size: 0.85rem !important;
+            font-size: 0.9rem !important;
             color: var(--text-secondary) !important;
-            margin-top: 0.25rem !important;
+            margin-top: 0.4rem !important; /* More space for info text */
         }
         /* Input row layout improvements */
         .input-row {
             display: flex !important;
-            gap: 1rem !important;
-            margin-bottom: 0.5rem !important; /* Reduced margin */
+            gap: 1.25rem !important; /* Consistent gap between query and state */
+            margin-bottom: 0.5rem !important;
         }
         .input-field {
             flex: 1 !important;
         }
+
         /* Button styling improvements */
         .button-row {
             display: flex !important;
             gap: 1rem !important;
-            justify-content: flex-end !important;
-            margin-top: 1rem !important;
+            justify-content: flex-end !important; /* Align buttons to the right */
+            margin-top: 1.5rem !important; /* More space above buttons */
         }
         .gradio-button {
-            padding: 0.75rem 1.5rem !important;
-            border-radius: 8px !important;
-            font-weight: 500 !important;
-            font-size: 0.9rem !important;
-            transition: all 0.2s ease !important;
+            padding: 0.85rem 1.8rem !important; /* More padding for bigger buttons */
+            border-radius: 9px !important; /* Slightly more rounded */
+            font-weight: 600 !important; /* Bolder text */
+            font-size: 1rem !important;
+            transition: all 0.25s ease !important;
             cursor: pointer !important;
             border: 2px solid transparent !important;
+            text-align: center !important; /* Ensure button text is centered */
         }
         .gr-button-primary {
             background: var(--primary-color) !important;
@@ -472,7 +507,7 @@ Answer:"""
         .gr-button-primary:hover {
             background: var(--primary-hover) !important;
             box-shadow: var(--shadow-md) !important;
-            transform: translateY(-1px) !important;
+            transform: translateY(-2px) !important; /* Subtle lift effect on hover */
         }
         .gr-button-secondary {
             background: transparent !important;
@@ -482,63 +517,83 @@ Answer:"""
         .gr-button-secondary:hover {
             background: var(--background-secondary) !important;
             border-color: var(--primary-color) !important;
+            transform: translateY(-2px) !important;
         }
-        /* Output styling with clear boundaries */
+
+        /* Output styling with clear boundaries (boundeyes are clear) */
         .output-content-wrapper {
             background: var(--background-primary) !important;
-            border: 2px solid var(--border-color) !important;
+            border: 2px solid var(--border-color) !important; /* Clear border */
             border-radius: 8px !important;
-            padding: 1rem !important;
-            min-height: 100px !important;
-            font-size: 0.95rem !important;
-            line-height: 1.6 !important;
+            padding: 1.5rem !important;
+            min-height: 150px !important; /* More space for output */
+            font-size: 1rem !important;
+            line-height: 1.7 !important;
             color: var(--text-primary) !important;
+            overflow-wrap: break-word; /* Ensure long words wrap */
+            word-break: break-word; /* Ensure long words wrap */
+            white-space: pre-wrap; /* Preserve formatting for newlines in markdown */
         }
         .response-header {
-            font-size: 1.2rem !important;
-            font-weight: 600 !important;
-            color: var(--text-primary) !important;
+            font-size: 1.3rem !important;
+            font-weight: 700 !important;
+            color: var(--primary-color) !important; /* Matches primary color */
             margin-bottom: 0.75rem !important;
             display: flex !important;
             align-items: center !important;
-            gap: 0.5rem !important;
+            gap: 0.6rem !important;
         }
         .response-icon {
-            font-size: 1.3rem !important;
+            font-size: 1.5rem !important;
             color: var(--primary-color) !important;
         }
         .divider {
             border: none !important;
-            border-top: 1px solid var(--border-color) !important;
-            margin: 0.75rem 0 !important;
+            border-top: 1px dashed var(--border-color) !important; /* Dashed divider for visual separation */
+            margin: 1rem 0 !important;
         }
+
         /* Error message styling */
         .error-message {
             background: var(--error-bg) !important;
             border: 2px solid var(--error-border) !important;
             color: var(--error-text) !important;
-            padding: 1rem !important;
+            padding: 1.25rem !important;
             border-radius: 8px !important;
             display: flex !important;
             align-items: flex-start !important;
-            gap: 0.75rem !important;
-            font-size: 0.9rem !important;
+            gap: 0.8rem !important;
+            font-size: 0.95rem !important;
+            font-weight: 500 !important;
+            line-height: 1.6 !important;
+        }
+        .error-message a {
+            color: var(--error-text) !important;
+            text-decoration: underline !important;
         }
         .error-icon {
-            font-size: 1.2rem !important;
+            font-size: 1.4rem !important;
             line-height: 1 !important;
             margin-top: 0.1rem !important;
         }
-        /* Placeholder styling */
+        .error-details {
+            font-size: 0.85rem !important;
+            color: var(--error-text) !important;
+            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;
             border-radius: 8px !important;
-            padding: 2rem 1rem !important;
+            padding: 2.5rem 1.5rem !important;
             text-align: center !important;
             color: var(--text-secondary) !important;
             font-style: italic !important;
+            font-size: 1.1rem !important;
         }
+
         /* Examples table styling */
         .examples-section .gr-samples-table {
             border: 2px solid var(--border-color) !important;
@@ -548,13 +603,14 @@ Answer:"""
         }
         .examples-section .gr-samples-table th,
         .examples-section .gr-samples-table td {
-            padding: 0.75rem !important;
+            padding: 0.9rem !important;
             border: none !important;
-            font-size: 0.9rem !important;
+            font-size: 0.95rem !important;
+            text-align: left !important; /* Ensure example text is left-aligned */
         }
         .examples-section .gr-samples-table th {
             background: var(--background-secondary) !important;
-            font-weight: 600 !important;
+            font-weight: 700 !important;
             color: var(--text-primary) !important;
         }
         .examples-section .gr-samples-table td {
@@ -566,66 +622,78 @@ Answer:"""
         .examples-section .gr-samples-table tr:hover td {
             background: var(--background-secondary) !important;
         }
-        /* Footer styling */
+        /* 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 */
         .app-footer-wrapper {
             background: var(--background-secondary) !important;
             border: 2px solid var(--border-color) !important;
             border-radius: 12px !important;
-            padding: 1.5rem !important;
+            padding: 1.75rem !important;
             margin-top: 1.5rem !important;
-            text-align: center !important;
+            text-align: center !important; /* Centered footer text */
         }
         .app-footer p {
-            margin: 0.5rem 0 !important;
-            font-size: 0.9rem !important;
+            margin: 0.6rem 0 !important;
+            font-size: 0.95rem !important;
             color: var(--text-secondary) !important;
-            line-height: 1.5 !important;
+            line-height: 1.6 !important;
         }
         .app-footer a {
             color: var(--primary-color) !important;
             text-decoration: none !important;
-            font-weight: 500 !important;
+            font-weight: 600 !important;
         }
         .app-footer a:hover {
             text-decoration: underline !important;
         }
-        /* Hide Gradio default elements */
-        .gr-examples .gr-label,
-        .gr-examples .label-wrap,
-        .gr-examples .gr-accordion-header {
-            display: none !important;
-        }
-        /* Responsive design */
+
+        /* Responsive design for smaller screens */
         @media (max-width: 768px) {
             .gradio-container {
-                padding: 0.5rem !important;
+                padding: 1rem !important;
             }
-            
             .app-header-title {
-                font-size: 2rem !important;
+                font-size: 2.2rem !important;
             }
-            
             .app-header-tagline {
                 font-size: 1rem !important;
             }
-            
+            .sub-section-title {
+                font-size: 1.4rem !important;
+            }
             .input-row {
-                flex-direction: column !important;
+                flex-direction: column !important; /* Stack inputs vertically */
             }
-            
             .button-row {
-                flex-direction: column !important;
+                flex-direction: column !important; /* Stack buttons vertically */
             }
-            
             .gradio-button {
-                width: 100% !important;
+                width: 100% !important; /* Full width buttons */
+            }
+            .dashboard-card-section {
+                padding: 1.25rem !important;
+            }
+            .output-content-wrapper {
+                min-height: 120px !important;
+            }
+            .placeholder {
+                padding: 1.5rem 1rem !important;
+                font-size: 1rem !important;
             }
         }
         """
 
-        with gr.Blocks(theme="soft", css=custom_css, title="Landlord-Tenant Rights Assistant") as demo:
-            # Header Section
+        # 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">
@@ -636,12 +704,12 @@ Answer:"""
                     """
                 )
 
-            # Main Dashboard Container
+            # 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>")
+                    gr.Markdown("<h3 class='sub-section-title'>Welcome & Disclaimer</h3>") # Centered by CSS
                     gr.Markdown(
                         """
                         Navigate landlord-tenant laws with ease. This assistant provides detailed, state-specific answers grounded in legal authority.
@@ -652,21 +720,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>")
+                    gr.Markdown("<h3 class='sub-section-title'>OpenAI API Key</h3>") # Centered by CSS
                     api_key_input = gr.Textbox(
                         label="API Key",
-                        type="password", 
-                        placeholder="Enter your API key (e.g., sk-...)",
-                        info="Required to process your query. Get one free from OpenAI.",
+                        type="password", # Hides the input for security
+                        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"]
+                        elem_classes=["input-field-group"] # Custom class for input styling
                     )
 
                 # 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>")
-                    with gr.Row(elem_classes="input-row"):
-                        with gr.Column(elem_classes="input-field", scale=3):
+                    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
                             query_input = gr.Textbox(
                                 label="Your Question",
                                 placeholder="E.g., What are the rules for security deposit returns in my state?",
@@ -674,7 +742,7 @@ Answer:"""
                                 max_lines=8,
                                 elem_classes=["input-field-group"]
                             )
-                        with gr.Column(elem_classes="input-field", scale=1):
+                        with gr.Column(elem_classes="input-field", scale=1): # State dropdown takes less space
                             state_input = gr.Dropdown(
                                 label="Select State",
                                 choices=dropdown_choices,
@@ -682,32 +750,32 @@ Answer:"""
                                 allow_custom_value=False,
                                 elem_classes=["input-field-group"]
                             )
-                    with gr.Row(elem_classes="button-row"):
+                    with gr.Row(elem_classes="button-row"): # Row for action buttons
                         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
                 with gr.Group(elem_classes="dashboard-card-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Legal Assistant's Response</h3>")
+                    gr.Markdown("<h3 class='sub-section-title'>Legal Assistant's Response</h3>") # Centered by CSS
                     output = gr.Markdown(
                         value="<div class='placeholder'>The answer will appear here after submitting your query.</div>",
-                        elem_classes="output-content-wrapper"
+                        elem_classes="output-content-wrapper" # Custom class for output styling
                     )
 
                 # Example Questions Section
                 with gr.Group(elem_classes="dashboard-card-section examples-section"):
-                    gr.Markdown("<h3 class='sub-section-title'>Example Questions</h3>")
+                    gr.Markdown("<h3 class='sub-section-title'>Example Questions</h3>") # Centered by CSS
                     if example_queries:
                         gr.Examples(
                             examples=example_queries,
                             inputs=[query_input, state_input],
                             examples_per_page=5,
-                            label=""
+                            label="" # Hide default Gradio label for examples to use our custom title
                         )
                     else:
-                        gr.Markdown("<div class='placeholder'>Sample questions could not be loaded.</div>")
+                        gr.Markdown("<div class='placeholder'>Sample questions could not be loaded. Please ensure the vector database is populated.</div>")
 
-            # Footer Section
+            # Footer Section - contains disclaimer and developer info
             with gr.Group(elem_classes="app-footer-wrapper"):
                 gr.Markdown(
                     """
@@ -716,20 +784,20 @@ Answer:"""
                     """
                 )
 
-            # Event Listeners
+            # 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"
+                api_name="submit_query" # Useful for debugging / external calls
             )
             
             clear_button.click(
                 fn=lambda: (
-                    "",
-                    "",
-                    initial_value,
-                    "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>"
+                    "", # 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
                 ),
                 inputs=[],
                 outputs=[api_key_input, query_input, state_input, output]
@@ -737,7 +805,7 @@ Answer:"""
 
             return demo
 
-# --- Main Execution Block (remains untouched from original logic, just fixed the exit) ---
+# --- Main Execution Block (UNCHANGED from original logic) ---
 if __name__ == "__main__":
     logging.info("Starting Landlord-Tenant Rights Bot application...")
     try:
@@ -748,28 +816,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) # This exit is correct: if file not found
+            exit(1) # Correctly exits if PDF is not found
 
         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) # This exit is correct: if file exists but is not readable
+            exit(1) # Correctly exits if PDF is unreadable
 
         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 if on Spaces/Cloud, else 7860
+        SERVER_PORT = int(os.getenv("PORT", 7860)) # Use PORT env var for Hugging Face Spaces
 
         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")