fix(ui): resolve static SPA routing for /, /index.html, and /studio endpoints

Author: ImdataScientistSachin

codeslim/ui/server.py

@@ -4,8 +4,10 @@
 Provides REST API endpoints for single-file analysis, optimization, codebase scanning,
 AI provider health checks, git hook installation, and interactive HTML report export.
 """
+
 from pathlib import Path
 from typing import Any
+
 import httpx
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse
@@ -24,6 +26,7 @@
 log = get_logger("codeslim.ui.server")
 settings = get_settings()
 
+
 # Pydantic Schemas
 class CodeAnalysisRequest(BaseModel):
     code: str = Field(..., description="Python source code snippet")
@@ -234,12 +237,17 @@ def export_html(directory: str = ".") -> HTMLResponse:
         html_content = generate_html_observatory_report(report)
         return HTMLResponse(content=html_content, status_code=200)
 
-    # 7. Serve Web Studio SPA Front-End
+    # 7. Serve Web Studio SPA Front-End at /, /index.html, and /studio
     @app.get("/", response_class=HTMLResponse)
+    @app.get("/index.html", response_class=HTMLResponse)
+    @app.get("/studio", response_class=HTMLResponse)
     def studio_home() -> HTMLResponse:
         index_file = static_dir / "index.html"
         if not index_file.exists():
-            return HTMLResponse("<h1>CodeSlim Web Studio Initializing...</h1>", status_code=200)
+            return HTMLResponse(
+                "<h1>CodeSlim Web Studio Initializing...</h1><p>Please refresh in a moment.</p>",
+                status_code=200,
+            )
         return HTMLResponse(content=index_file.read_text(encoding="utf-8"), status_code=200)
 
     app.mount("/static", StaticFiles(directory=static_dir), name="static")

pyproject.toml

@@ -45,6 +45,9 @@ codeslim = "codeslim.cli:main"
 where = ["."]
 include = ["codeslim*"]
 
+[tool.setuptools.package-data]
+codeslim = ["ui/static/*.html", "ui/static/*"]
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 markers = ["llm: tests that make real LLM calls (slow)"]

tests/ui/test_server.py

@@ -10,6 +10,18 @@
 client = TestClient(app)
 
 
+def test_ui_root_endpoint():
+    response = client.get("/")
+    assert response.status_code == 200
+    assert "CodeSlim Web Studio" in response.text
+
+
+def test_ui_studio_endpoint():
+    response = client.get("/studio")
+    assert response.status_code == 200
+    assert "CodeSlim Web Studio" in response.text
+
+
 def test_ui_healthcheck():
     response = client.get("/api/v1/healthcheck")
     assert response.status_code == 200
@@ -32,7 +44,7 @@ def test_ui_analyze_endpoint():
     assert response.status_code == 200
     data = response.json()
     assert data["grade"] in ["A", "B", "C", "D", "F"]
-    assert "sys" in data["dead_code_items"] or len(data["dead_code_items"]) >= 0
+    assert len(data["dead_code_items"]) >= 0
 
 
 def test_ui_optimize_endpoint_no_llm():
@@ -46,7 +58,6 @@ def test_ui_optimize_endpoint_no_llm():
 
 
 def test_ui_scan_endpoint(tmp_path):
-    # Create test python file in temp path
     test_file = tmp_path / "sample.py"
     test_file.write_text("import sys\ndef hello(): pass\n", encoding="utf-8")
 
@@ -55,3 +66,12 @@ def test_ui_scan_endpoint(tmp_path):
     data = response.json()
     assert data["total_files"] == 1
     assert data["overall_grade"] in ["A", "B", "C", "D", "F"]
+
+
+def test_ui_export_html_endpoint(tmp_path):
+    test_file = tmp_path / "sample.py"
+    test_file.write_text("import sys\ndef hello(): pass\n", encoding="utf-8")
+
+    response = client.get(f"/api/v1/export-html?directory={tmp_path}")
+    assert response.status_code == 200
+    assert "<!DOCTYPE html>" in response.text