backend/server.py

import os
import socketio
import uvicorn
from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
import asyncio
import base64
import cv2
import numpy as np
import random
from deepface import DeepFace
from supabase import create_client, Client

# --- 1. CONFIGURATION SUPABASE ---
SUPABASE_URL = os.getenv("SUPABASE_URL", "https://gwjrwejdjpctizolfkcz.supabase.co")
SUPABASE_KEY = os.getenv("SUPABASE_KEY")

try:
    if SUPABASE_KEY:
        supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
        print("☁️ Connecté à Supabase")
    else:
        print("⚠️ ATTENTION: SUPABASE_KEY manquante !")
except Exception as e:
    print(f"❌ Erreur connexion Supabase : {e}")

# --- 2. CONFIGURATION SOCKET.IO (BYPASS TOTAL) ---
# Cette fonction force l'acceptation de tout le monde et affiche qui se connecte
def force_allow_origins(origin, environ, **kwargs):
    print(f"🔍 TENTATIVE CONNEXION DEPUIS : {origin}")
    return True # Retourne toujours VRAI (Autorisé)

sio = socketio.AsyncServer(
    async_mode='asgi',
    cors_allowed_origins=force_allow_origins, # On utilise la fonction au lieu de '*'
    logger=True,
    engineio_logger=True,
    always_connect=True
)

app = FastAPI()

# --- 3. MIDDLEWARE CORS API (REST) ---
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

socket_app = socketio.ASGIApp(sio, app)

# --- 4. API REST (ADMIN) ---
@app.get("/")
def home():
    return {"status": "online", "message": "Startech API is running"}

@app.get("/api/sessions")
def get_sessions():
    response = supabase.table('sessions').select("*").order('id', desc=True).execute()
    return response.data

@app.get("/api/sessions/{session_id}")
def get_session_details(session_id: int):
    sess = supabase.table('sessions').select("*").eq('id', session_id).execute()
    if not sess.data: raise HTTPException(status_code=404, detail="Session introuvable")
    meas = supabase.table('measurements').select("*").eq('session_id', session_id).order('session_time', desc=False).execute()
    return {"info": sess.data[0], "data": meas.data}

@app.delete("/api/sessions/{session_id}")
def delete_session(session_id: int):
    supabase.table('sessions').delete().eq('id', session_id).execute()
    return {"message": "Supprimé"}

# --- 5. LOGIQUE MÉTIER ---
camera_state = { "emotion": "neutral", "emotion_score": 0, "face_coords": None }
active_sessions = {}

def calculate_kpis(emotion):
    valence = 0.0; arousal = 0.0; noise = random.uniform(-0.05, 0.05)
    if emotion == "happy": valence = 0.8 + noise; arousal = 0.6 + noise
    elif emotion == "surprise": valence = 0.2 + noise; arousal = 0.9 + noise
    elif emotion in ["fear", "angry"]: valence = -0.7 + noise; arousal = 0.8 + noise
    elif emotion == "disgust": valence = -0.8 + noise; arousal = 0.5 + noise
    elif emotion == "sad": valence = -0.6 + noise; arousal = 0.2 + noise
    else: valence = 0.0 + noise; arousal = 0.3 + noise

    def clamp(n): return max(0, min(100, int(n)))
    val_eng = clamp((arousal * 100) + random.uniform(0, 5))
    val_sat = clamp(((valence + 1) / 2) * 100)
    val_tru = clamp(50 + (valence * 40) + random.uniform(0, 5)) if valence > 0 else clamp(50 - (abs(valence) * 40) + random.uniform(0, 5))
    val_loy = clamp((val_sat * 0.7) + (val_tru * 0.3))
    val_opi = val_sat

    lbl_eng = "Engagement Fort 🔥" if val_eng >= 75 else ("Engagement Moyen" if val_eng >= 40 else "Désengagement 💤")
    lbl_sat = "Très Satisfait 😃" if val_sat >= 70 else ("Neutre 😐" if val_sat >= 45 else "Insatisfait 😡")
    
    return { "engagement": val_eng, "satisfaction": val_sat, "trust": val_tru, "loyalty": val_loy, "opinion": val_opi, "lbl_eng": lbl_eng, "lbl_sat": lbl_sat }

# --- 6. SOCKET EVENTS ---
@sio.event
async def connect(sid, environ):
    print(f"✅ Client CONNECTÉ: {sid}")
    active_sessions[sid] = { "is_recording": False, "session_time": 0, "db_id": None }

@sio.event
async def disconnect(sid):
    if sid in active_sessions: del active_sessions[sid]

@sio.event
async def process_frame(sid, data_uri):
    try:
        encoded_data = data_uri.split(',')[1]
        nparr = np.frombuffer(base64.b64decode(encoded_data), np.uint8)
        frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
        result = DeepFace.analyze(frame, actions=['emotion'], enforce_detection=False, silent=True)
        data = result[0] if isinstance(result, list) else result
        camera_state["emotion"] = data['dominant_emotion']
        region = data['region']
        camera_state["face_coords"] = {'x': region['x'], 'y': region['y'], 'w': region['w'], 'h': region['h']} if region['w'] > 0 else None
    except:
        pass

@sio.event
async def start_session(sid, data):
    user = active_sessions.get(sid)
    if user:
        user["is_recording"] = True
        user["session_time"] = 0
        try:
            res = supabase.table('sessions').insert({ "first_name": data.get('firstName'), "last_name": data.get('lastName'), "client_id": data.get('clientId') }).execute()
            user["db_id"] = res.data[0]['id']
        except Exception as e: print(f"Erreur DB: {e}")

@sio.event
async def stop_session(sid):
    if sid in active_sessions: active_sessions[sid]["is_recording"] = False

async def loop():
    while True:
        for sid, user in list(active_sessions.items()):
            kpis = calculate_kpis(camera_state["emotion"])
            if user["is_recording"]:
                user["session_time"] += 1
                if user["db_id"]:
                    try:
                        supabase.table('measurements').insert({
                            "session_id": user["db_id"], "session_time": user["session_time"],
                            "emotion": camera_state["emotion"], 
                            "engagement_val": kpis["engagement"], "engagement_lbl": kpis["lbl_eng"],
                            "satisfaction_val": kpis["satisfaction"], "satisfaction_lbl": kpis["lbl_sat"],
                            "trust_val": kpis["trust"], "loyalty_val": kpis["loyalty"], "opinion_val": kpis["opinion"]
                        }).execute()
                    except: pass
            await sio.emit('metrics_update', { "emotion": camera_state["emotion"], "metrics": kpis, "face_coords": camera_state["face_coords"], "session_time": user["session_time"], "is_recording": user["is_recording"] }, room=sid)
        await asyncio.sleep(1)

if __name__ == "__main__":
    @app.on_event("startup")
    async def startup(): asyncio.create_task(loop())
    uvicorn.run(socket_app, host="0.0.0.0", port=7860)