Update app.py

app.py

@@ -13,6 +13,7 @@ import json
 from typing import Optional, Dict, List, Any
 import uvicorn
 import traceback
+from pathlib import Path  # <-- FIX 1: Added this import
 
 # ============================================================================
 # 2. SHARED GLOBALS & CONFIGURATION
@@ -559,8 +560,13 @@ def get_schema_info_ui():
     """
     md = f"# 📚 Database Schema\n\n"
     md += f"**Repo:** [{INDEXED_REPO_ID}](https://huggingface.co/datasets/{INDEXED_REPO_ID})\n"
-    md += f"**Database File:** `{DB_PATH}`\n\n"
     
+    if DB_PATH:
+        md += f"**Database File:** `{DB_PATH}`\n\n"
+    else:
+        md += "**Database File:** `NOT LOADED`\n\n"
+        return md
+
     try:
         with get_db_connection() as conn:
             cursor = conn.cursor()
@@ -667,7 +673,7 @@ def create_gradio_ui():
                 raw_status = gr.Markdown()
                 raw_results = gr.DataFrame(label="SQL Results", interactive=False)
             
-            with gr.TabItem("📊 Schema"):
+            with gr.TabItem("📊 Schema") as schema_tab: # <-- FIX 2: Added variable 'schema_tab'
                 gr.Markdown("View the database schema and table counts.")
                 schema_btn = gr.Button("📊 Load Schema Info")
                 schema_output = gr.Markdown()
@@ -711,12 +717,13 @@ def create_gradio_ui():
             outputs=[raw_results, raw_status]
         )
         
-        # Use .load() to run this when the tab is first clicked
-        schema_output.load(
+        # --- FIX 2: Use schema_tab.select() to load data when the tab is clicked ---
+        schema_tab.select(
             fn=get_schema_info_ui,
             inputs=None,
             outputs=[schema_output]
         )
+        # We can also keep the button for a manual refresh
         schema_btn.click(
             fn=get_schema_info_ui,
             inputs=None,