1. import time
  2. import numpy as np
  3. from smbus2 import SMBus
  4. import requests
  5.  
  6. # =========================
  7. # 1) SUPABASE CONFIG
  8. # =========================
  9. SUPABASE_URL = "https://bnrqoovyawxyjbdqmtix.supabase.co";
  10. SUPABASE_ANON_KEY = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImJucnFvb3Z5YXd4eWpiZHFtdGl4Iiwicm9sZSI6ImFub24iLCJpYXQiOjE3NjE2NjA3ODEsImV4cCI6MjA3NzIzNjc4MX0.2aXc6ZDLss-nXsj2WQNkvD9AnImr73xVGExKK1LQerM"
  11. TABLE = "heart_reading"
  12.  
  13. UPLOAD_INTERVAL = 5.0  # Upload every 5 seconds
  14.  
  15. def send_to_supabase(bpm, ir, red, spo2=98):
  16.     url = f"{SUPABASE_URL}/rest/v1/{TABLE}"
  17.     headers = {
  18.         "apikey": SUPABASE_ANON_KEY,
  19.         "Authorization": f"Bearer {SUPABASE_ANON_KEY}",
  20.         "Content-Type": "application/json",
  21.         "Prefer": "return=minimal",
  22.     }
  23.     payload = {
  24.         "bpm": int(round(bpm)) if bpm else None,
  25.         "spo2": int(spo2),
  26.         "ir": int(ir),
  27.         "red": int(red)
  28.     }
  29.     try:
  30.         r = requests.post(url, headers=headers, json=payload, timeout=5)
  31.         return r.ok
  32.     except:
  33.         return False
  34.  
  35. # =========================
  36. # 2) MAX30102 CORE
  37. # =========================
  38. MAX30102_ADDR = 0x57
  39. REG_FIFO_DATA = 0x07
  40. REG_MODE_CONFIG = 0x09
  41. REG_SPO2_CONFIG = 0x0A
  42. REG_LED1_PA = 0x0C 
  43. REG_LED2_PA = 0x0D 
  44.  
  45. def init_max30102(bus):
  46.     bus.write_byte_data(MAX30102_ADDR, REG_MODE_CONFIG, 0x40) # Reset
  47.     time.sleep(0.1)
  48.     bus.write_byte_data(MAX30102_ADDR, REG_MODE_CONFIG, 0x03) # SpO2 Mode
  49.     bus.write_byte_data(MAX30102_ADDR, REG_SPO2_CONFIG, 0x27) # 100Hz
  50.     bus.write_byte_data(MAX30102_ADDR, REG_LED1_PA, 0x24)    
  51.     bus.write_byte_data(MAX30102_ADDR, REG_LED2_PA, 0x24)    
  52.  
  53. def read_fifo(bus):
  54.     d = bus.read_i2c_block_data(MAX30102_ADDR, REG_FIFO_DATA, 6)
  55.     red = ((d[0] << 16) | (d[1] << 8) | d[2]) & 0x3FFFF
  56.     ir = ((d[3] << 16) | (d[4] << 8) | d[5]) & 0x3FFFF
  57.     return red, ir
  58.  
  59. def estimate_bpm_autocorr(ir_signal, fs_hz):
  60.     x = np.array(ir_signal, dtype=np.float64)
  61.     x = x - np.mean(x)
  62.     w = 3 # Smaller smoothing window for smaller buffer
  63.     x = np.convolve(x, np.ones(w)/w, mode='same')
  64.     ac = np.correlate(x, x, mode='full')
  65.     ac = ac[len(ac)//2:]
  66.     min_lag = int(fs_hz * 60 / 180) 
  67.     max_lag = int(fs_hz * 60 / 45)  
  68.     if max_lag >= len(ac): return None
  69.     seg = ac[min_lag:max_lag]
  70.     if len(seg) == 0: return None
  71.     return 60.0 * fs_hz / (np.argmax(seg) + min_lag)
  72.  
  73. # =========================
  74. # 3) MAIN LOGIC
  75. # =========================
  76. def main():
  77.     fs = 50
  78.     N = fs * 2 # <--- CHANGED: Buffer is now 2 seconds (100 samples)
  79.     ir_buf = []
  80.     last_upload_time = time.time()
  81.     bpm_smooth = None
  82.  
  83.     with SMBus(1) as bus:
  84.         init_max30102(bus)
  85.         print("System Online. Buffer: 2 Seconds. Uploading every 5 seconds.")
  86.  
  87.         while True:
  88.             try:
  89.                 red_raw, ir_raw = read_fifo(bus)
  90.             except:
  91.                 continue
  92.  
  93.             if ir_raw > 35000: # Finger detected
  94.                 ir_buf.append(ir_raw)
  95.                 if len(ir_buf) > N: ir_buf.pop(0)
  96.  
  97.                 if len(ir_buf) == N:
  98.                     bpm = estimate_bpm_autocorr(ir_buf, fs)
  99.                     if bpm and 40 < bpm < 180:
  100.                         # Smoothing factor (slightly faster response)
  101.                         bpm_smooth = bpm if bpm_smooth is None else 0.7*bpm_smooth + 0.3*bpm
  102.  
  103.                 print(f"RED: {red_raw} | IR: {ir_raw} | BPM: {int(bpm_smooth) if bpm_smooth else 'Wait 2s...'}")
  104.             else:
  105.                 ir_buf = []
  106.                 bpm_smooth = None
  107.                 print("Searching for Finger...")
  108.  
  109.             # Upload Check
  110.             current_time = time.time()
  111.             if (current_time - last_upload_time) >= UPLOAD_INTERVAL:
  112.                 if ir_raw > 35000 and bpm_smooth is not None:
  113.                     print(">>> Uploading snapshot...")
  114.                     send_to_supabase(bpm_smooth, ir_raw, red_raw)
  115.                 last_upload_time = current_time
  116.  
  117.             time.sleep(1/fs)
  118.  
  119. if __name__ == "__main__":
  120.     main()