Improve: Add configuration checks, environment validation, and better error messages for missing HF_TOKEN and DATABASE_URL

.env.example

@@ -0,0 +1,23 @@
+# =========================================================
+# ENVIRONMENT CONFIGURATION
+# =========================================================
+
+# Hugging Face API Token (required for Mistral LLM features)
+# Get it from: https://huggingface.co/settings/tokens
+HF_TOKEN=your_hf_token_here
+
+# PostgreSQL Database URL (required for database features)
+# Format: postgresql://username:password@host:port/database
+# Example: postgresql://user:pass@localhost:5432/vehicle_logs
+DATABASE_URL=postgresql://username:password@host:port/database_name
+
+# =========================================================
+# HOW TO SET UP IN HUGGING FACE SPACES
+# =========================================================
+#
+# 1. Go to your Space Settings
+# 2. Click "Repository secrets"
+# 3. Add HF_TOKEN and DATABASE_URL as secrets
+# 4. Restart the Space
+#
+# =========================================================

app.py

@@ -16,7 +16,11 @@ from database import (
     get_vehicles_by_state,
     get_hourly_traffic,
     get_top_plates,
-    get_suspicious_vehicles
+    get_suspicious_vehicles,
+    HF_TOKEN,
+    DATABASE_URL,
+    client,
+    engine
 )
 
 # =========================================================
@@ -279,6 +283,45 @@ AI-powered Vehicle Detection + NLP-to-SQL Intelligence Platform
     else:
         gr.Warning(msg)
 
+    # =====================================================
+    # CONFIGURATION STATUS
+    # =====================================================
+
+    gr.Markdown("### 🔧 Configuration Status")
+
+    config_status = []
+
+    if HF_TOKEN:
+        gr.Success("✅ Mistral LLM Configured - NLP features enabled")
+    else:
+        gr.Warning("❌ HF_TOKEN missing - NLP/AI features disabled. Add HF_TOKEN to Space Secrets")
+        config_status.append("HF_TOKEN")
+
+    if DATABASE_URL:
+        gr.Success("✅ Database Configured - Query features enabled")
+    else:
+        gr.Warning("❌ DATABASE_URL missing - Database features disabled. Add DATABASE_URL to Space Secrets")
+        config_status.append("DATABASE_URL")
+
+    if config_status:
+        with gr.Group():
+            gr.Markdown(f"""
+### ⚙️ Setup Instructions
+
+Your Space is **missing these environment variables**:
+- **{', '.join(config_status)}**
+
+**To fix this:**
+1. Go to Space Settings → Repository secrets
+2. Add the following variables:
+   - `HF_TOKEN`: Get from https://huggingface.co/settings/tokens
+   - `DATABASE_URL`: Your PostgreSQL connection string
+3. Restart the Space
+
+Without these, NLP queries and database operations won't work.
+""")
+
+
     # =====================================================
     # TAB 1 - DETECTION
     # =====================================================

database.py

@@ -22,15 +22,28 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 DATABASE_URL = os.getenv("DATABASE_URL")
 
 # Initialize Mistral client
-client = InferenceClient(
-    model="mistralai/Mistral-7B-Instruct-v0.2",
-    token=HF_TOKEN
-)
+client = None
+try:
+    if HF_TOKEN:
+        client = InferenceClient(
+            model="mistralai/Mistral-7B-Instruct-v0.2",
+            token=HF_TOKEN
+        )
+        print("✅ Mistral client initialized")
+    else:
+        print("⚠️ HF_TOKEN not set - LLM features disabled")
+except Exception as e:
+    print(f"⚠️ Mistral client error: {e}")
+    client = None
 
 # Initialize database engine
+engine = None
 try:
-    engine = create_engine(DATABASE_URL)
-    print("✅ Database connection initialized")
+    if DATABASE_URL:
+        engine = create_engine(DATABASE_URL)
+        print("✅ Database connection initialized")
+    else:
+        print("⚠️ DATABASE_URL not set - Database features disabled")
 except Exception as e:
     print(f"⚠️ Database connection warning: {e}")
     engine = None
@@ -659,6 +672,10 @@ Generate PostgreSQL SQL query for:
 
     try:
 
+        if client is None:
+            print("❌ Mistral client not initialized - HF_TOKEN missing")
+            raise Exception("LLM service unavailable - HF_TOKEN not configured")
+
         response = client.chat_completion(
             messages=[
                 {
@@ -729,6 +746,14 @@ def run_query(user_query):
         print(sql)
         print("==============================")
 
+        if engine is None:
+            return {
+                "query": user_query,
+                "error": "❌ Database not configured - DATABASE_URL missing",
+                "sql": sql,
+                "result": []
+            }
+
         with engine.connect() as conn:
 
             result = conn.execute(text(sql))