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. SEND_EVERY_SECONDS = 1.0  # Upload rate
  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=2)
  31.         return r.ok
  32.     except:
  33.         return False
  34.  
  35. # =========================
  36. # 2) MAX30102 REGISTERS
  37. # =========================
  38. MAX30102_ADDR = 0x57
  39. REG_FIFO_DATA = 0x07
  40. REG_FIFO_WR_PTR = 0x04
  41. REG_FIFO_RD_PTR = 0x06
  42. REG_MODE_CONFIG = 0x09
  43. REG_SPO2_CONFIG = 0x0A
  44. REG_LED1_PA = 0x0C # RED
  45. REG_LED2_PA = 0x0D # IR
  46.  
  47. def init_max30102(bus):
  48.     bus.write_byte_data(MAX30102_ADDR, REG_MODE_CONFIG, 0x40) # Reset
  49.     time.sleep(0.1)
  50.     bus.write_byte_data(MAX30102_ADDR, REG_MODE_CONFIG, 0x03) # SpO2 Mode
  51.     bus.write_byte_data(MAX30102_ADDR, REG_SPO2_CONFIG, 0x27) # 100Hz, 411us
  52.     bus.write_byte_data(MAX30102_ADDR, REG_LED1_PA, 0x24)    # Red LED
  53.     bus.write_byte_data(MAX30102_ADDR, REG_LED2_PA, 0x24)    # IR LED
  54.     bus.write_byte_data(MAX30102_ADDR, REG_FIFO_WR_PTR, 0x00)
  55.     bus.write_byte_data(MAX30102_ADDR, REG_FIFO_RD_PTR, 0x00)
  56.  
  57. def read_fifo(bus):
  58.     d = bus.read_i2c_block_data(MAX30102_ADDR, REG_FIFO_DATA, 6)
  59.     red = ((d[0] << 16) | (d[1] << 8) | d[2]) & 0x3FFFF
  60.     ir = ((d[3] << 16) | (d[4] << 8) | d[5]) & 0x3FFFF
  61.     return red, ir
  62.  
  63. # =========================
  64. # 3) BPM ALGORITHM
  65. # =========================
  66. def estimate_bpm_autocorr(ir_signal, fs_hz):
  67.     x = np.array(ir_signal, dtype=np.float64)
  68.     x = x - np.mean(x)
  69.     # Bandpass filter simulation (Smoothing)
  70.     w = 5
  71.     x = np.convolve(x, np.ones(w)/w, mode='same')
  72.     ac = np.correlate(x, x, mode='full')
  73.     ac = ac[len(ac)//2:]
  74.  
  75.     min_lag = int(fs_hz * 60 / 180) # 180 BPM
  76.     max_lag = int(fs_hz * 60 / 45)  # 45 BPM
  77.  
  78.     if max_lag >= len(ac): return None
  79.     seg = ac[min_lag:max_lag]
  80.     if len(seg) == 0: return None
  81.  
  82.     peak = np.argmax(seg) + min_lag
  83.     return 60.0 * fs_hz / peak
  84.  
  85. # =========================
  86. # 4) MAIN LOOP
  87. # =========================
  88. def main():
  89.     fs = 50
  90.     N = fs * 4 # 4-second window for faster response
  91.     ir_buf = []
  92.     last_send = 0
  93.     bpm_smooth = None
  94.  
  95.     with SMBus(1) as bus:
  96.         init_max30102(bus)
  97.         print("System Ready. Place finger on sensor...")
  98.  
  99.         while True:
  100.             # Read sample
  101.             try:
  102.                 red_raw, ir_raw = read_fifo(bus)
  103.             except:
  104.                 continue
  105.  
  106.             if ir_raw > 30000: # Finger detected
  107.                 ir_buf.append(ir_raw)
  108.                 if len(ir_buf) > N: ir_buf.pop(0)
  109.  
  110.                 if len(ir_buf) == N:
  111.                     bpm = estimate_bpm_autocorr(ir_buf, fs)
  112.                     if bpm and 40 < bpm < 180:
  113.                         bpm_smooth = bpm if bpm_smooth is None else 0.8*bpm_smooth + 0.2*bpm
  114.  
  115.                 print(f"RED: {red_raw} | IR: {ir_raw} | BPM: {int(bpm_smooth) if bpm_smooth else 'Calculating...'}")
  116.             else:
  117.                 ir_buf = []
  118.                 bpm_smooth = None
  119.                 print("No Finger Detected")
  120.  
  121.             # Supabase Upload
  122.             now = time.time()
  123.             if ir_raw > 30000 and (now - last_send) >= SEND_EVERY_SECONDS:
  124.                 success = send_to_supabase(bpm_smooth, ir_raw, red_raw)
  125.                 if success: print(">>> Sent to Supabase ✅")
  126.                 else: print(">>> Supabase Error ❌")
  127.                 last_send = now
  128.  
  129.             time.sleep(1/fs)
  130.  
  131. if __name__ == "__main__":
  132.     main()