Update app.py

app.py

@@ -1,12 +1,13 @@
 import sqlite3
 import pandas as pd
 from fastapi import FastAPI
+from fastapi.responses import HTMLResponse
 
 DB_PATH = "CarOwners_Full_Final.db"
 
 app = FastAPI()
 
-# Connect to DB
+# Connect DB
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
 
@@ -27,69 +28,92 @@ def get_schema(table_name):
     return cursor.fetchall()
 
 
+# =========================
+# GET ROW COUNT
+# =========================
+def get_row_count(table_name):
+    cursor.execute(f"SELECT COUNT(*) FROM {table_name}")
+    return cursor.fetchone()[0]
+
+
 # =========================
 # GET FIRST 100 ROWS
 # =========================
 def get_preview(table_name):
-    query = f"SELECT * FROM {table_name} LIMIT 100;"
-    return pd.read_sql_query(query, conn)
+    return pd.read_sql_query(f"SELECT * FROM {table_name} LIMIT 100;", conn)
 
 
 # =========================
-# INSPECT DATABASE (startup log)
+# INSPECT DATABASE (PRINT EVERYTHING)
 # =========================
 def inspect_db():
     tables = get_tables()
-    print("\n📊 TABLES FOUND:", tables)
+
+    print("\n================ DATABASE INFO ================")
+    print("Tables:", tables)
 
     for table in tables:
         print(f"\n🔹 TABLE: {table}")
 
-        # schema
+        # Schema
         schema = get_schema(table)
         print("Columns:")
         for col in schema:
             print(col)
 
-        # row count
-        cursor.execute(f"SELECT COUNT(*) FROM {table}")
-        count = cursor.fetchone()[0]
+        # Row count
+        count = get_row_count(table)
         print("Total rows:", count)
 
-        # preview
+        # First 5 rows
         df = pd.read_sql_query(f"SELECT * FROM {table} LIMIT 5;", conn)
         print(df)
 
+    print("=============================================")
+
 
-# Run at startup
 inspect_db()
 
 
 # =========================
-# API ROUTES
+# UI HOMEPAGE
 # =========================
-
-@app.get("/")
+@app.get("/", response_class=HTMLResponse)
 def home():
-    return {"message": "Car DB API is running"}
+    tables = get_tables()
 
+    html = "<h1>🚗 Car Database Explorer</h1>"
 
-@app.get("/tables")
-def tables():
-    return {"tables": get_tables()}
+    if not tables:
+        return "<h2>❌ No tables found. DB not loaded properly.</h2>"
 
+    for table in tables:
+        html += f"<h2>📊 Table: {table}</h2>"
+
+        # Row count
+        count = get_row_count(table)
+        html += f"<p><b>Total Rows:</b> {count}</p>"
 
-@app.get("/schema/{table_name}")
-def schema(table_name: str):
-    return {"schema": get_schema(table_name)}
+        # Schema
+        schema = get_schema(table)
+        html += "<h3>Columns:</h3><ul>"
+        for col in schema:
+            html += f"<li>{col}</li>"
+        html += "</ul>"
 
+        # First 100 rows
+        df = get_preview(table)
+        html += "<h3>First 100 Rows:</h3>"
+        html += df.to_html(index=False)
 
-@app.get("/preview/{table_name}")
-def preview(table_name: str):
-    df = get_preview(table_name)
-    return df.to_dict(orient="records")
+        html += "<hr>"
 
+    return html
 
+
+# =========================
+# SEARCH API
+# =========================
 @app.get("/search")
 def search(table: str, column: str, query: str):
     sql = f"""