Create server.py

backend/server.py

@@ -0,0 +1,159 @@
+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") # On récupère la clé secrète des variables d'environnement
+
+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 (CORS TOTAL) ---
+# On configure le serveur pour tout accepter, sans restriction.
+mgr = socketio.AsyncRedisManager(os.environ.get('REDIS_URL', '')) if os.environ.get('REDIS_URL') else None
+sio = socketio.AsyncServer(
+    async_mode='asgi',
+    cors_allowed_origins='*',  # L'étoile magique
+    logger=True,               # Logs activés pour voir les connexions
+    engineio_logger=True,
+    always_connect=True        # Force la connexion même si le transport est bizarre
+)
+
+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)