fix: add PYTHONPATH environment variable and ensure src is importable in app.py

Dockerfile

@@ -26,6 +26,7 @@ ENV TOKENIZERS_PARALLELISM=false
 ENV MESA_GL_VERSION_OVERRIDE=3.3
 ENV PYOPENGL_PLATFORM=egl
 ENV PYTHONUNBUFFERED=1
+ENV PYTHONPATH=/app
 
 EXPOSE 7860
 CMD ["python", "spaces/app.py"]

spaces/app.py

@@ -1,10 +1,17 @@
 from __future__ import annotations
 
 import os
+import sys
+from pathlib import Path
 
 os.environ.setdefault("MODEL_CACHE_DIR", "/data/models")
 os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
+# Ensure `src` is importable when running `python spaces/app.py`.
+ROOT = Path(__file__).resolve().parent.parent
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
 import uvicorn
 
 from src.api.main import app