File size: 4,401 Bytes
cc2ed2f | 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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | """Лаунчер двух процессов Ru2SQL: FastAPI + Streamlit.
Запуск:
python scripts/run_app.py
Скрипт стартует uvicorn в фоне, ждёт пока /health начнёт отвечать
(модель Qwen может качаться 5–10 минут при первом запуске), и
поднимает Streamlit как основной процесс. По Ctrl+C корректно
останавливает оба.
Полезно при разработке и для демо на защите: один Ctrl+C — и оба
процесса завершены, не нужно искать висящий uvicorn в taskmgr.
"""
from __future__ import annotations
import os
import signal
import subprocess
import sys
import time
from pathlib import Path
import httpx
ROOT = Path(__file__).resolve().parent.parent
API_HOST = os.environ.get("API_HOST", "127.0.0.1")
API_PORT = int(os.environ.get("API_PORT", "8000"))
STREAMLIT_PORT = int(os.environ.get("STREAMLIT_PORT", "8501"))
# Streamlit слушает 0.0.0.0 в контейнере (нужен внешний доступ),
# а локально по умолчанию 127.0.0.1 — это управляется отдельной переменной.
STREAMLIT_HOST = os.environ.get("STREAMLIT_HOST", "127.0.0.1")
API_URL = f"http://{API_HOST}:{API_PORT}"
WAIT_API_SECONDS = 600 # модель скачивается долго
def wait_for_api(timeout: int) -> bool:
"""Опрашиваем /health, пока не получим 200 или не выйдет timeout."""
deadline = time.time() + timeout
while time.time() < deadline:
try:
r = httpx.get(f"{API_URL}/health", timeout=2.0)
if r.status_code == 200:
return True
except Exception:
pass
time.sleep(1.5)
return False
def main():
print(f"[run_app] корень проекта: {ROOT}")
print(f"[run_app] стартую uvicorn на {API_URL}")
creation_flags = 0
if sys.platform == "win32":
creation_flags = subprocess.CREATE_NEW_PROCESS_GROUP
uvicorn_proc = subprocess.Popen(
[
sys.executable, "-m", "uvicorn", "src.api.main:app",
"--host", API_HOST,
"--port", str(API_PORT),
],
cwd=str(ROOT),
creationflags=creation_flags,
)
try:
print(f"[run_app] жду готовности API ({WAIT_API_SECONDS} сек макс) — "
"при первом запуске Qwen качается с HuggingFace")
if not wait_for_api(WAIT_API_SECONDS):
print("[run_app] API не поднялся за отведённое время. Проверь логи uvicorn.")
uvicorn_proc.terminate()
sys.exit(1)
print(f"[run_app] API готов, стартую Streamlit на http://{STREAMLIT_HOST}:{STREAMLIT_PORT}")
env = os.environ.copy()
env["RU2SQL_API_URL"] = API_URL
streamlit_cmd = [
sys.executable, "-m", "streamlit", "run", "streamlit_app.py",
"--server.port", str(STREAMLIT_PORT),
"--server.address", STREAMLIT_HOST,
]
streamlit_proc = subprocess.Popen(streamlit_cmd, cwd=str(ROOT), env=env)
# Главный цикл — ждём, пока streamlit или uvicorn не упадёт
try:
while True:
if uvicorn_proc.poll() is not None:
print("[run_app] uvicorn завершился, останавливаю streamlit")
streamlit_proc.terminate()
break
if streamlit_proc.poll() is not None:
print("[run_app] streamlit завершился, останавливаю uvicorn")
break
time.sleep(1.0)
except KeyboardInterrupt:
print("\n[run_app] Ctrl+C получен, завершаю процессы…")
streamlit_proc.terminate()
finally:
try:
uvicorn_proc.terminate()
uvicorn_proc.wait(timeout=5)
except subprocess.TimeoutExpired:
uvicorn_proc.kill()
except Exception:
pass
print("[run_app] оба процесса остановлены")
if __name__ == "__main__":
main()
|