Fix ID Troubleshoot

app/main.py

@@ -33,6 +33,15 @@ app.add_middleware(
 # Add JWT middleware to inject user info into request.state
 app.add_middleware(JWTMiddleware)
 
+# ── 404 Logger ────────────────────────────────────────────────────────
+@app.middleware("http")
+async def log_404_errors(request: Request, call_next):
+    response = await call_next(request)
+    if response.status_code == 404:
+        referer = request.headers.get("referer", "No Referer")
+        print(f"DEBUG 404: {request.method} {request.url} - Referer: {referer}")
+    return response
+
 
 # ── Startup ───────────────────────────────────────────────────────────
 @app.on_event("startup")

app/services/data_loader.py

@@ -89,8 +89,11 @@ class DataLoader:
             data = clean_nans(df.to_dict(orient='records'))
             df = pd.DataFrame(data)
             
-            # Add 'id' column for backward compatibility (1-indexed)
-            df['id'] = range(1, len(df) + 1)
+            # Note: We keep the 'id' from the SQL view 'companies_unified' (a.id)
+            # instead of overwriting it with a virtual range.
+            # This ensures stable IDs that match enriched_companies.company_id.
+            if 'id' not in df.columns:
+                df['id'] = range(1, len(df) + 1)
             
             # Recalculate capital divergence if not already handled by SQL
             # (In our current SQL view, we could add this, but keeping it here for safety for now)