# Capteur intelligent avec IA embarquée
import machine, network, time
import urequests as requests
from machine import Pin, I2C, PWM
import tensorflow_lite as tflite

class SmartSensor:
    def __init__(self):
        self.led = Pin(2, Pin.OUT)
        self.sensor = ADC(Pin(36))
        self.model = tflite.load('anomaly_detection.tflite')
        
    def connect_wifi(self):
        wlan = network.WLAN(network.STA_IF)
        wlan.active(True)
        wlan.connect('IoT_Network', 'password')
        
        while not wlan.isconnected():
            self.led.on()
            time.sleep(0.1)
            self.led.off()
            time.sleep(0.1)
            
        print('📡 Connecté:', wlan.ifconfig()[0])
        
    def predict_anomaly(self, sensor_data):
        """IA embarquée pour détection d'anomalies"""
        input_tensor = tflite.tensor(sensor_data)
        self.model.invoke(input_tensor)
        prediction = self.model.output_tensor(0)
        
        return prediction[0] > 0.8  # Seuil anomalie
        
    def send_alert(self, data):
        """Envoi alerte si anomalie détectée"""
        payload = {
            'device_id': 'esp32_001',
            'alert_type': 'anomaly_detected',
            'sensor_data': data,
            'timestamp': time.time(),
            'confidence': 0.95
        }
        
        try:
            response = requests.post(
                'https://iot-api.oxelya.com/alerts',
                json=payload,
                headers={'Authorization': 'Bearer token'}
            )
            print('🚨 Alerte envoyée:', response.status_code)
        except Exception as e:
            print('❌ Erreur envoi:', e)
    
    def main_loop(self):
        self.connect_wifi()
        
        while True:
            # Lecture capteur
            sensor_value = self.sensor.read()
            temperature = (sensor_value / 4095.0) * 100
            
            # Prédiction IA embarquée
            is_anomaly = self.predict_anomaly([temperature])
            
            if is_anomaly:
                print(f'🚨 ANOMALIE: {temperature}°C')
                self.send_alert({
                    'temperature': temperature,
                    'anomaly': True
                })
                
                # LED rouge clignotante
                for _ in range(5):
                    self.led.on()
                    time.sleep(0.2)
                    self.led.off()
                    time.sleep(0.2)
            else:
                print(f'✅ Normal: {temperature}°C')
                self.led.on()
                time.sleep(0.1)
                self.led.off()
            
            time.sleep(30)  # Mesure toutes les 30s

# Démarrage du capteur intelligent
sensor = SmartSensor()
sensor.main_loop()