Upload folder using huggingface_hub

README.md

@@ -31,7 +31,7 @@ pinned: false
 
 ---
 
-HeartFlow OS is an end-to-end telemetry and inference platform designed to stream cardiovascular vitals, predict anomalies using a soft-voting ensemble model, and export those models to highly-constrained edge devices (like ESP32/Arduino).
+HeartFlow OS streams cardiovascular vitals, predicts anomalies using a soft-voting ensemble, and exports models to highly-constrained edge devices (ESP32/Arduino).
 
 ## System Architecture
 

frontend/src/context/TelemetryContext.tsx

@@ -1,14 +1,13 @@
 "use client";
 
-import { createContext, useContext, useEffect, useState, useRef } from "react";
+import { createContext, useContext, useState, useCallback } from "react";
+import { useResilientSocket, ConnectionStatus } from "@/hooks/useResilientSocket";
 
 type StreamState = {
   hr: number[];
   temp: number[];
 };
 
-type ConnectionStatus = 'connected' | 'reconnecting' | 'disconnected';
-
 type TelemetryContextType = {
   data: any;
   stream: StreamState;
@@ -23,60 +22,20 @@ export function TelemetryProvider({ children }: { children: React.ReactNode }) {
     hr: Array(50).fill(75),
     temp: Array(50).fill(37.0)
   });
-  const [status, setStatus] = useState<ConnectionStatus>('disconnected');
-  
-  const reconnectTimeoutRef = useRef<NodeJS.Timeout | null>(null);
-  const reconnectAttemptRef = useRef(0);
-
-  useEffect(() => {
-    let ws: WebSocket;
-    let isComponentMounted = true;
-
-    const connect = () => {
-      ws = new WebSocket(process.env.NEXT_PUBLIC_WS_URL || "ws://localhost:8000/ws/feed");
-      
-      ws.onopen = () => {
-        if (!isComponentMounted) return;
-        setStatus('connected');
-        reconnectAttemptRef.current = 0; // Reset attempts
-      };
-
-      ws.onmessage = (event) => {
-        if (!isComponentMounted) return;
-        const msg = JSON.parse(event.data);
-        setData(msg);
-        if (msg?.sensor_data) {
-          setStream(prev => ({
-            hr: [...prev.hr.slice(1), msg.sensor_data.Heart_Rate],
-            temp: [...prev.temp.slice(1), msg.sensor_data.Body_Temp]
-          }));
-        }
-      };
 
-      ws.onclose = () => {
-        if (!isComponentMounted) return;
-        setStatus('reconnecting');
-        const timeout = Math.min(1000 * (2 ** reconnectAttemptRef.current), 30000); // Max 30s
-        reconnectAttemptRef.current += 1;
-        
-        reconnectTimeoutRef.current = setTimeout(connect, timeout);
-      };
-
-      ws.onerror = () => {
-        if (!isComponentMounted) return;
-        ws.close(); // Force trigger onclose
-      };
-    };
-
-    connect();
-
-    return () => {
-      isComponentMounted = false;
-      if (reconnectTimeoutRef.current) clearTimeout(reconnectTimeoutRef.current);
-      if (ws) ws.close();
-    };
+  const handleMessage = useCallback((msg: any) => {
+    setData(msg);
+    if (msg?.sensor_data) {
+      setStream(prev => ({
+        hr: [...prev.hr.slice(1), msg.sensor_data.Heart_Rate],
+        temp: [...prev.temp.slice(1), msg.sensor_data.Body_Temp]
+      }));
+    }
   }, []);
 
+  const wsUrl = process.env.NEXT_PUBLIC_WS_URL || "ws://localhost:8000/ws/feed";
+  const status = useResilientSocket(wsUrl, handleMessage);
+
   return (
     <TelemetryContext.Provider value={{ data, stream, status }}>
       {children}

frontend/src/hooks/useResilientSocket.ts

@@ -0,0 +1,59 @@
+import { useEffect, useRef, useState } from "react";
+
+export type ConnectionStatus = 'connected' | 'reconnecting' | 'disconnected';
+
+export function useResilientSocket(url: string, onMessage: (msg: any) => void) {
+  const [status, setStatus] = useState<ConnectionStatus>('disconnected');
+  const reconnectTimeoutRef = useRef<NodeJS.Timeout | null>(null);
+  const reconnectAttemptRef = useRef(0);
+  const savedCallback = useRef(onMessage);
+
+  useEffect(() => {
+    savedCallback.current = onMessage;
+  }, [onMessage]);
+
+  useEffect(() => {
+    let ws: WebSocket;
+    let isComponentMounted = true;
+
+    const connect = () => {
+      ws = new WebSocket(url);
+      
+      ws.onopen = () => {
+        if (!isComponentMounted) return;
+        setStatus('connected');
+        reconnectAttemptRef.current = 0; // Reset attempts
+      };
+
+      ws.onmessage = (event) => {
+        if (!isComponentMounted) return;
+        const msg = JSON.parse(event.data);
+        savedCallback.current(msg);
+      };
+
+      ws.onclose = () => {
+        if (!isComponentMounted) return;
+        setStatus('reconnecting');
+        const timeout = Math.min(1000 * (2 ** reconnectAttemptRef.current), 30000); // Max 30s
+        reconnectAttemptRef.current += 1;
+        
+        reconnectTimeoutRef.current = setTimeout(connect, timeout);
+      };
+
+      ws.onerror = () => {
+        if (!isComponentMounted) return;
+        ws.close(); // Force trigger onclose
+      };
+    };
+
+    connect();
+
+    return () => {
+      isComponentMounted = false;
+      if (reconnectTimeoutRef.current) clearTimeout(reconnectTimeoutRef.current);
+      if (ws) ws.close();
+    };
+  }, [url]);
+
+  return status;
+}