Fix webservice interactions with Docker

Author: jbarlow83

misc/_webservice.py

@@ -16,6 +16,7 @@
 import subprocess
 import sys
 import time
+import urllib
 from functools import partial
 from operator import getitem
 from pathlib import Path
@@ -29,6 +30,17 @@
 
 from ocrmypdf._defaults import DEFAULT_ROTATE_PAGES_THRESHOLD
 
+
+def get_host_url_with_port(port: int) -> str:
+    """Get the host URL for the web service. Hacky."""
+    host_url = st.context.headers["host"]
+    try:
+        host, _streamlit_port = host_url.split(":", maxsplit=1)
+    except ValueError:
+        host = host_url
+    return f"//{host}:{port}"  # Use the same protocol
+
+
 st.title("OCRmyPDF Web Service")
 
 if not which("ttyd"):
@@ -218,7 +230,7 @@
         ttyd_proc = subprocess.Popen(
             ttyd_args + args, stdout=subprocess.PIPE, stderr=subprocess.PIPE
         )
-        iframe(src=f"http://localhost:{port}", height=400)
+        iframe(src=get_host_url_with_port(port), height=400)
 
         while ttyd_proc.poll() is None:
             ttyd_proc.poll()

misc/webservice.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 # SPDX-FileCopyrightText: 2025 James R. Barlow
 # SPDX-License-Identifier: AGPL-3.0-or-later
 

pyproject.toml

@@ -163,4 +163,5 @@ dev = [
   "pymupdf>=1.24.14",
   "streamlit-pdf-viewer>=0.0.19",
   "streamlit>=1.40.2",
+  "ipykernel>=6.29.5",
 ]