add session clenaup and timeout messages

backend/routes/__init__.py

@@ -5,6 +5,7 @@ from .benchmark import router as benchmark_router, active_tasks
 from .questions import router as questions_router
 from .download import router as download_router
 from .evaluation import router as evaluation_router, active_evaluation_tasks
+from .cleanup import router as cleanup_router
 
 # Exposer les routeurs
 routers = [
@@ -13,11 +14,13 @@ routers = [
     benchmark_router,
     questions_router,
     download_router,
-    evaluation_router
+    evaluation_router,
+    cleanup_router
 ]
 
 # Référencer les données partagées entre routes
 benchmark_router.session_files = session_files
+cleanup_router.session_files = session_files
 
 # Exposer les variables partagées pour main.py
 __all__ = ['routers', 'session_files', 'active_tasks', 'active_evaluation_tasks'] 

backend/routes/cleanup.py

@@ -0,0 +1,58 @@
+from fastapi import APIRouter, HTTPException
+import os
+import shutil
+from .upload import session_files
+
+router = APIRouter(tags=["cleanup"])
+
+# Dossier racine pour les uploads
+UPLOAD_ROOT = "uploaded_files"
+
+# Liste des documents de base qui ne doivent pas être supprimés
+BASE_DOCUMENTS = ["the-bitter-lesson", "hurricane-faq", "pokemon-guide"]
+
+@router.delete("/cleanup-session/{session_id}")
+async def cleanup_session(session_id: str):
+    """
+    Supprime le dossier de session après que l'utilisateur a visualisé les résultats d'évaluation.
+    Ne supprime pas les documents de base.
+    
+    Args:
+        session_id: ID de la session à supprimer
+        
+    Returns:
+        Dictionary avec statut et message
+    """
+    # Vérifier si le session_id existe et n'est pas un document de base
+    if session_id in BASE_DOCUMENTS:
+        return {
+            "success": False,
+            "message": f"Cannot delete base document: {session_id}"
+        }
+    
+    session_dir = os.path.join(UPLOAD_ROOT, session_id)
+    
+    # Vérifier si le dossier existe
+    if not os.path.exists(session_dir):
+        return {
+            "success": False,
+            "message": f"Session directory not found: {session_id}"
+        }
+    
+    try:
+        # Supprimer la référence du fichier de session
+        if session_id in session_files:
+            del session_files[session_id]
+            
+        # Supprimer le dossier de session
+        shutil.rmtree(session_dir)
+        
+        return {
+            "success": True,
+            "message": f"Session cleaned up successfully: {session_id}"
+        }
+    except Exception as e:
+        return {
+            "success": False,
+            "message": f"Error cleaning up session: {str(e)}"
+        } 

frontend/server.js

@@ -42,6 +42,7 @@ app.use(
     "/evaluation-logs",
     "/evaluation-results",
     "/download-dataset",
+    "/cleanup-session",
   ],
   createProxyMiddleware({
     target: backendUrl,

frontend/src/components/BenchmarkEvaluation.jsx

@@ -282,8 +282,35 @@ const BenchmarkEvaluation = ({ sessionId, isDefaultDocument, onComplete }) => {
         alignItems: "center",
         justifyContent: "center",
         minHeight: 200,
+        position: "relative",
       }}
     >
+      {/* Temps estimé */}
+      <Box
+        sx={{
+          position: "absolute",
+          top: 12,
+          right: 12,
+          backgroundColor: "rgba(0, 0, 0, 0.04)",
+          borderRadius: "4px",
+          px: 1,
+          py: 0.5,
+          display: "inline-flex",
+          alignItems: "center",
+        }}
+      >
+        <Typography
+          variant="caption"
+          sx={{
+            fontSize: "0.675rem",
+            color: "text.secondary",
+            fontWeight: 500,
+          }}
+        >
+          Estimated time: ~1 min
+        </Typography>
+      </Box>
+
       {error ? (
         <Alert severity="error" sx={{ width: "100%" }}>
           {error}

frontend/src/components/BenchmarkGenerator.jsx

@@ -86,12 +86,33 @@ const BenchmarkGenerator = ({ sessionId, isDefaultDocument, onComplete }) => {
     // Set start time
     startTimeRef.current = Date.now();
 
+    // Référence pour le timeout
+    let timeoutRef = null;
+
     // Start timer
     timerIntervalRef.current = setInterval(() => {
       const timeElapsed = Math.floor(
         (Date.now() - startTimeRef.current) / 1000
       );
       setElapsedTime(timeElapsed);
+
+      // Vérifier si le temps écoulé dépasse 8 minutes (480 secondes) et que nous ne sommes pas en mode simulation
+      if (timeElapsed > 480 && !isDefault && !generationComplete) {
+        // Afficher un message d'erreur en cas de timeout
+        setError(
+          "The benchmark generation is taking too long. The demo is currently under heavy load, please try again later."
+        );
+        setGenerationComplete(true);
+
+        // Nettoyer les intervalles
+        if (pollingIntervalRef.current) {
+          clearInterval(pollingIntervalRef.current);
+        }
+
+        if (timerIntervalRef.current) {
+          clearInterval(timerIntervalRef.current);
+        }
+      }
     }, 1000);
 
     // Gestionnaire pour détecter quand la page redevient visible
@@ -483,8 +504,35 @@ const BenchmarkGenerator = ({ sessionId, isDefaultDocument, onComplete }) => {
         alignItems: "center",
         justifyContent: "center",
         minHeight: 200,
+        position: "relative",
       }}
     >
+      {/* Temps estimé */}
+      <Box
+        sx={{
+          position: "absolute",
+          top: 12,
+          right: 12,
+          backgroundColor: "rgba(0, 0, 0, 0.04)",
+          borderRadius: "4px",
+          px: 1,
+          py: 0.5,
+          display: "inline-flex",
+          alignItems: "center",
+        }}
+      >
+        <Typography
+          variant="caption"
+          sx={{
+            fontSize: "0.675rem",
+            color: "text.secondary",
+            fontWeight: 500,
+          }}
+        >
+          Estimated time: ~2 min
+        </Typography>
+      </Box>
+
       {error ? (
         <Alert severity="error" sx={{ width: "100%" }}>
           {error}

frontend/src/components/EvaluationDisplay.jsx

@@ -85,47 +85,7 @@ const getMedalStyle = (rank) => {
   };
 };
 
-const EvaluationDisplay = ({ sessionId }) => {
-  const [results, setResults] = useState(null);
-  const [loading, setLoading] = useState(true);
-  const [error, setError] = useState(null);
-
-  useEffect(() => {
-    const fetchEvaluationResults = async () => {
-      if (!sessionId) {
-        setError("No session ID provided");
-        setLoading(false);
-        return;
-      }
-
-      try {
-        // Fetch evaluation results from the API
-        const response = await fetch(
-          `http://localhost:3001/evaluation-results/${sessionId}`
-        );
-
-        if (!response.ok) {
-          throw new Error(`Failed to fetch results: ${response.status}`);
-        }
-
-        const data = await response.json();
-
-        if (!data.success) {
-          throw new Error(data.message || "Failed to fetch evaluation results");
-        }
-
-        setResults(data.results);
-      } catch (err) {
-        console.error("Error fetching evaluation results:", err);
-        setError(err.message);
-      } finally {
-        setLoading(false);
-      }
-    };
-
-    fetchEvaluationResults();
-  }, [sessionId]);
-
+const EvaluationDisplay = ({ sessionId, results }) => {
   // Format accuracy as percentage
   const formatAccuracy = (value) => {
     return `${(value * 100).toFixed(2)}\u2009%`;
@@ -152,34 +112,6 @@ const EvaluationDisplay = ({ sessionId }) => {
     return `${seconds.toFixed(2)}s`;
   };
 
-  if (loading) {
-    return (
-      <Box
-        sx={{
-          width: "100%",
-          mt: 4,
-          mb: 4,
-          display: "flex",
-          flexDirection: "column",
-          alignItems: "center",
-        }}
-      >
-        <Typography variant="h5" gutterBottom>
-          Loading Evaluation Results...
-        </Typography>
-        <CircularProgress />
-      </Box>
-    );
-  }
-
-  if (error) {
-    return (
-      <Alert severity="error" sx={{ mt: 4, mb: 4 }}>
-        {error}
-      </Alert>
-    );
-  }
-
   if (
     !results ||
     !results.models_comparison ||
@@ -187,7 +119,21 @@ const EvaluationDisplay = ({ sessionId }) => {
   ) {
     return (
       <Alert severity="info" sx={{ mt: 4, mb: 4 }}>
-        No evaluation results found for this benchmark.
+        No evaluation results found for this benchmark. The demo is currently
+        under heavy load, please try again later.
+      </Alert>
+    );
+  }
+
+  // Vérifier s'il n'y a aucun modèle réussi
+  const successfulModels = results.models_comparison.filter(
+    (model) => model.success
+  );
+  if (successfulModels.length === 0) {
+    return (
+      <Alert severity="warning" sx={{ mt: 4, mb: 4 }}>
+        All model evaluations failed. The demo is currently under heavy load,
+        please try again later.
       </Alert>
     );
   }
@@ -241,98 +187,96 @@ const EvaluationDisplay = ({ sessionId }) => {
             </TableRow>
           </TableHead>
           <TableBody>
-            {results.models_comparison
-              .filter((model) => model.success)
-              .map((model, index) => (
-                <TableRow
-                  key={`${model.model_name}-${model.provider}`}
+            {successfulModels.map((model, index) => (
+              <TableRow
+                key={`${model.model_name}-${model.provider}`}
+                sx={{
+                  "&:nth-of-type(even)": {
+                    backgroundColor: "rgba(0, 0, 0, 0.02)",
+                  },
+                }}
+              >
+                <TableCell>
+                  <Box sx={{ display: "flex", alignItems: "center" }}>
+                    <Box sx={getMedalStyle(index + 1)}>{index + 1}</Box>
+                  </Box>
+                </TableCell>
+                <TableCell component="th" scope="row">
+                  <Tooltip title={model.model_name} placement="top">
+                    <Link
+                      href={`https://huggingface.co/${model.model_name}`}
+                      target="_blank"
+                      rel="noopener noreferrer"
+                      sx={{
+                        textDecoration: "none",
+                        "&:hover": {
+                          textDecoration: "underline",
+                        },
+                        display: "flex",
+                        alignItems: "center",
+                      }}
+                    >
+                      {model.model_name.length > 40
+                        ? `${model.model_name.substring(0, 40)}...`
+                        : model.model_name}
+                      <OpenInNewIcon sx={{ ml: 0.5, fontSize: 16 }} />
+                    </Link>
+                  </Tooltip>
+                </TableCell>
+                <TableCell
+                  align="left"
                   sx={{
-                    "&:nth-of-type(even)": {
-                      backgroundColor: "rgba(0, 0, 0, 0.02)",
-                    },
+                    padding: 0,
+                    position: "relative",
+                    overflow: "hidden",
                   }}
                 >
-                  <TableCell>
-                    <Box sx={{ display: "flex", alignItems: "center" }}>
-                      <Box sx={getMedalStyle(index + 1)}>{index + 1}</Box>
-                    </Box>
-                  </TableCell>
-                  <TableCell component="th" scope="row">
-                    <Tooltip title={model.model_name} placement="top">
-                      <Link
-                        href={`https://huggingface.co/${model.model_name}`}
-                        target="_blank"
-                        rel="noopener noreferrer"
-                        sx={{
-                          textDecoration: "none",
-                          "&:hover": {
-                            textDecoration: "underline",
-                          },
-                          display: "flex",
-                          alignItems: "center",
-                        }}
-                      >
-                        {model.model_name.length > 40
-                          ? `${model.model_name.substring(0, 40)}...`
-                          : model.model_name}
-                        <OpenInNewIcon sx={{ ml: 0.5, fontSize: 16 }} />
-                      </Link>
-                    </Tooltip>
-                  </TableCell>
-                  <TableCell
-                    align="left"
+                  <Box
                     sx={{
-                      padding: 0,
-                      position: "relative",
-                      overflow: "hidden",
+                      position: "absolute",
+                      width: "100%",
+                      height: "100%",
+                      left: 0,
+                      top: 0,
+                      display: "flex",
+                      alignItems: "center",
+                      justifyContent: "flex-start",
+                      pl: 2,
                     }}
                   >
                     <Box
                       sx={{
                         position: "absolute",
-                        width: "100%",
-                        height: "100%",
                         left: 0,
                         top: 0,
-                        display: "flex",
-                        alignItems: "center",
-                        justifyContent: "flex-start",
-                        pl: 2,
+                        height: "100%",
+                        width: `${model.accuracy * 100}%`,
+                        backgroundColor: getColorForScore(model.accuracy),
+                        opacity: 0.2,
+                        zIndex: 0,
+                      }}
+                    />
+                    <Typography
+                      sx={{
+                        position: "relative",
+                        zIndex: 1,
+                        fontWeight: model.accuracy > 0.7 ? "bold" : "normal",
+                        py: 1.5,
+                        textAlign: "left",
                       }}
                     >
-                      <Box
-                        sx={{
-                          position: "absolute",
-                          left: 0,
-                          top: 0,
-                          height: "100%",
-                          width: `${model.accuracy * 100}%`,
-                          backgroundColor: getColorForScore(model.accuracy),
-                          opacity: 0.2,
-                          zIndex: 0,
-                        }}
-                      />
-                      <Typography
-                        sx={{
-                          position: "relative",
-                          zIndex: 1,
-                          fontWeight: model.accuracy > 0.7 ? "bold" : "normal",
-                          py: 1.5,
-                          textAlign: "left",
-                        }}
-                      >
-                        {formatAccuracy(model.accuracy)}
-                      </Typography>
-                    </Box>
-                  </TableCell>
-                  <TableCell align="left">
-                    {formatTime(model.evaluation_time)}
-                  </TableCell>
-                  <TableCell align="right">
-                    <span style={{ color: "green" }}>✓ Success</span>
-                  </TableCell>
-                </TableRow>
-              ))}
+                      {formatAccuracy(model.accuracy)}
+                    </Typography>
+                  </Box>
+                </TableCell>
+                <TableCell align="left">
+                  {formatTime(model.evaluation_time)}
+                </TableCell>
+                <TableCell align="right">
+                  <span style={{ color: "green" }}>✓ Success</span>
+                </TableCell>
+              </TableRow>
+            ))}
           </TableBody>
         </Table>
       </TableContainer>

frontend/src/pages/EvaluationDisplayPage.jsx

@@ -1,19 +1,26 @@
 import React, { useState, useEffect } from "react";
-import { Box, CircularProgress } from "@mui/material";
+import { Box, CircularProgress, Alert } from "@mui/material";
 import { useSearchParams, Navigate } from "react-router-dom";
 import Intro from "../components/Intro";
 import EvaluationDisplay from "../components/EvaluationDisplay";
 import { useThemeMode } from "../hooks/useThemeMode";
 import getTheme from "../config/theme";
+import API_CONFIG from "../config/api";
 
 function EvaluationDisplayPage() {
   const [searchParams] = useSearchParams();
   const sessionId = searchParams.get("session");
   const [isValidSession, setIsValidSession] = useState(true);
   const [isLoading, setIsLoading] = useState(true);
+  const [evaluationResults, setEvaluationResults] = useState(null);
+  const [error, setError] = useState(null);
   const { mode } = useThemeMode();
   const theme = getTheme(mode);
 
+  // Liste des documents de base qui ne doivent pas être supprimés
+  const baseDocuments = ["the-bitter-lesson", "hurricane-faq", "pokemon-guide"];
+  const isBaseDocument = baseDocuments.includes(sessionId);
+
   useEffect(() => {
     if (!sessionId) {
       console.log(
@@ -23,29 +30,88 @@ function EvaluationDisplayPage() {
       return;
     }
 
-    const checkSession = async () => {
+    const fetchEvaluationResults = async () => {
       try {
-        const response = await fetch(
-          `http://localhost:3001/benchmark-questions/${sessionId}`
+        // Vérifier d'abord si la session existe
+        const sessionCheckResponse = await fetch(
+          `${API_CONFIG.BASE_URL}/benchmark-questions/${sessionId}`
         );
 
-        if (!response.ok) {
+        if (!sessionCheckResponse.ok) {
           console.error(
-            `Session invalide ou erreur serveur: ${response.status}`
+            `Session invalide ou erreur serveur: ${sessionCheckResponse.status}`
           );
           setIsValidSession(false);
+          return;
         }
+
+        // Récupérer les résultats d'évaluation
+        const evalResponse = await fetch(
+          `${API_CONFIG.BASE_URL}/evaluation-results/${sessionId}`
+        );
+
+        if (!evalResponse.ok) {
+          setError(`Failed to fetch results: ${evalResponse.status}`);
+          setIsLoading(false);
+          return;
+        }
+
+        const data = await evalResponse.json();
+
+        if (!data.success) {
+          setError(data.message || "Failed to fetch evaluation results");
+          setIsLoading(false);
+          return;
+        }
+
+        setEvaluationResults(data.results);
       } catch (error) {
-        console.error("Erreur lors de la vérification de la session:", error);
-        setIsValidSession(false);
+        console.error("Error fetching evaluation results:", error);
+        setError(error.message);
       } finally {
         setIsLoading(false);
       }
     };
 
-    checkSession();
+    fetchEvaluationResults();
   }, [sessionId]);
 
+  // Effet pour nettoyer le dossier de session après avoir affiché les résultats
+  useEffect(() => {
+    // Ne pas nettoyer si c'est un document de base ou si les résultats ne sont pas encore chargés
+    if (isBaseDocument || isLoading || !evaluationResults) {
+      return;
+    }
+
+    // Fonction pour supprimer le dossier de session
+    const cleanupSession = async () => {
+      try {
+        const response = await fetch(
+          `${API_CONFIG.BASE_URL}/cleanup-session/${sessionId}`,
+          {
+            method: "DELETE",
+          }
+        );
+
+        if (response.ok) {
+          console.log(`Session ${sessionId} cleaned up successfully`);
+        } else {
+          console.warn(`Failed to clean up session ${sessionId}`);
+        }
+      } catch (error) {
+        console.error("Error cleaning up session:", error);
+      }
+    };
+
+    // Appeler la fonction après un délai pour s'assurer que l'utilisateur a eu le temps de voir les résultats
+    const cleanupTimeout = setTimeout(() => {
+      cleanupSession();
+    }, 2000);
+
+    // Nettoyer le timeout si le composant est démonté
+    return () => clearTimeout(cleanupTimeout);
+  }, [sessionId, isBaseDocument, isLoading, evaluationResults]);
+
   if (!isValidSession) {
     return <Navigate to="/" />;
   }
@@ -65,6 +131,10 @@ function EvaluationDisplayPage() {
         >
           <CircularProgress size={60} />
         </Box>
+      ) : error ? (
+        <Alert severity="error" sx={{ mt: 4, mb: 4 }}>
+          {error}
+        </Alert>
       ) : (
         <Box
           sx={{
@@ -74,7 +144,10 @@ function EvaluationDisplayPage() {
             bgcolor: "background.paper",
           }}
         >
-          <EvaluationDisplay sessionId={sessionId} />
+          <EvaluationDisplay
+            sessionId={sessionId}
+            results={evaluationResults}
+          />
         </Box>
       )}
     </>