| from pathlib import Path
|
| import os
|
| import sys
|
|
|
|
|
| ROOT = Path(__file__).resolve().parent
|
| ROOT_ENV = ROOT / ".env"
|
| LOCAL_ENV = ROOT / "data" / "local.env"
|
| SRC = ROOT / "src"
|
| if str(SRC) not in sys.path:
|
| sys.path.insert(0, str(SRC))
|
|
|
|
|
| def _load_local_env(path: Path, *, override: bool = False) -> None:
|
| if not path.exists():
|
| return
|
| for raw_line in path.read_text(encoding="utf-8").splitlines():
|
| line = raw_line.strip()
|
| if not line or line.startswith("#") or "=" not in line:
|
| continue
|
| key, value = line.split("=", 1)
|
| key = key.strip()
|
| value = value.strip().strip('"').strip("'")
|
| if key and (override or key not in os.environ):
|
| os.environ[key] = value
|
|
|
|
|
| _load_local_env(ROOT_ENV, override=False)
|
| _load_local_env(LOCAL_ENV, override=False)
|
|
|
| from time_machine.application.container import create_container |
| from time_machine.ui.gradio_app import create_app |
| from time_machine.ui.realtime import realtime_voice_socket |
|
|
| import gradio as gr |
| from fastapi import FastAPI, WebSocket |
| from fastapi.responses import JSONResponse, RedirectResponse, Response |
| from fastapi.staticfiles import StaticFiles |
|
|
|
|
| gr.set_static_paths([ROOT / "static"]) |
| container = create_container() |
| demo = create_app(container) |
| fastapi_app = FastAPI(title="AI Time Machine") |
| fastapi_app.mount("/static", StaticFiles(directory=ROOT / "static"), name="static") |
|
|
|
|
| @fastapi_app.websocket("/realtime/voice")
|
| async def websocket_realtime_voice(websocket: WebSocket):
|
| await realtime_voice_socket(websocket, container)
|
|
|
|
|
| @fastapi_app.get("/manifest.json") |
| async def manifest(): |
| return JSONResponse( |
| { |
| "name": "AI Time Machine", |
| "short_name": "AI Time Machine", |
| "start_url": "/", |
| "display": "standalone", |
| "background_color": "#0b1117", |
| "theme_color": "#0b1117", |
| } |
| ) |
|
|
|
|
| @fastapi_app.get("/favicon.ico") |
| async def favicon(): |
| return Response(status_code=204) |
|
|
|
|
| @fastapi_app.get("/blank") |
| @fastapi_app.get("/blank/") |
| @fastapi_app.get("/app") |
| @fastapi_app.get("/app/") |
| async def blank_redirect(): |
| return RedirectResponse(url="/") |
|
|
|
|
| runtime_js = getattr(demo, "time_machine_js", None) |
| fastapi_app_with_gradio = gr.mount_gradio_app(fastapi_app, demo, path="/app", js=runtime_js) |
| app = fastapi_app_with_gradio if os.getenv("TIME_MACHINE_FASTAPI_WRAPPER") else demo |
|
|
|
|
| if __name__ == "__main__": |
| import uvicorn |
|
|
| host = os.getenv("GRADIO_SERVER_NAME", "0.0.0.0") |
| port = int(os.getenv("GRADIO_SERVER_PORT", "7860")) |
| if os.getenv("TIME_MACHINE_FASTAPI_WRAPPER"): |
| uvicorn.run(fastapi_app_with_gradio, host=host, port=port) |
| else: |
| demo.launch(server_name=host, server_port=port, js=runtime_js) |
|
|
