"""Run the J-Wash server.""" import argparse import os from pathlib import Path def main(): parser = argparse.ArgumentParser(description="Run the J-Wash server.") parser.add_argument( "--hf-cache", metavar="PATH", default=None, help="Hugging Face cache directory. Default: the shared HF cache " "(HF_HOME, else ~/.cache/huggingface). Pass a path — e.g. ./hf_cache — " "to keep downloads isolated in a project-local cache.", ) parser.add_argument( "--data-dir", metavar="PATH", default=None, help="Runtime data directory (history, frames, presets, edits). " "Default: ./data. Give each instance its own when running several " "servers side by side.", ) parser.add_argument( "--port", type=int, default=None, help="HTTP port (default: 8381). Use a distinct port per instance.", ) args = parser.parse_args() # Set the env overrides BEFORE importing config (which reads them). if args.data_dir: os.environ["JWASH_DATA_DIR"] = str(Path(args.data_dir).expanduser().resolve()) if args.hf_cache: os.environ["HF_HOME"] = str(Path(args.hf_cache).expanduser().resolve()) else: # HF cache chosen in the Options tab (data/settings.json); the # --hf-cache flag wins over it, the plain HF_HOME env loses to it. import json data_dir = Path( os.environ.get("JWASH_DATA_DIR") or Path(__file__).resolve().parent / "data" ) try: saved = json.loads((data_dir / "settings.json").read_text(encoding="utf-8")) if saved.get("hf_cache"): os.environ["HF_HOME"] = str(Path(saved["hf_cache"]).expanduser().resolve()) except Exception: pass import config config.setup_env() import uvicorn uvicorn.run("api.app:app", host=config.HOST, port=args.port or config.PORT, log_level="info") if __name__ == "__main__": main()