Spaces:
Running on Zero
Running on Zero
| """Hugging Face Gradio entry point with the React dashboard mounted in-app.""" | |
| from __future__ import annotations | |
| import os | |
| import sys | |
| from pathlib import Path | |
| os.environ["GRADIO_SSR_MODE"] = "false" | |
| import gradio as gr | |
| from starlette.routing import Mount | |
| ROOT = Path(__file__).resolve().parent | |
| sys.path.insert(0, str(ROOT)) | |
| sys.path.insert(0, str(ROOT / "src")) | |
| sys.path.insert(0, str(ROOT / "src" / "near_miss")) | |
| try: | |
| import src.near_miss as near_miss | |
| sys.modules["near_miss"] = near_miss | |
| except Exception: | |
| pass | |
| from dashboard_api import app as fastapi_app | |
| try: | |
| import spaces | |
| except ImportError: | |
| class spaces: | |
| def GPU(fn=None, duration=120): | |
| if fn is None: | |
| return lambda function: function | |
| return fn | |
| def zero_gpu_pipeline_worker(value: str = "") -> str: | |
| """Register the Space's ZeroGPU worker with Hugging Face's supervisor.""" | |
| return value | |
| with gr.Blocks(title="AI Traffic Adaptation System") as demo: | |
| gr.HTML( | |
| """ | |
| <style> | |
| html, body, .gradio-container { | |
| max-width: 100% !important; | |
| width: 100% !important; | |
| height: 100% !important; | |
| margin: 0 !important; | |
| padding: 0 !important; | |
| overflow: hidden !important; | |
| background: #020617 !important; | |
| } | |
| footer { display: none !important; } | |
| </style> | |
| <div style="width:100%;height:100vh;margin:0;padding:0;overflow:hidden;background:#020617"> | |
| <iframe src="/dashboard/" title="AI Traffic Adaptation Dashboard" | |
| style="width:100%;height:100%;border:0;display:block"></iframe> | |
| </div> | |
| """ | |
| ) | |
| # Gradio registers catch-all routes during Blocks construction. Put the mounted | |
| # application first so its API, static files, and WebSocket routes are matched. | |
| demo.app.router.routes.insert( | |
| 0, | |
| Mount("/dashboard", app=fastapi_app, name="traffic_dashboard"), | |
| ) | |
| app = demo.app | |
| if __name__ == "__main__": | |
| demo.launch( | |
| server_name="0.0.0.0", | |
| server_port=int(os.environ.get("PORT", "7860")), | |
| theme=gr.themes.Soft(primary_hue="indigo"), | |
| _app=app, | |
| ) | |