Cleanup init commit
This commit is contained in:
@@ -0,0 +1,56 @@
|
||||
"""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()
|
||||
Reference in New Issue
Block a user