UI v18

app.py

@@ -5,7 +5,10 @@ from functools import lru_cache
 import re
 
 import gradio as gr
+import gradio.themes as themes # Import gradio.themes
+
 try:
+    # Assuming vector_db.py exists in the same directory or is installed
     from vector_db import VectorDatabase
 except ImportError:
     print("Error: Could not import VectorDatabase from vector_db.py.")
@@ -33,7 +36,7 @@ logging.basicConfig(
     format='%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)d] - %(message)s'
 )
 
-# --- RAGSystem Class ---
+# --- RAGSystem Class (Processing Logic - kept intact as requested) ---
 class RAGSystem:
     def __init__(self, vector_db: Optional[VectorDatabase] = None):
         logging.info("Initializing RAGSystem")
@@ -57,7 +60,7 @@ Statutes from context:
 Context information:
 --- START CONTEXT ---
 {context}
---- END CONTEXT ---
+--- END CONCONTEXT ---
 Answer:"""
         self.prompt_template = PromptTemplate(
             input_variables=["query", "context", "state", "statutes"],
@@ -72,7 +75,7 @@ Answer:"""
         for statute in statutes:
             statute = statute.strip()
             if '§' in statute and any(char.isdigit() for char in statute):
-                 if not re.match(r'^\([\w\.]+\)$', statute) and 'http' not in statute:
+                if not re.match(r'^\([\w\.]+\)$', statute) and 'http' not in statute:
                     if len(statute) > 5:
                         valid_statutes.append(statute)
 
@@ -169,7 +172,7 @@ Answer:"""
 
             if not answer_text:
                 logging.warning("LLM returned an empty answer.")
-                answer_text = "<div class='error-message'>The AI model returned an empty response. This might be due to the query, context limitations, or temporary issues. Please try rephrasing your question or try again later.</div>"
+                answer_text = "<div class='error-message'><span class='error-icon'>⚠️</span>The AI model returned an empty response. This might be due to the query, context limitations, or temporary issues. Please try rephrasing your question or try again later.</div>"
             else:
                 logging.info("LLM generated answer successfully.")
 
@@ -189,10 +192,10 @@ Answer:"""
                 error_message = "Error: The request was too long for the AI model. This can happen with very complex questions or extensive retrieved context."
                 details = "Try simplifying your question or asking about a more specific aspect."
             elif "timeout" in str(e).lower():
-                 error_message = "Error: The request to the AI model timed out. The service might be busy."
-                 details = "Please try again in a few moments."
+                error_message = "Error: The request to the AI model timed out. The service might be busy."
+                details = "Please try again in a few moments."
 
-            formatted_error = f"<div class='error-message'>{error_message}</div>"
+            formatted_error = f"<div class='error-message'><span class='error-icon'>❌</span>{error_message}</div>"
             if details:
                 formatted_error += f"<div class='error-details'>{details}</div>"
 
@@ -220,6 +223,7 @@ 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()
@@ -237,15 +241,12 @@ 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 ---
+    # --- GRADIO INTERFACE (Refactored to use earneleh/paris theme) ---
     def gradio_interface(self):
-        # Wrapper function for the Gradio interface logic
         def query_interface_wrapper(api_key: str, query: str, state: str) -> str:
-            logging.info(f"Gradio interface received query: '{query[:50]}...', state: '{state}'")
-
-            # Re-validate inputs robustly
+            # 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 here</a>.</div>"
+                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:
                 return "<div class='error-message'><span class='error-icon'>⚠️</span>Please select a valid state from the dropdown.</div>"
             if not query or not query.strip():
@@ -253,785 +254,540 @@ Answer:"""
 
             # 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>")
 
-            # Format the response for display
-            answer = result.get("answer", "<div class='error-message'><span class='error-icon'>⚠️</span>An unexpected error occurred, and no answer was generated. Please check the logs or try again.</div>")
-
-            # Add a header *only* if the answer is not an error message itself
-            if not "<div class='error-message'>" in answer:
-                 formatted_response = f"<div class='response-header'><span class='response-icon'>📜</span>Response for {state}</div><hr class='divider'>{answer}"
-            else:
-                 formatted_response = answer # Pass through error messages directly
-
-            # Log context length for debugging (optional)
-            context_used = result.get("context_used", "N/A")
-            if isinstance(context_used, str) and "N/A" not in context_used:
-                logging.debug(f"Context length used for query: {len(context_used)} characters.")
+            # Check if the answer already contains an error message (from deeper within process_query)
+            if "<div class='error-message'>" in answer:
+                return answer # Return the pre-formatted error message directly
             else:
-                logging.debug(f"No context was used or available for this query ({context_used}).")
+                # Format the successful response with the new UI structure
+                formatted_response = f"<div class='response-header'><span class='response-icon'>📜</span>Response for {state}</div><hr class='divider'>{answer}"
+                return formatted_response
 
-            return formatted_response
-
-        # --- Get Available States for Dropdown ---
         try:
             available_states_list = self.get_states()
-            if not available_states_list or "Error" in available_states_list[0]:
-                dropdown_choices = ["Error: Could not load states"]
-                initial_value = dropdown_choices[0]
-                logging.error("Could not load states for dropdown. UI will show error.")
-            else:
-                dropdown_choices = ["Select a state..."] + available_states_list
-                initial_value = dropdown_choices[0]
-        except Exception as e:
-            logging.error(f"Unexpected critical error getting states: {e}", exc_info=True)
+            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]
+        except Exception: # Catch-all for safety
             dropdown_choices = ["Error: Critical failure loading states"]
             initial_value = dropdown_choices[0]
 
-        # --- Prepare Example Queries ---
+        # 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"],
             ["My landlord hasn't made necessary repairs. What can I do?", "Texas"],
-            ["What are the limits on rent increases in my state?", "Florida"],
-            ["Is my lease automatically renewed if I don't move out?", "Illinois"],
-            ["What happens if I break my lease early?", "Washington"]
+            ["How much notice must a landlord give to raise rent?", "Florida"],
+            ["What is an implied warranty of habitability?", "Illinois"]
         ]
         example_queries = []
-        if available_states_list and "Error" not in available_states_list[0]:
+        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]
-        if not example_queries:
-             fallback_state = available_states_list[0] if available_states_list and "Error" not in available_states_list[0] else "California"
-             example_queries.append(["What basic rights do tenants have?", fallback_state])
+            # Add a generic example if no specific state examples match or if list is empty
+            if not example_queries:
+                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
+            example_queries.append(["What basic rights do tenants have?", "California"])
 
-        # --- Refined Custom CSS ---
+        # Improved Custom CSS for better UI design and HuggingFace compatibility
         custom_css = """
-        @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;500;600;700&display=swap');
-
-        /* --- CSS Variables for Theme Consistency --- */
+        /* Import legible 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 {
-            --primary-color: #2563EB; /* Tailwind Blue 600 */
-            --primary-hover: #1D4ED8; /* Tailwind Blue 700 */
-            --secondary-color: #4B5563; /* Gray 600 */
-            --secondary-hover: #374151; /* Gray 700 */
-            --text-primary: #111827; /* Gray 900 */
-            --text-secondary: #6B7280; /* Gray 500 */
-            --background: #F3F4F6; /* Gray 100 */
-            --card-background: #FFFFFF; /* White */
-            --border-color: #D1D5DB; /* Gray 300 */
-            --shadow: 0 2px 4px rgba(0, 0, 0, 0.04), 0 4px 8px rgba(0, 0, 0, 0.06); /* Softer, layered shadow */
-            --error-bg: #FEF2F2; /* Red 50 */
-            --error-border: #FECACA; /* Red 300 */
-            --error-accent: #EF4444; /* Red 500 */
-            --error-text: #B91C1C; /* Red 700 */
-            --success-bg: #F0FDF4; /* Green 50 */
-            --success-border: #A7F3D0; /* Green 300 */
-            --success-text: #15803D; /* Green 700 */
-            --divider: #E5E7EB; /* Gray 200 */
-            --focus-ring: rgba(37, 99, 235, 0.3); /* Based on new primary */
-        }
-
-        /* Dark Mode Variables */
+            --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;
+        }
+        
+        /* Dark mode variables */
         @media (prefers-color-scheme: dark) {
             :root {
-                --primary-color: #3B82F6; /* Tailwind Blue 500 (lighter for dark mode primary) */
-                --primary-hover: #60A5FA; /* Tailwind Blue 400 */
-                --text-primary: #F3F4F6; /* Gray 100 */
-                --text-secondary: #9CA3AF; /* Gray 400 */
-                --background: #111827; /* Gray 900 (very dark) */
-                --card-background: #1F2937; /* Gray 800 (main content card bg) */
-                --border-color: #4B5563; /* Gray 600 */
-                --shadow: 0 2px 4px rgba(0, 0, 0, 0.1), 0 4px 8px rgba(0, 0, 0, 0.2);
-                --error-bg: #450A0A; /* Darker Red */
-                --error-border: #7F1D1D; /* Darker Red */
-                --error-accent: #F87171; /* Lighter Red for accent */
-                --error-text: #FECACA; /* Lighter Red for text */
-                --success-bg: #064E3B; /* Darker Green */
-                --success-border: #15803D; /* Darker Green */
-                --success-text: #A7F3D0; /* Lighter Green */
-                --divider: #374151; /* Gray 700 */
-                --focus-ring: rgba(59, 130, 246, 0.4);
+                --background-primary: #0f172a;
+                --background-secondary: #1e293b;
+                --text-primary: #f1f5f9;
+                --text-secondary: #94a3b8;
+                --border-color: #334155;
+                --error-bg: #1e1b1b;
+                --error-border: #451a1a;
+                --error-text: #f87171;
             }
         }
-
-        /* --- Base & Body --- */
-        body, .gradio-container {
-            font-family: 'Poppins', -apple-system, BlinkMacSystemFont, sans-serif !important;
-            background: var(--background) !important;
-            color: var(--text-primary) !important;
-            margin: 0;
-            padding: 0;
-            min-height: 100vh;
-            font-size: 15px; /* Slightly smaller base font */
-            line-height: 1.6; /* Increased line height for readability */
-            -webkit-font-smoothing: antialiased;
-            -moz-osx-font-smoothing: grayscale;
-        }
-        * {
-            box-sizing: border-box;
-        }
-
-        /* --- Main Content Container --- */
-        .gradio-container > .flex.flex-col {
-            max-width: 960px; /* Slightly narrower for focus */
+        /* Base container improvements */
+        .gradio-container {
+            max-width: 1000px !important;
             margin: 0 auto !important;
-            padding: 2.5rem 1.5rem !important; /* Adjusted padding */
-            gap: 2rem !important; /* Adjusted gap */
-            background: transparent !important;
-        }
-
-        /* --- Card Styling --- */
-        .card-style {
-            background: var(--card-background) !important;
-            border: 1px solid var(--border-color) !important;
-            border-radius: 12px !important; /* More modern radius */
-            padding: 1.75rem !important; /* Adjusted padding */
-            box-shadow: var(--shadow) !important;
-            transition: transform 0.2s ease, background 0.2s ease, border 0.2s ease;
-        }
-        .card-style:hover {
-            transform: translateY(-2px);
-        }
-
-        /* --- Header Section --- */
-        .header-section {
-            background: var(--primary-color) !important; /* Solid primary color */
-            border-radius: 12px !important;
-            padding: 2.5rem 2rem !important;
+            padding: 1rem !important;
+            font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
+        }
+        /* Header styling */
+        .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;
+            margin-bottom: 1.5rem !important;
             text-align: center !important;
-            color: #FFFFFF !important;
-            box-shadow: var(--shadow) !important;
-            position: relative;
-            overflow: hidden;
-        }
-        .header-section::before { /* Subtle background pattern */
-            content: '';
-            position: absolute;
-            top: 0; left: 0; width: 100%; height: 100%;
-            background-image: radial-gradient(rgba(255, 255, 255, 0.07) 1px, transparent 1.2px);
-            background-size: 8px 8px;
-            opacity: 0.5;
-            pointer-events: none;
-        }
-        .header-logo {
-            font-size: 2.5rem; /* Adjusted size */
-            margin-bottom: 0.75rem;
-        }
-        .header-title {
-            font-size: 2rem; /* Adjusted size */
-            font-weight: 600; /* Adjusted weight */
-            margin: 0 0 0.5rem 0;
+            box-shadow: var(--shadow-md) !important;
         }
-        .header-tagline {
-            font-size: 1.1rem; /* Adjusted size */
-            font-weight: 400; /* Adjusted weight */
-            opacity: 0.85;
-        }
-
-        /* --- Introduction Section --- */
-        .intro-card h3 { /* Title like "Know Your Rights" */
-            font-size: 1.5rem; /* Adjusted relative to new base */
-            font-weight: 600;
-            color: var(--primary-color);
-            margin: 0 0 1rem 0;
-            padding-bottom: 0.5rem;
-            border-bottom: 2px solid var(--primary-color);
-            display: inline-block;
-        }
-        .intro-card p {
-            font-size: 0.95rem; /* Adjusted relative to new base */
-            line-height: 1.7;
-            color: var(--text-secondary);
-            margin: 0 0 0.75rem 0;
-        }
-        .intro-card a {
-            color: var(--primary-color);
-            text-decoration: none;
-            font-weight: 500;
-            transition: color 0.2s ease;
-        }
-        .intro-card a:hover {
-            color: var(--primary-hover);
-            text-decoration: underline;
+        .app-header-logo {
+            font-size: 3rem !important;
+            margin-bottom: 0.5rem !important;
+            display: block !important;
         }
-        .intro-card strong {
-            font-weight: 600;
-            color: var(--text-primary);
+        .app-header-title {
+            font-family: 'Poppins', sans-serif !important;
+            font-size: 2.5rem !important;
+            font-weight: 700 !important;
+            color: var(--text-primary) !important;
+            margin: 0 0 0.5rem 0 !important;
+            line-height: 1.2 !important;
         }
-
-        /* --- Input Form Section --- */
-        .input-form-card h3 { /* Title like "Ask Your Question" */
-            font-size: 1.4rem; /* Adjusted */
-            font-weight: 600;
-            color: var(--text-primary);
-            margin: 0 0 1.25rem 0;
-            padding-bottom: 0.5rem;
-            border-bottom: 1px solid var(--divider); /* Thinner divider */
+        .app-header-tagline {
+            font-size: 1.1rem !important;
+            color: var(--text-secondary) !important;
+            font-weight: 400 !important;
+            margin: 0 !important;
+        }
+        /* Main container with compact spacing */
+        .main-dashboard-container {
+            display: flex !important;
+            flex-direction: column !important;
+            gap: 1rem !important; /* Reduced gap for more compact look */
+        }
+        /* Card sections with better boundaries */
+        .dashboard-card-section {
+            background: var(--background-primary) !important;
+            border: 2px solid var(--border-color) !important;
+            border-radius: 12px !important;
+            padding: 1.5rem !important; /* Reduced padding for compactness */
+            box-shadow: var(--shadow-sm) !important;
+            transition: all 0.2s ease !important;
         }
-        .input-field-group {
-            margin-bottom: 1.25rem;
+        .dashboard-card-section:hover {
+            box-shadow: var(--shadow-md) !important;
         }
-        .input-row {
-            display: flex;
-            gap: 1.25rem;
-            flex-wrap: wrap;
-            margin-bottom: 1.25rem;
+        /* Centered section titles with better typography */
+        .sub-section-title {
+            font-family: 'Poppins', sans-serif !important;
+            font-size: 1.5rem !important;
+            font-weight: 600 !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;
+            display: block !important;
         }
-        .input-field {
-            flex: 1;
-            min-width: 200px; /* Adjusted min-width */
+        /* Improved input styling with clear boundaries */
+        .gradio-textbox, .gradio-dropdown {
+            margin-bottom: 0.75rem !important; /* Reduced margin for compactness */
         }
         .gradio-textbox textarea,
-        .gradio-dropdown select,
-        .gradio-textbox input[type=password] {
-            border: 1px solid var(--border-color) !important;
-            border-radius: 8px !important; /* Sharper radius */
-            padding: 0.8rem 1rem !important; /* Adjusted padding */
-            font-size: 0.95rem !important; /* Adjusted font size */
-            background: var(--card-background) !important; /* Use card for consistency, can be var(--background) for contrast */
+        .gradio-textbox input,
+        .gradio-dropdown select {
+            background: var(--background-primary) !important;
+            border: 2px solid var(--border-color) !important;
+            border-radius: 8px !important;
+            padding: 0.75rem !important;
+            font-size: 0.95rem !important;
+            font-family: 'Inter', sans-serif !important;
             color: var(--text-primary) !important;
-            transition: border-color 0.2s ease, box-shadow 0.2s ease, background 0.2s ease;
-            width: 100% !important;
-            box-shadow: inset 0 1px 2px rgba(0, 0, 0, 0.03);
-        }
-        .gradio-textbox textarea {
-            min-height: 110px; /* Adjusted height */
-            resize: vertical;
+            transition: all 0.2s ease !important;
+            box-shadow: var(--shadow-sm) !important;
         }
         .gradio-textbox textarea:focus,
-        .gradio-dropdown select:focus,
-        .gradio-textbox input[type=password]:focus {
-            border-color: var(--primary-color) !important;
-            box-shadow: 0 0 0 3px var(--focus-ring) !important; /* Slightly smaller focus ring */
+        .gradio-textbox input:focus,
+        .gradio-dropdown select:focus {
             outline: none !important;
-            /* background: var(--background) !important; */ /* Keep card background on focus or change to main for contrast */
+            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;
         }
-        .gradio-input-label,
-        .gradio-output-label {
-            font-size: 0.9rem !important; /* Adjusted */
+        /* Label styling for better readability */
+        .gradio-textbox label,
+        .gradio-dropdown label {
             font-weight: 500 !important;
             color: var(--text-primary) !important;
-            margin-bottom: 0.4rem !important;
+            font-size: 0.9rem !important;
+            margin-bottom: 0.5rem !important;
             display: block !important;
         }
-        .gradio-input-info {
-            font-size: 0.8rem !important; /* Adjusted */
+        /* Info text styling */
+        .gradio-textbox .gr-form,
+        .gradio-dropdown .gr-form {
+            font-size: 0.85rem !important;
             color: var(--text-secondary) !important;
-            margin-top: 0.3rem;
-            font-style: italic;
+            margin-top: 0.25rem !important;
+        }
+        /* Input row layout improvements */
+        .input-row {
+            display: flex !important;
+            gap: 1rem !important;
+            margin-bottom: 0.5rem !important; /* Reduced margin */
         }
-        /* Buttons */
+        .input-field {
+            flex: 1 !important;
+        }
+        /* Button styling improvements */
         .button-row {
-            display: flex;
-            gap: 0.75rem; /* Tighter gap */
-            margin-top: 1.25rem;
-            flex-wrap: wrap;
-            justify-content: flex-end;
+            display: flex !important;
+            gap: 1rem !important;
+            justify-content: flex-end !important;
+            margin-top: 1rem !important;
         }
         .gradio-button {
-            border-radius: 8px !important; /* Sharper radius */
-            padding: 0.75rem 1.5rem !important; /* Adjusted padding */
-            font-size: 0.95rem !important; /* Adjusted font size */
+            padding: 0.75rem 1.5rem !important;
+            border-radius: 8px !important;
             font-weight: 500 !important;
-            border: none !important;
-            cursor: pointer;
-            transition: background-color 0.2s ease, transform 0.1s ease, box-shadow 0.2s ease;
-            box-shadow: 0 1px 2px rgba(0,0,0,0.05) !important; /* More subtle shadow */
-        }
-        .gradio-button:hover:not(:disabled) {
-            transform: translateY(-1px);
-            box-shadow: 0 2px 4px rgba(0,0,0,0.07) !important;
-        }
-        .gradio-button:active:not(:disabled) {
-            transform: scale(0.98) translateY(0);
-        }
-        .gradio-button:disabled {
-            background: var(--border-color) !important;
-            color: var(--text-secondary) !important;
-            cursor: not-allowed;
-            box-shadow: none !important;
+            font-size: 0.9rem !important;
+            transition: all 0.2s ease !important;
+            cursor: pointer !important;
+            border: 2px solid transparent !important;
         }
         .gr-button-primary {
             background: var(--primary-color) !important;
-            color: #FFFFFF !important;
+            color: white !important;
+            box-shadow: var(--shadow-sm) !important;
         }
-        .gr-button-primary:hover:not(:disabled) {
+        .gr-button-primary:hover {
             background: var(--primary-hover) !important;
+            box-shadow: var(--shadow-md) !important;
+            transform: translateY(-1px) !important;
         }
         .gr-button-secondary {
-            background: var(--card-background) !important; /* Changed from transparent */
+            background: transparent !important;
             color: var(--text-primary) !important;
-            border: 1px solid var(--border-color) !important;
-            box-shadow: none !important;
-        }
-        .gr-button-secondary:hover:not(:disabled) {
-            background: var(--background) !important; /* Use main bg for hover */
-            border-color: var(--secondary-hover) !important;
-        }
-
-        /* --- Output Section --- */
-        .output-card .response-header {
-            font-size: 1.3rem; /* Adjusted */
-            font-weight: 600;
-            color: var(--text-primary);
-            margin: 0 0 0.75rem 0;
-            display: flex;
-            align-items: center;
-            gap: 0.5rem;
-        }
-        .output-card .response-icon {
-            font-size: 1.5rem; /* Adjusted */
-        }
-        .output-card .divider {
-            border: none;
-            border-top: 1px solid var(--divider);
-            margin: 0.75rem 0 1.25rem 0; /* Adjusted margins */
+            border-color: var(--border-color) !important;
         }
-        .output-card .output-content-wrapper {
-            font-size: 0.95rem; /* Adjusted */
-            line-height: 1.7; /* Adjusted */
-            color: var(--text-primary);
-        }
-        .output-card .output-content-wrapper p {
-            margin-bottom: 0.85rem;
-        }
-        .output-card .output-content-wrapper ul,
-        .output-card .output-content-wrapper ol {
-            margin-left: 1.25rem;
-            margin-bottom: 0.85rem;
-            padding-left: 0.85rem;
-        }
-        .output-card .output-content-wrapper li {
-            margin-bottom: 0.4rem;
-        }
-        .output-card .output-content-wrapper strong,
-        .output-card .output-content-wrapper b {
-            font-weight: 600;
-            color: var(--text-primary);
-        }
-        .output-card .output-content-wrapper a {
-            color: var(--primary-color);
-            text-decoration: none;
-            font-weight: 500;
-        }
-        .output-card .output-content-wrapper a:hover {
-            color: var(--primary-hover);
-            text-decoration: underline;
-        }
-        /* Error and Success Messages */
-        .output-card .error-message,
-        .output-card .success-message {
-            display: flex;
-            align-items: flex-start;
-            gap: 0.6rem;
-            border-radius: 8px; /* Match other radii */
-            padding: 0.85rem 1.25rem; /* Adjusted padding */
-            margin-top: 0.5rem;
-            font-weight: 500;
-            line-height: 1.5;
-        }
-        .output-card .error-message {
-            background: var(--error-bg);
-            border: 1px solid var(--error-border);
-            border-left: 3px solid var(--error-accent); /* Thinner accent line */
-            color: var(--error-text);
+        .gr-button-secondary:hover {
+            background: var(--background-secondary) !important;
+            border-color: var(--primary-color) !important;
         }
-        .output-card .success-message {
-            background: var(--success-bg);
-            border: 1px solid var(--success-border);
-            color: var(--success-text);
-            border-left: 3px solid var(--success-text);
+        /* Output styling with clear boundaries */
+        .output-content-wrapper {
+            background: var(--background-primary) !important;
+            border: 2px solid var(--border-color) !important;
+            border-radius: 8px !important;
+            padding: 1rem !important;
+            min-height: 100px !important;
+            font-size: 0.95rem !important;
+            line-height: 1.6 !important;
+            color: var(--text-primary) !important;
         }
-        .output-card .error-icon,
-        .output-card .success-icon {
-            font-size: 1.1rem; /* Adjusted */
-            line-height: 1.5;
-            margin-top: 2px; /* Align icon better with text */
+        .response-header {
+            font-size: 1.2rem !important;
+            font-weight: 600 !important;
+            color: var(--text-primary) !important;
+            margin-bottom: 0.75rem !important;
+            display: flex !important;
+            align-items: center !important;
+            gap: 0.5rem !important;
         }
-        .output-card .error-details {
-            font-size: 0.85rem; /* Adjusted */
-            color: var(--error-text); /* Ensure correct text color for dark mode if needed */
-            margin-top: 0.4rem;
-            font-style: italic;
+        .response-icon {
+            font-size: 1.3rem !important;
+            color: var(--primary-color) !important;
         }
-        /* Placeholder text */
-        .output-card .placeholder {
-            color: var(--text-secondary);
-            font-style: italic;
-            text-align: center;
-            padding: 1.5rem 1rem; /* Adjusted padding */
-            display: block;
-            font-size: 1rem; /* Adjusted */
+        .divider {
+            border: none !important;
+            border-top: 1px solid var(--border-color) !important;
+            margin: 0.75rem 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;
+            border-radius: 8px !important;
+            display: flex !important;
+            align-items: flex-start !important;
+            gap: 0.75rem !important;
+            font-size: 0.9rem !important;
+        }
+        .error-icon {
+            font-size: 1.2rem !important;
+            line-height: 1 !important;
+            margin-top: 0.1rem !important;
+        }
+        /* Placeholder styling */
+        .placeholder {
+            background: var(--background-secondary) !important;
+            border: 2px dashed var(--border-color) !important;
+            border-radius: 8px !important;
+            padding: 2rem 1rem !important;
+            text-align: center !important;
+            color: var(--text-secondary) !important;
+            font-style: italic !important;
+        }
+        /* Examples table styling */
+        .examples-section .gr-samples-table {
+            border: 2px solid var(--border-color) !important;
+            border-radius: 8px !important;
+            overflow: hidden !important;
+            margin-top: 1rem !important;
+        }
+        .examples-section .gr-samples-table th,
+        .examples-section .gr-samples-table td {
+            padding: 0.75rem !important;
+            border: none !important;
+            font-size: 0.9rem !important;
         }
-
-        /* --- Examples Section --- */
-        .examples-card .gr-examples-header {
-            font-size: 1.3rem !important; /* Adjusted */
+        .examples-section .gr-samples-table th {
+            background: var(--background-secondary) !important;
             font-weight: 600 !important;
             color: var(--text-primary) !important;
-            margin: 0 0 1.25rem 0 !important;
-            padding-bottom: 0.5rem !important;
-            border-bottom: 1px solid var(--divider) !important; /* Thinner divider */
         }
-        .examples-card .gr-examples-table {
-            border-collapse: collapse !important;
-            width: 100% !important;
-            background: var(--card-background) !important;
-            border-radius: 8px !important; /* Match other radii */
-            overflow: hidden;
-            border: 1px solid var(--border-color) !important; /* Add outer border to table */
-        }
-        .examples-card .gr-examples-table th,
-        .examples-card .gr-examples-table td {
-            text-align: left !important;
-            padding: 0.75rem 1rem !important; /* Adjusted padding */
-            border: 1px solid var(--border-color) !important;
-            font-size: 0.9rem !important; /* Adjusted font size */
+        .examples-section .gr-samples-table td {
+            background: var(--background-primary) !important;
             color: var(--text-primary) !important;
-            background: transparent !important;
-        }
-        .examples-card .gr-examples-table th {
-            font-weight: 500 !important;
-            background: var(--background) !important; /* Use main bg for header */
-        }
-        .examples-card .gr-examples-table tr {
-            cursor: pointer;
-            transition: background 0.2s ease;
+            border-top: 1px solid var(--border-color) !important;
+            cursor: pointer !important;
         }
-        .examples-card .gr-examples-table tr:hover td {
-            background: var(--background) !important; /* Use main bg for hover */
+        .examples-section .gr-samples-table tr:hover td {
+            background: var(--background-secondary) !important;
         }
-
-        /* --- Footer Section --- */
-        .footer-section {
-            background: transparent !important;
-            border-top: 1px solid var(--divider) !important;
-            padding: 1.5rem 1rem !important; /* Adjusted padding */
-            margin-top: 1.5rem !important; /* Adjusted margin */
+        /* Footer styling */
+        .app-footer-wrapper {
+            background: var(--background-secondary) !important;
+            border: 2px solid var(--border-color) !important;
+            border-radius: 12px !important;
+            padding: 1.5rem !important;
+            margin-top: 1.5rem !important;
             text-align: center !important;
+        }
+        .app-footer p {
+            margin: 0.5rem 0 !important;
+            font-size: 0.9rem !important;
             color: var(--text-secondary) !important;
-            font-size: 0.85rem !important; /* Adjusted font size */
             line-height: 1.5 !important;
         }
-        .footer-section strong {
-            color: var(--text-primary);
-            font-weight: 500;
-        }
-        .footer-section a {
-            color: var(--primary-color);
-            text-decoration: none;
-            font-weight: 500;
+        .app-footer a {
+            color: var(--primary-color) !important;
+            text-decoration: none !important;
+            font-weight: 500 !important;
         }
-        .footer-section a:hover {
-            color: var(--primary-hover);
-            text-decoration: underline;
+        .app-footer a:hover {
+            text-decoration: underline !important;
         }
-
-        /* --- Accessibility & Focus --- */
-        :focus-visible { /* Standard focus visibility */
-            outline: 2px solid var(--primary-color) !important;
-            outline-offset: 2px;
-        }
-        /* Remove custom box-shadow focus for inputs/selects if :focus-visible is preferred */
-        .gradio-textbox textarea:focus,
-        .gradio-dropdown select:focus,
-        .gradio-textbox input[type=password]:focus {
-            border-color: var(--primary-color) !important;
-            box-shadow: 0 0 0 3px var(--focus-ring) !important; /* Keep this for consistent focus */
-            outline: none !important;
-        }
-        .gradio-button span:focus { /* Remove Gradio's default focus on button text */
-            outline: none !important;
+        /* Hide Gradio default elements */
+        .gr-examples .gr-label,
+        .gr-examples .label-wrap,
+        .gr-examples .gr-accordion-header {
+            display: none !important;
         }
-
-
-        /* --- Responsive Adjustments --- */
+        /* Responsive design */
         @media (max-width: 768px) {
-            body { font-size: 14px; }
-            .gradio-container > .flex.flex-col {
-                padding: 2rem 1rem !important;
-                gap: 1.5rem !important;
+            .gradio-container {
+                padding: 0.5rem !important;
             }
-            .card-style {
-                padding: 1.5rem !important;
-                border-radius: 10px !important;
+            
+            .app-header-title {
+                font-size: 2rem !important;
             }
-            .header-section {
-                padding: 2rem 1.5rem !important;
-                border-radius: 10px !important;
+            
+            .app-header-tagline {
+                font-size: 1rem !important;
             }
-            .header-title { font-size: 1.8rem; }
-            .header-tagline { font-size: 1rem; }
+            
             .input-row {
-                flex-direction: column;
-                gap: 1rem;
-            }
-            .button-row { justify-content: center; }
-            .intro-card h3, .input-form-card h3, .output-card .response-header, .examples-card .gr-examples-header {
-                font-size: 1.2rem !important;
-            }
-        }
-        @media (max-width: 480px) {
-            body { font-size: 14px; } /* Keep 14px or adjust if too small */
-            .gradio-container > .flex.flex-col {
-                padding: 1.25rem 0.75rem !important;
-                gap: 1.25rem !important;
-            }
-            .card-style {
-                padding: 1rem !important;
-                border-radius: 8px !important;
-            }
-            .header-section {
-                padding: 1.5rem 1rem !important;
-                border-radius: 8px !important;
-            }
-            .header-logo { font-size: 2rem; }
-            .header-title { font-size: 1.5rem; }
-            .header-tagline { font-size: 0.9rem; }
-
-            .intro-card h3, .input-form-card h3, .output-card .response-header, .examples-card .gr-examples-header {
-                font-size: 1.1rem !important;
-            }
-            .gradio-textbox textarea,
-            .gradio-dropdown select,
-            .gradio-textbox input[type=password] {
-                font-size: 0.9rem !important;
-                padding: 0.7rem 0.9rem !important;
-            }
-            .gradio-button {
-                width: 100%;
-                padding: 0.7rem 1.25rem !important;
-                font-size: 0.9rem !important;
+                flex-direction: column !important;
             }
+            
             .button-row {
-                flex-direction: column;
-                gap: 0.5rem;
+                flex-direction: column !important;
             }
-            .examples-card .gr-examples-table th,
-            .examples-card .gr-examples-table td {
-                padding: 0.5rem 0.7rem !important;
-                font-size: 0.85rem !important;
+            
+            .gradio-button {
+                width: 100% !important;
             }
         }
-
-        /* --- Gradio Overrides --- */
-        .gradio-container > .flex {
-            gap: 2rem !important; /* Match main gap */
-        }
-        .gradio-markdown > *:first-child { margin-top: 0; }
-        .gradio-markdown > *:last-child { margin-bottom: 0; }
-        .gradio-dropdown,
-        .gradio-textbox { /* Remove Gradio default borders/padding around components */
-            border: none !important;
-            padding: 0 !important;
-            background: transparent !important;
-        }
         """
 
-        # --- Gradio Blocks Layout ---
-        with gr.Blocks(theme=None, css=custom_css, title="Landlord-Tenant Rights Assistant") as demo:
+        with gr.Blocks(theme="soft", css=custom_css, title="Landlord-Tenant Rights Assistant") as demo:
             # Header Section
-            with gr.Group(elem_classes="header-section"):
+            with gr.Group(elem_classes="app-header-wrapper"):
                 gr.Markdown(
                     """
-                    <span class="header-logo">⚖️</span>
-                    <h1 class="header-title">Landlord-Tenant Rights Assistant</h1>
-                    <p class="header-tagline">Empowering You with State-Specific Legal Insights</p>
-                    """, elem_id="app-title"
-                )
-
-            # Introduction Section
-            with gr.Group(elem_classes="card-style intro-card"):
-                gr.Markdown(
+                    <div class="app-header">
+                        <span class="app-header-logo">⚖️</span>
+                        <h1 class="app-header-title">Landlord-Tenant Rights Assistant</h1>
+                        <p class="app-header-tagline">Empowering You with State-Specific Legal Insights</p>
+                    </div>
                     """
-                    <h3>Know Your Rights</h3>
-                    <p>Navigate landlord-tenant laws with ease. Enter your <strong>OpenAI API key</strong>, select your state, and ask your question to get detailed, state-specific answers.</p>
-                    <p>Don't have an API key? <a href='https://platform.openai.com/api-keys' target='_blank'>Get one free from OpenAI</a>.</p>
-                    <p><strong>Disclaimer:</strong> This tool provides information only, not legal advice. For legal guidance, consult a licensed attorney.</p>
-                    """, elem_id="app-description"
                 )
 
-            # Input Form Section
-            with gr.Group(elem_classes="card-style input-form-card"):
-                gr.Markdown("<h3>Ask Your Question</h3>", elem_id="form-heading")
+            # Main Dashboard Container
+            with gr.Column(elem_classes="main-dashboard-container"):
 
-                with gr.Column(elem_classes="input-field-group"):
-                    api_key_input = gr.Textbox(
-                        label="OpenAI API Key",
-                        type="password",
-                        placeholder="Enter your API key (e.g., sk-...)",
-                        info="Required to process your query. Securely used per request, not stored.",
-                        elem_id="api-key-input",
-                        lines=1
+                # Introduction and Disclaimer Card
+                with gr.Group(elem_classes="dashboard-card-section"):
+                    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.
+                        
+                        **Disclaimer:** This tool is for informational purposes only and does not constitute legal advice. For specific legal guidance, always consult a licensed attorney in your jurisdiction.
+                        """
                     )
 
-                with gr.Row(elem_classes="input-row"):
-                    with gr.Column(elem_classes="input-field"):
-                        query_input = gr.Textbox(
-                            label="Your Question",
-                            placeholder="E.g., What are the rules for security deposit returns in my state?",
-                            lines=4,
-                            max_lines=8,
-                            elem_id="query-input"
-                        )
-                    with gr.Column(elem_classes="input-field"):
-                        state_input = gr.Dropdown(
-                            label="Select State",
-                            choices=dropdown_choices,
-                            value=initial_value,
-                            allow_custom_value=False,
-                            elem_id="state-dropdown"
-                        )
-
-                with gr.Row(elem_classes="button-row"):
-                    clear_button = gr.Button(
-                        "Clear",
-                        variant="secondary",
-                        elem_id="clear-button",
-                        elem_classes=["gr-button-secondary"]
-                    )
-                    submit_button = gr.Button(
-                        "Submit Query",
-                        variant="primary",
-                        elem_id="submit-button",
-                        elem_classes=["gr-button-primary"]
+                # OpenAI API Key Input Card
+                with gr.Group(elem_classes="dashboard-card-section"):
+                    gr.Markdown("<h3 class='sub-section-title'>OpenAI API Key</h3>")
+                    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.",
+                        lines=1,
+                        elem_classes=["input-field-group"]
                     )
 
-            # Output Section
-            with gr.Group(elem_classes="card-style output-card"):
-                with gr.Column():
+                # 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):
+                            query_input = gr.Textbox(
+                                label="Your Question",
+                                placeholder="E.g., What are the rules for security deposit returns in my state?",
+                                lines=4,
+                                max_lines=8,
+                                elem_classes=["input-field-group"]
+                            )
+                        with gr.Column(elem_classes="input-field", scale=1):
+                            state_input = gr.Dropdown(
+                                label="Select State",
+                                choices=dropdown_choices,
+                                value=initial_value,
+                                allow_custom_value=False,
+                                elem_classes=["input-field-group"]
+                            )
+                    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
+                with gr.Group(elem_classes="dashboard-card-section"):
+                    gr.Markdown("<h3 class='sub-section-title'>Legal Assistant's Response</h3>")
                     output = gr.Markdown(
-                        value="<div class='placeholder'>Your answer will appear here after submitting your query.</div>",
-                        elem_id="output-content",
+                        value="<div class='placeholder'>The answer will appear here after submitting your query.</div>",
                         elem_classes="output-content-wrapper"
                     )
 
-            # Example Questions Section
-            if example_queries:
-                with gr.Group(elem_classes="card-style examples-card"):
-                    gr.Examples(
-                        examples=example_queries,
-                        inputs=[query_input, state_input],
-                        label="Explore Sample Questions",
-                        examples_per_page=6
-                    )
-            else:
-                with gr.Group(elem_classes="card-style examples-card"):
-                    gr.Markdown(
-                        "<div class='placeholder'>Sample questions could not be loaded. Please ensure states are available.</div>"
-                    )
+                # Example Questions Section
+                with gr.Group(elem_classes="dashboard-card-section examples-section"):
+                    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=""
+                        )
+                    else:
+                        gr.Markdown("<div class='placeholder'>Sample questions could not be loaded.</div>")
 
             # Footer Section
-            with gr.Group(elem_classes="footer-section"):
+            with gr.Group(elem_classes="app-footer-wrapper"):
                 gr.Markdown(
                     """
-                    **Disclaimer**: This tool is for informational purposes only and does not constitute legal advice.
-                    <br><br>
-                    Developed by **Nischal Subedi**. Connect on <a href="https://www.linkedin.com/in/nischal1/" target="_blank">LinkedIn</a> or explore insights at <a href="https://datascientistinsights.substack.com/" target="_blank">Substack</a>.
-                    """, elem_id="app-footer"
+                    This tool is for informational purposes only and does not constitute legal advice. For legal guidance, always consult with a licensed attorney in your jurisdiction.
+                    Developed by **Nischal Subedi**. Connect on [LinkedIn](https://www.linkedin.com/in/nischal1/) or explore insights at [Substack](https://datascientistinsights.substack.com/).
+                    """
                 )
 
-            # --- Event Listeners ---
+            # Event Listeners
             submit_button.click(
                 fn=query_interface_wrapper,
                 inputs=[api_key_input, query_input, state_input],
                 outputs=output,
                 api_name="submit_query"
             )
-
+            
             clear_button.click(
                 fn=lambda: (
-                    "", "", initial_value,
+                    "",
+                    "",
+                    initial_value,
                     "<div class='placeholder'>Inputs cleared. Ready for your next question.</div>"
                 ),
                 inputs=[],
                 outputs=[api_key_input, query_input, state_input, output]
             )
 
-        logging.info("Refined Gradio interface created successfully.")
-        return demo
+            return demo
 
-# --- Main Execution Block ---
+# --- Main Execution Block (remains untouched from original logic, just fixed the exit) ---
 if __name__ == "__main__":
     logging.info("Starting Landlord-Tenant Rights Bot application...")
     try:
         SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
-        DEFAULT_PDF_PATH = os.path.join(SCRIPT_DIR, "data/tenant-landlord.pdf")
-        DEFAULT_DB_PATH = os.path.join(SCRIPT_DIR, "data/chroma_db")
+        DEFAULT_PDF_PATH = os.path.join(SCRIPT_DIR, "tenant-landlord.pdf")
+        DEFAULT_DB_PATH = os.path.join(SCRIPT_DIR, "chroma_db")
 
         PDF_PATH = os.getenv("PDF_PATH", DEFAULT_PDF_PATH)
         VECTOR_DB_PATH = os.getenv("VECTOR_DB_PATH", DEFAULT_DB_PATH)
 
         os.makedirs(os.path.dirname(VECTOR_DB_PATH), exist_ok=True)
-        os.makedirs(os.path.dirname(PDF_PATH), exist_ok=True)
-
-        logging.info(f"Using PDF path: {PDF_PATH}")
-        logging.info(f"Using Vector DB path: {VECTOR_DB_PATH}")
 
+        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 ---")
-            print(f"The required PDF file ('{os.path.basename(PDF_PATH)}') was not found at:")
-            print(f"  {PDF_PATH}")
-            print(f"Please ensure the file exists or set 'PDF_PATH' environment variable.")
-            print(f"---------------------------\n")
-            exit(1)
+            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
+
+        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
+
+        logging.info(f"PDF file '{os.path.basename(PDF_PATH)}' found and is readable.")
 
-        logging.info("Initializing Vector Database...")
         vector_db_instance = VectorDatabase(persist_directory=VECTOR_DB_PATH)
-        logging.info("Initializing RAG System...")
         rag = RAGSystem(vector_db=vector_db_instance)
 
-        logging.info(f"Loading/Verifying data from PDF: {PDF_PATH}")
-        states_loaded_count = rag.load_pdf(PDF_PATH)
-        doc_count = vector_db_instance.document_collection.count() if vector_db_instance.document_collection else 0
-        state_count = vector_db_instance.state_collection.count() if vector_db_instance.state_collection else 0
-        total_items = doc_count + state_count
+        rag.load_pdf(PDF_PATH)
 
-        if total_items > 0:
-            logging.info(f"Data loading/verification complete. Vector DB contains {total_items} items. Found {states_loaded_count} distinct states.")
-        else:
-            logging.warning("Potential issue: PDF processed but Vector DB appears empty. Check PDF content/format and logs.")
-            print("\nWarning: No data loaded from PDF or found in DB. Application might not function correctly.\n")
-
-        logging.info("Setting up Gradio interface...")
         app_interface = rag.gradio_interface()
+        SERVER_PORT = int(os.getenv("PORT", 7860)) # Use PORT env var if on Spaces/Cloud, else 7860
 
-        SERVER_PORT = 7860
         logging.info(f"Launching Gradio app on http://0.0.0.0:{SERVER_PORT}")
-        print("\n--- Gradio App Running ---")
-        print(f"Access the interface in your browser at: http://localhost:{SERVER_PORT} or http://<your-ip-address>:{SERVER_PORT}")
-        print("--------------------------\n")
-        app_interface.launch(
-            server_name="0.0.0.0",
-            server_port=SERVER_PORT,
-            share=True # Set to False if you don't want public sharing links
-        )
+        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
 
-    except FileNotFoundError as fnf_error:
-        logging.error(f"Initialization failed due to a missing file: {str(fnf_error)}", exc_info=True)
-        print(f"\n--- STARTUP ERROR: File Not Found ---")
-        print(f"{str(fnf_error)}")
-        print(f"---------------------------------------\n")
-        exit(1)
-    except ImportError as import_error:
-        logging.error(f"Import error: {str(import_error)}. Check dependencies.", exc_info=True)
-        print(f"\n--- STARTUP ERROR: Missing Dependency ---")
-        print(f"Import Error: {str(import_error)}")
-        print(f"Please ensure required libraries are installed (e.g., pip install -r requirements.txt).")
-        print(f"-----------------------------------------\n")
+    except ModuleNotFoundError as e:
+        if "vector_db" in str(e):
+             logging.error(f"FATAL: Could not import VectorDatabase. Ensure 'vector_db.py' is in the same directory and 'chromadb', 'langchain', 'pypdf', 'sentence-transformers' are installed.", exc_info=True)
+             print(f"\n--- MISSING DEPENDENCY OR FILE ---\nCould not find/import 'vector_db.py' or one of its dependencies.\nError: {e}\nPlease ensure 'vector_db.py' is present and all required packages (chromadb, langchain, pypdf, sentence-transformers, etc.) are in your requirements.txt and installed.\n---------------------------\n")
+        else:
+            logging.error(f"Application startup failed due to a missing module: {str(e)}", exc_info=True)
+            print(f"\n--- FATAL STARTUP ERROR - MISSING MODULE ---\n{str(e)}\nPlease ensure all dependencies are installed.\nCheck logs for more details.\n---------------------------\n")
         exit(1)
-    except RuntimeError as runtime_error:
-        logging.error(f"A runtime error occurred during setup: {str(runtime_error)}", exc_info=True)
-        print(f"\n--- STARTUP ERROR: Runtime Problem ---")
-        print(f"Runtime Error: {str(runtime_error)}")
-        print(f"Check logs for details, often related to data loading or DB setup.")
-        print(f"--------------------------------------\n")
+    except FileNotFoundError as e:
+        logging.error(f"Application startup failed due to a missing file: {str(e)}", exc_info=True)
+        print(f"\n--- FATAL STARTUP ERROR - FILE NOT FOUND ---\n{str(e)}\nPlease ensure the file exists at the specified path.\nCheck logs for more details.\n---------------------------\n")
         exit(1)
     except Exception as e:
-        logging.error(f"An unexpected error occurred during application startup: {str(e)}", exc_info=True)
-        print(f"\n--- FATAL STARTUP ERROR ---")
-        print(f"An unexpected error stopped the application: {str(e)}")
-        print(f"Check logs for detailed traceback.")
-        print(f"---------------------------\n")
+        logging.error(f"Application startup failed: {str(e)}", exc_info=True)
+        print(f"\n--- FATAL STARTUP ERROR ---\n{str(e)}\nCheck logs for more details.\n---------------------------\n")
         exit(1)