Spaces:
Sleeping
Sleeping
| from __future__ import annotations | |
| import gradio as gr | |
| from loguru import logger | |
| from src.config import settings | |
| from src.embeddings import warm_up_embeddings | |
| from src.monitoring import add_monitoring_route | |
| from src.ui.helpers import CSS, THEME | |
| from src.ui.interactive import INTERACTIVE_HEAD_HTML | |
| def warm_up_if_enabled() -> bool: | |
| if not settings.embedding_warmup_enabled: | |
| return False | |
| try: | |
| latency_ms = warm_up_embeddings() | |
| except Exception as exc: | |
| logger.warning("embedding_warmup_failed error={}", type(exc).__name__) | |
| return False | |
| logger.info("embedding_warmup_completed latency_ms={:.3f}", latency_ms) | |
| return True | |
| def launch_demo(demo: gr.Blocks) -> None: | |
| settings.export_dir.mkdir(parents=True, exist_ok=True) | |
| _ = warm_up_if_enabled() | |
| auth = None | |
| if settings.gradio_username and settings.gradio_password: | |
| auth = (settings.gradio_username, settings.gradio_password) | |
| queued_demo = demo.queue(default_concurrency_limit=2) | |
| add_monitoring_route(queued_demo.app) | |
| _ = queued_demo.launch( | |
| allowed_paths=[str(settings.export_dir.resolve())], | |
| auth=auth, | |
| css=CSS, | |
| head=INTERACTIVE_HEAD_HTML, | |
| max_file_size=settings.max_upload_bytes, | |
| server_name=settings.server_name, | |
| server_port=settings.server_port, | |
| theme=THEME, | |
| _app=queued_demo.app, | |
| ) | |