Add an endpoint to initialize the GUI

imagedephi/gui.py

@@ -5,7 +5,7 @@
 from pathlib import Path
 
 from fastapi import BackgroundTasks, FastAPI, Form, HTTPException, Request
-from fastapi.responses import HTMLResponse, PlainTextResponse
+from fastapi.responses import HTMLResponse, PlainTextResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
 from jinja2 import FunctionLoader
 from starlette.background import BackgroundTask
@@ -63,11 +63,22 @@ def reset_shutdown_event() -> None:
     shutdown_event.clear()
 
 
-@app.get("/", response_class=HTMLResponse)
+@app.get("/", response_class=RedirectResponse)
+def home(request: Request):
+    # On Windows, there may be multiple roots, so pick the one that's an ancestor of the CWD
+    # On Linux, this should typically resolve to "/"
+    root_directory = Path.cwd().root
+
+    return request.url_for("select_directory").include_query_params(
+        input_directory=str(root_directory), output_directory=str(root_directory)
+    )
+
+
+@app.get("/select-directory/", response_class=HTMLResponse)
 def select_directory(
     request: Request,
-    input_directory: Path = Path("/"),  # noqa: B008
-    output_directory: Path = Path("/"),  # noqa: B008
+    input_directory: Path,
+    output_directory: Path,
 ):
     # TODO: if input_directory is specified but an empty string, it gets instantiated as the CWD
     if not input_directory.is_dir():
@@ -85,7 +96,7 @@ def select_directory(
     )
 
 
-@app.post("/redact/")
+@app.post("/redact/", response_class=HTMLResponse)
 def redact(
     background_tasks: BackgroundTasks,
     input_directory: Path = Form(),  # noqa: B008

imagedephi/main.py

@@ -105,6 +105,7 @@ async def announce_ready() -> None:
         # To avoid race conditions, ensure that the webserver is
         # actually running before launching the browser
         await wait_for_port(port)
+        app.url_path_for("home")
         url = f"http://{host}:{port}/"
         click.echo(f"Server is running at {url} .")
         webbrowser.open(url)

tests/test_gui.py

@@ -11,8 +11,8 @@ def client() -> TestClient:
     return TestClient(app)
 
 
-def test_gui_select_directory(client: TestClient) -> None:
-    response = client.get("/")
+def test_gui_home(client: TestClient) -> None:
+    response = client.get(app.url_path_for("home"), follow_redirects=True)
 
     assert response.status_code == 200
     assert "Select Directory" in response.text