Upload 10 files

Dockerfile

@@ -11,6 +11,7 @@ COPY services ./services
 COPY utils ./utils
 COPY main.py ./main.py
 
+COPY --from=frontend /app/frontend/dist ./static
 
 EXPOSE 7860
 

main.py

@@ -21,4 +21,12 @@ async def start_aggregator_service():
                 storage.aggregate(repo_id)
             await asyncio.sleep(5)
 
-    asyncio.create_task(loop())
+    asyncio.create_task(loop())
+
+
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
+
+
+@app.get("/{full_path:path}")
+async def serve_spa():
+    return FileResponse("static/index.html")

services/registry.py

@@ -69,6 +69,7 @@ class RepoConnection:
                                 try:
                                     data = json.loads(data_str)
                                     stage = data.get("compute", {}).get("status", {}).get("stage", "Unknown")
+                                    logger.info(f"Stopped connections for {self.namespace}/{self.repo}  {stage}")
                                     storage.update_status(self.repo_id, "CONNECTED", stage)
 
                                     if stage == "Running" and not self._metrics_active: