Upload 26 files

Dockerfile

@@ -5,7 +5,7 @@ COPY frontend ./frontend
 WORKDIR /app/frontend
 RUN npm install && npm run build
 
-FROM python:3.11
+FROM python:3.11-slim
 
 WORKDIR /app
 COPY . .
@@ -13,6 +13,6 @@ COPY . .
 # copy frontend build
 COPY --from=frontend /app/frontend/dist ./static
 
-RUN pip install fastapi uvicorn httpx
+RUN pip install fastapi uvicorn httpx pydantic
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

api/routes/status.py

@@ -1,8 +1,7 @@
 from fastapi import APIRouter
-from services.storage import InMemoryStorage
+from services.storage import storage
 
 router = APIRouter()
-storage = InMemoryStorage()
 
 
 @router.get("/")

frontend/src/main.tsx

@@ -3,4 +3,14 @@ import ReactDOM from "react-dom/client";
 import App from "./App";
 import "./index.css";
 
-ReactDOM.createRoot(document.getElementById("root")!).render(<App />);
+import { QueryClient, QueryClientProvider } from "@tanstack/react-query";
+
+const queryClient = new QueryClient();
+
+ReactDOM.createRoot(document.getElementById("root")!).render(
+  <React.StrictMode>
+    <QueryClientProvider client={queryClient}>
+      <App />
+    </QueryClientProvider>
+  </React.StrictMode>
+);

services/connection_manager.py

@@ -35,6 +35,11 @@ class RepoConnection:
                         self.is_running = False
                 
             except Exception:
+                storage.set_status(self.repo_id, RepoStatus(
+                    state="ERROR",
+                    stage="Disconnected",
+                    last_updated=datetime.utcnow()
+                ))
                 attempt += 1
                 await exponential_backoff(attempt)
 

services/storage.py

@@ -85,6 +85,16 @@ class InMemoryStorage:
             return []
 
         return repo.get(range, [])
+        
+    def set_status(self, repo_id: str, status: RepoStatus):
+        self.status[repo_id] = status
+
+    def get_status(self, repo_id: str):
+        return self.status.get(repo_id, {
+            "state": "DISCONNECTED",
+            "stage": "Unknown",
+            "last_updated": None
+        })
 
 # Global instance
 storage = InMemoryStorage()