| """Hugging Face Space entry point (Gradio SDK on ZeroGPU hardware). | |
| ZeroGPU expects the Space to be served by Gradio's own launch() and at least one @spaces.GPU function. | |
| So Gradio owns port 7860, and the Cortex FastAPI routes are inserted ahead of Gradio's own routes, | |
| which makes "/" serve the Cortex UI. | |
| """ | |
| import os | |
| import threading | |
| import gradio as gr | |
| import spaces | |
| from app.bootstrap import STATUS, start_background | |
| from app.server import app as api, housekeeping | |
| def gpu_embed(texts): | |
| """Embeds document chunks; registered for ZeroGPU (runs on CPU outside ZeroGPU).""" | |
| from core.embeddings import LocalEmbeddings | |
| return LocalEmbeddings().embed_documents(texts) | |
| with gr.Blocks(title="Cortex status") as demo: | |
| gr.Markdown("## Cortex status\nThe Cortex app is served at [/](/).") | |
| out = gr.JSON() | |
| gr.Button("Refresh").click(lambda: STATUS, outputs=out) | |
| if __name__ == "__main__": | |
| start_background() | |
| threading.Thread(target=housekeeping, daemon=True).start() | |
| # ssr_mode=False: on Spaces, Gradio SSR puts a Node server in front of 7860, which would hide the Cortex routes. | |
| demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")), ssr_mode=False, prevent_thread_lock=True) | |
| # Cortex routes take priority over Gradio's (the Gradio page itself is not used). | |
| demo.app.router.routes[0:0] = list(api.router.routes) | |
| demo.block_thread() | |