File size: 1,608 Bytes
00b2f48 4345562 00b2f48 4345562 00b2f48 4345562 00b2f48 4345562 00b2f48 4345562 00b2f48 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 | """Hugging Face Spaces entry point.
HF Spaces looks for ``app.py`` at the repo root. We just import the
actual app from ``demo/`` so the demo code stays tucked away and the
root stays uncluttered.
"""
from __future__ import annotations
import importlib.util
import sys
from pathlib import Path
_REPO_ROOT = Path(__file__).resolve().parent
# Add packages to path (needed before importing demo modules)
for pkg in ("pdfsys-core", "pdfsys-router", "pdfsys-parser-mupdf", "pdfsys-bench"):
src = _REPO_ROOT / "packages" / pkg / "src"
if src.is_dir() and str(src) not in sys.path:
sys.path.insert(0, str(src))
# Add demo dir to path (needed for pipeline import)
_DEMO_DIR = _REPO_ROOT / "demo"
if str(_DEMO_DIR) not in sys.path:
sys.path.insert(0, str(_DEMO_DIR))
# Load demo app from demo/app.py using importlib to avoid name collision
_demo_module_path = _DEMO_DIR / "app.py"
spec = importlib.util.spec_from_file_location("demo_app", _demo_module_path)
_demo_module = importlib.util.module_from_spec(spec)
sys.modules["demo_app"] = _demo_module
spec.loader.exec_module(_demo_module)
demo = _demo_module.demo # noqa: F401 — re-exported for HF Spaces
if __name__ == "__main__":
import os
import gradio as gr
CSS = """
.small-num input { font-weight: 600; font-size: 1.1rem; }
footer { display: none !important; }
"""
demo.queue(max_size=8).launch(
server_name=os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0"),
server_port=int(os.environ.get("GRADIO_SERVER_PORT", "7860")),
theme=gr.themes.Soft(primary_hue="emerald"),
css=CSS,
)
|