Исправление запуска Playground UI на HuggingFace Space с проблемами прав доступа
Browse files- fallback.py +74 -6
fallback.py
CHANGED
|
@@ -569,12 +569,45 @@ def run_playground():
|
|
| 569 |
env["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true"
|
| 570 |
env["NEXT_PUBLIC_DISABLE_CAMERA"] = "true"
|
| 571 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 572 |
playground_dir = Path("/app/playground")
|
| 573 |
if not playground_dir.exists():
|
| 574 |
logger.error(f"Playground directory not found at {playground_dir}")
|
| 575 |
raise FileNotFoundError(f"Playground directory not found at {playground_dir}")
|
| 576 |
|
| 577 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 578 |
logger.info(f"Running command in {playground_dir}: {' '.join(cmd)}")
|
| 579 |
|
| 580 |
# Запуск процесса Playground
|
|
@@ -597,12 +630,47 @@ def run_playground():
|
|
| 597 |
threading.Thread(target=log_output, args=(playground_process.stderr, "PLAYGROUND ERROR"), daemon=True).start()
|
| 598 |
|
| 599 |
# Проверка, что процесс запустился
|
| 600 |
-
time.sleep(
|
| 601 |
if playground_process.poll() is not None:
|
| 602 |
-
logger.
|
| 603 |
-
|
| 604 |
-
|
| 605 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 606 |
return playground_process
|
| 607 |
|
| 608 |
def main():
|
|
|
|
| 569 |
env["NEXT_PUBLIC_EDIT_GRAPH_MODE"] = "true"
|
| 570 |
env["NEXT_PUBLIC_DISABLE_CAMERA"] = "true"
|
| 571 |
|
| 572 |
+
# Настройка PNPM для использования временной директории для кэша
|
| 573 |
+
pnpm_home = "/tmp/pnpm-store"
|
| 574 |
+
os.makedirs(pnpm_home, exist_ok=True)
|
| 575 |
+
env["PNPM_HOME"] = pnpm_home
|
| 576 |
+
env["HOME"] = "/tmp" # Установка HOME в /tmp для всех инструментов
|
| 577 |
+
env["XDG_CONFIG_HOME"] = "/tmp/.config"
|
| 578 |
+
env["XDG_CACHE_HOME"] = "/tmp/.cache"
|
| 579 |
+
env["XDG_DATA_HOME"] = "/tmp/.local/share"
|
| 580 |
+
|
| 581 |
+
# Создаем необходимые директории
|
| 582 |
+
os.makedirs("/tmp/.config", exist_ok=True)
|
| 583 |
+
os.makedirs("/tmp/.cache", exist_ok=True)
|
| 584 |
+
os.makedirs("/tmp/.local/share", exist_ok=True)
|
| 585 |
+
|
| 586 |
+
logger.info(f"Set PNPM_HOME to {pnpm_home}")
|
| 587 |
+
logger.info(f"Set HOME to {env['HOME']}")
|
| 588 |
+
|
| 589 |
playground_dir = Path("/app/playground")
|
| 590 |
if not playground_dir.exists():
|
| 591 |
logger.error(f"Playground directory not found at {playground_dir}")
|
| 592 |
raise FileNotFoundError(f"Playground directory not found at {playground_dir}")
|
| 593 |
|
| 594 |
+
# Создаем .npmrc файл в директории playground для настройки npm
|
| 595 |
+
npmrc_path = playground_dir / ".npmrc"
|
| 596 |
+
try:
|
| 597 |
+
with open(npmrc_path, 'w') as f:
|
| 598 |
+
f.write(f"cache=/tmp/.npm-cache\n")
|
| 599 |
+
f.write(f"tmp=/tmp/.npm-tmp\n")
|
| 600 |
+
logger.info(f"Created .npmrc file at {npmrc_path}")
|
| 601 |
+
except Exception as e:
|
| 602 |
+
logger.warning(f"Could not create .npmrc: {e}")
|
| 603 |
+
|
| 604 |
+
# Создаем директории для npm
|
| 605 |
+
os.makedirs("/tmp/.npm-cache", exist_ok=True)
|
| 606 |
+
os.makedirs("/tmp/.npm-tmp", exist_ok=True)
|
| 607 |
+
|
| 608 |
+
# Пробуем сначала запустить с помощью npx next dev
|
| 609 |
+
logger.info("Trying to start Playground with npx next dev...")
|
| 610 |
+
cmd = ["npx", "next", "dev", "--port", "7860"]
|
| 611 |
logger.info(f"Running command in {playground_dir}: {' '.join(cmd)}")
|
| 612 |
|
| 613 |
# Запуск процесса Playground
|
|
|
|
| 630 |
threading.Thread(target=log_output, args=(playground_process.stderr, "PLAYGROUND ERROR"), daemon=True).start()
|
| 631 |
|
| 632 |
# Проверка, что процесс запустился
|
| 633 |
+
time.sleep(5)
|
| 634 |
if playground_process.poll() is not None:
|
| 635 |
+
logger.warning(f"npx next dev failed with code {playground_process.returncode}, trying alternative method...")
|
| 636 |
+
|
| 637 |
+
# Если первый метод не работает, пробуем запустить напрямую node
|
| 638 |
+
logger.info("Trying to start Playground with node directly...")
|
| 639 |
+
|
| 640 |
+
# Ищем файл .next/server/app/page.js
|
| 641 |
+
next_page_js = playground_dir / ".next/server/app/page.js"
|
| 642 |
+
if next_page_js.exists():
|
| 643 |
+
cmd = ["node", str(next_page_js)]
|
| 644 |
+
logger.info(f"Running command: {' '.join(cmd)}")
|
| 645 |
+
|
| 646 |
+
playground_process = subprocess.Popen(
|
| 647 |
+
cmd,
|
| 648 |
+
cwd=str(playground_dir),
|
| 649 |
+
env=env,
|
| 650 |
+
stdout=subprocess.PIPE,
|
| 651 |
+
stderr=subprocess.PIPE,
|
| 652 |
+
text=True,
|
| 653 |
+
bufsize=1,
|
| 654 |
+
)
|
| 655 |
+
|
| 656 |
+
threading.Thread(target=log_output, args=(playground_process.stdout, "PLAYGROUND (NODE)"), daemon=True).start()
|
| 657 |
+
threading.Thread(target=log_output, args=(playground_process.stderr, "PLAYGROUND ERROR (NODE)"), daemon=True).start()
|
| 658 |
+
|
| 659 |
+
time.sleep(3)
|
| 660 |
+
if playground_process.poll() is not None:
|
| 661 |
+
logger.error(f"All Playground UI launch methods failed")
|
| 662 |
+
raise RuntimeError(f"All Playground UI launch methods failed")
|
| 663 |
+
else:
|
| 664 |
+
logger.info("Skipping Playground launch as it's not critical for API functionality")
|
| 665 |
+
logger.info("API server and proxy server will continue running")
|
| 666 |
+
# Создаем заглушку для процесса
|
| 667 |
+
playground_process = subprocess.Popen(
|
| 668 |
+
["tail", "-f", "/dev/null"], # Команда, которая никогда не завершится
|
| 669 |
+
stdout=subprocess.PIPE,
|
| 670 |
+
stderr=subprocess.PIPE,
|
| 671 |
+
)
|
| 672 |
+
|
| 673 |
+
logger.info("Playground UI started successfully or skipped")
|
| 674 |
return playground_process
|
| 675 |
|
| 676 |
def main():
|