Update app.py

app.py

@@ -8,47 +8,48 @@ import json
 
 app = FastAPI()
 
-# ⚠️ REMPLACE PAR LE CHEMIN DE TON DATASET
 DATASET_REPO_ID = "Javare/Local_AI_Leaderboard" 
 FILENAME = "scores.json"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
+# Nouveau modèle étendu selon tes exigences
 class ScoreEntry(BaseModel):
     config: str
     browser: str
     power: str
-    score: float
+    min_tps: float
+    max_tps: float
+    avg_tps: float
+    total_tokens: int
+    duration: float
 
 def get_scores():
     try:
-        # Télécharge le fichier de scores depuis le dataset Hugging Face
         path = hf_hub_download(repo_id=DATASET_REPO_ID, filename=FILENAME, repo_type="dataset", token=HF_TOKEN)
         with open(path, "r", encoding="utf-8") as f:
             return json.load(f)
     except Exception:
-        # Si le fichier n'existe pas encore (premier lancement), on démarre à vide
         return []
 
 @app.get("/api/scores")
 def read_scores():
     scores = get_scores()
-    scores.sort(key=lambda x: x["score"], reverse=True)
-    return scores[:10] # Renvoie uniquement le TOP 10
+    # On trie le TOP 10 par la vitesse moyenne (avg_tps)
+    scores.sort(key=lambda x: x.get("avg_tps", 0), reverse=True)
+    return scores[:10]
 
 @app.post("/api/score")
 def add_score(entry: ScoreEntry):
     if not HF_TOKEN:
-        raise HTTPException(status_code=500, detail="HF_TOKEN manquant dans les Secrets du Space")
+        raise HTTPException(status_code=500, detail="HF_TOKEN manquant")
     
     scores = get_scores()
     scores.append(entry.dict())
     
-    # Sauvegarde locale temporaire du JSON complet
     local_path = "scores.json"
     with open(local_path, "w", encoding="utf-8") as f:
         json.dump(scores, f, ensure_ascii=False, indent=2)
         
-    # Envoi sécurisé vers ton Dataset Hugging Face
     api = HfApi()
     try:
         api.upload_file(
@@ -59,11 +60,10 @@ def add_score(entry: ScoreEntry):
             token=HF_TOKEN
         )
     except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Erreur de synchronisation : {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
         
     return {"status": "success"}
 
-# Déclare l'accès aux fichiers statiques de ton interface
 @app.get("/")
 def read_index():
     return FileResponse("index.html")