add url importer | improve yourbench error handling | refactor

backend/pyproject.toml

@@ -26,6 +26,7 @@ dependencies = [
     "PyPDF2>=3.0.0",
     "beautifulsoup4>=4.12.0",
     "evaluate>=0.4.0",
+    "requests>=2.31.0",
 ]
 
 [build-system]

backend/routes/cleanup.py

@@ -26,8 +26,8 @@ async def cleanup_session(session_id: str):
         Dictionary with status and message
     """
     # Check if we are in development mode
-    # if os.environ.get("ENVIRONEMENT", "").lower() == "development":
-    if True:
+    if os.environ.get("ENVIRONEMENT", "").lower() == "development":
+    # if True:
         logging.info(f"[DEV MODE] Cleanup called for session: {session_id} - No action taken in development mode")
         return {
             "success": True,

backend/routes/download.py

@@ -1,5 +1,5 @@
 from fastapi import APIRouter, HTTPException
-from fastapi.responses import StreamingResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 from huggingface_hub import hf_hub_download, snapshot_download
 import os
 import tempfile
@@ -7,6 +7,8 @@ import shutil
 import zipfile
 import io
 import logging
+import json
+from datasets import load_dataset
 
 router = APIRouter(tags=["download"])
 
@@ -71,4 +73,85 @@ async def download_dataset(session_id: str):
         raise HTTPException(
             status_code=500, 
             detail=f"Erreur lors du téléchargement: {str(e)}"
+        )
+
+@router.get("/download-questions/{session_id}")
+async def download_questions(session_id: str):
+    """
+    Télécharge les questions générées pour une session au format JSON
+    
+    Args:
+        session_id: Identifiant de la session
+        
+    Returns:
+        Fichier JSON contenant les questions générées
+    """
+    try:
+        # Identifiant du repo HuggingFace
+        dataset_repo_id = f"yourbench/yourbench_{session_id}"
+        
+        # Initialize questions list
+        all_questions = []
+        
+        # Try to load single-shot questions
+        try:
+            single_dataset = load_dataset(dataset_repo_id, 'single_shot_questions')
+            if single_dataset and len(single_dataset['train']) > 0:
+                for idx in range(len(single_dataset['train'])):
+                    all_questions.append({
+                        "id": str(idx),
+                        "question": single_dataset['train'][idx].get("question", ""),
+                        "answer": single_dataset['train'][idx].get("self_answer", "No answer available"),
+                        "type": "single_shot"
+                    })
+                logging.info(f"Loaded {len(all_questions)} single-shot questions")
+        except Exception as e:
+            logging.error(f"Error loading single-shot questions: {str(e)}")
+        
+        # Try to load multi-hop questions
+        try:
+            multi_dataset = load_dataset(dataset_repo_id, 'multi_hop_questions')
+            if multi_dataset and len(multi_dataset['train']) > 0:
+                start_idx = len(all_questions)
+                for idx in range(len(multi_dataset['train'])):
+                    all_questions.append({
+                        "id": str(start_idx + idx),
+                        "question": multi_dataset['train'][idx].get("question", ""),
+                        "answer": multi_dataset['train'][idx].get("self_answer", "No answer available"),
+                        "type": "multi_hop"
+                    })
+                logging.info(f"Loaded {len(multi_dataset['train'])} multi-hop questions")
+        except Exception as e:
+            logging.error(f"Error loading multi-hop questions: {str(e)}")
+        
+        # If we couldn't load any questions, the dataset might not exist
+        if len(all_questions) == 0:
+            raise HTTPException(status_code=404, detail="Aucune question trouvée pour cette session")
+        
+        # Convert questions to JSON
+        questions_json = json.dumps({
+            "session_id": session_id,
+            "questions": all_questions
+        }, ensure_ascii=False, indent=2)
+        
+        # Create a BytesIO object with the JSON data
+        json_bytes = io.BytesIO(questions_json.encode('utf-8'))
+        json_bytes.seek(0)
+        
+        # Return the JSON file for download
+        filename = f"yourbench_{session_id}_questions.json"
+        return StreamingResponse(
+            json_bytes,
+            media_type="application/json",
+            headers={"Content-Disposition": f"attachment; filename={filename}"}
+        )
+            
+    except HTTPException:
+        # Re-raise HTTP exceptions
+        raise
+    except Exception as e:
+        logging.error(f"Erreur lors de la récupération des questions: {str(e)}")
+        raise HTTPException(
+            status_code=500, 
+            detail=f"Erreur lors du téléchargement des questions: {str(e)}"
         ) 

backend/routes/questions.py

@@ -50,26 +50,26 @@ async def get_benchmark_questions(session_id: str):
         except Exception as e:
             print(f"Error loading single-shot questions: {str(e)}")
         
-        try:
-            # Essayer de charger les questions multi-hop si nécessaire
-            if len(questions) < 2:
-                multi_dataset = load_dataset(dataset_repo_id, 'multi_hop_questions')
-                if multi_dataset and len(multi_dataset['train']) > 0:
-                    # Prendre les questions multi-hop pour compléter, en évitant aussi la première
-                    start_idx = 1
-                    remaining = 2 - len(questions)
-                    max_questions = min(remaining, max(0, len(multi_dataset['train']) - start_idx))
-                    for i in range(max_questions):
-                        idx = start_idx + i
-                        questions.append({
-                            "id": str(idx),
-                            "question": multi_dataset['train'][idx].get("question", ""),
-                            "answer": multi_dataset['train'][idx].get("self_answer", "No answer available"),
-                            "type": "multi_hop"
-                        })
-                    print(f"Loaded {len(questions)} multi-hop questions")
-        except Exception as e:
-            print(f"Error loading multi-hop questions: {str(e)}")
+        # try:
+        #     # Essayer de charger les questions multi-hop si nécessaire
+        #     if len(questions) < 2:
+        #         multi_dataset = load_dataset(dataset_repo_id, 'multi_hop_questions')
+        #         if multi_dataset and len(multi_dataset['train']) > 0:
+        #             # Prendre les questions multi-hop pour compléter, en évitant aussi la première
+        #             start_idx = 1
+        #             remaining = 2 - len(questions)
+        #             max_questions = min(remaining, max(0, len(multi_dataset['train']) - start_idx))
+        #             for i in range(max_questions):
+        #                 idx = start_idx + i
+        #                 questions.append({
+        #                     "id": str(idx),
+        #                     "question": multi_dataset['train'][idx].get("question", ""),
+        #                     "answer": multi_dataset['train'][idx].get("self_answer", "No answer available"),
+        #                     "type": "multi_hop"
+        #                 })
+        #             print(f"Loaded {len(questions)} multi-hop questions")
+        # except Exception as e:
+        #     print(f"Error loading multi-hop questions: {str(e)}")
         
         # If we couldn't load any questions, the dataset might not exist
         if len(questions) == 0:

backend/routes/upload.py

@@ -4,28 +4,31 @@ import shutil
 import uuid
 from bs4 import BeautifulSoup
 from PyPDF2 import PdfReader
+import requests
+from fastapi import Form
+from typing import Optional
 
 router = APIRouter(tags=["files"])
 
-# Définir le stockage des fichiers par session (importé dans main.py)
+# Define file storage by session (imported in main.py)
 session_files = {}
 
-# Dossier racine pour les uploads
+# Root folder for uploads
 UPLOAD_ROOT = "uploaded_files"
 os.makedirs(UPLOAD_ROOT, exist_ok=True)
 
-# Longueur minimale pour tout fichier (en caractères)
+# Minimum length for any file (in characters)
 MIN_FILE_LENGTH = 500
 
 def validate_pdf(file_path: str) -> bool:
     """Validate if file is a valid PDF."""
     try:
         reader = PdfReader(file_path)
-        # Vérifier que le PDF a au moins une page
+        # Check that the PDF has at least one page
         if len(reader.pages) == 0:
             return False
             
-        # Extraire le texte pour vérifier la longueur
+        # Extract text to check length
         text = ""
         for page in reader.pages:
             text += page.extract_text()
@@ -39,7 +42,7 @@ def validate_markdown(file_path: str) -> bool:
     try:
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
-            # Vérifier longueur minimale et présence d'éléments markdown
+            # Check minimum length and presence of markdown elements
             return len(content) >= MIN_FILE_LENGTH and any(marker in content for marker in ['#', '-', '*', '`', '[', '>'])
     except:
         return False
@@ -49,7 +52,7 @@ def validate_html(file_path: str) -> bool:
     try:
         with open(file_path, 'r', encoding='utf-8') as f:
             content = f.read()
-            # Vérifier longueur minimale et structure HTML
+            # Check minimum length and HTML structure
             if len(content) < MIN_FILE_LENGTH:
                 return False
             BeautifulSoup(content, 'html.parser')
@@ -100,7 +103,7 @@ async def upload_file(file: UploadFile = File(...)):
     Returns:
         Dictionary with filename, status and session_id
     """
-    # Vérifier si le fichier est un PDF, TXT, HTML ou MD
+    # Check if the file is a PDF, TXT, HTML or MD
     if not file.filename.endswith(('.pdf', '.txt', '.html', '.md')):
         raise HTTPException(status_code=400, detail="Only PDF, TXT, HTML and MD files are accepted")
     
@@ -121,11 +124,11 @@ async def upload_file(file: UploadFile = File(...)):
     # Create the full path to save the file
     file_path = os.path.join(uploaded_files_dir, standardized_filename)
     
-    # Sauvegarder le fichier
+    # Save the file
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
     
-    # Valider le fichier selon son type
+    # Validate the file according to its type
     is_valid = False
     error_detail = ""
     
@@ -194,11 +197,83 @@ async def upload_file(file: UploadFile = File(...)):
             is_valid = False
     
     if not is_valid:
-        # Supprimer le fichier invalide
+        # Delete the invalid file
         os.remove(file_path)
         raise HTTPException(status_code=400, detail=error_detail or f"Invalid {file_extension[1:].upper()} file")
     
     # Store file path for later use
     session_files[session_id] = file_path
     
-    return {"filename": standardized_filename, "status": "uploaded", "session_id": session_id} 
+    return {"filename": standardized_filename, "status": "uploaded", "session_id": session_id}
+
+@router.post("/upload-url")
+async def upload_url(url: str = Form(...)):
+    """
+    Upload content from a URL, extract text and store it as a document
+    
+    Args:
+        url: The URL to download content from
+        
+    Returns:
+        Dictionary with status and session_id
+    """
+    try:
+        # Retrieve the content from the URL
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()  # Raise an exception if the HTTP status is not 200
+        
+        # Extract text from HTML with BeautifulSoup
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        # Remove script and style tags
+        for script in soup(["script", "style"]):
+            script.extract()
+            
+        # Extract the text
+        text = soup.get_text()
+        
+        # Clean the text (remove multiple spaces and empty lines)
+        lines = (line.strip() for line in text.splitlines())
+        chunks = (phrase.strip() for line in lines for phrase in line.split("  "))
+        text = '\n'.join(chunk for chunk in chunks if chunk)
+        
+        # Limit to 1000 characters if necessary
+        if len(text) > 25000:
+            text = text[:25000]
+            
+        # Check if the text is long enough
+        if len(text.strip()) < MIN_FILE_LENGTH:
+            raise HTTPException(
+                status_code=400, 
+                detail=f"The content is too short ({len(text.strip())} characters). Minimum required: {MIN_FILE_LENGTH} characters."
+            )
+        
+        # Generate a session ID
+        session_id = str(uuid.uuid4())
+        # Create the directory structure for the session
+        session_dir = os.path.join(UPLOAD_ROOT, session_id)
+        uploaded_files_dir = os.path.join(session_dir, "uploaded_files")
+        os.makedirs(uploaded_files_dir, exist_ok=True)
+        
+        # Path of the file to save
+        file_path = os.path.join(uploaded_files_dir, "document.txt")
+        
+        # Save the text
+        with open(file_path, "w", encoding="utf-8") as f:
+            f.write(text)
+            
+        # Store the file path for later use
+        session_files[session_id] = file_path
+        
+        return {
+            "status": "uploaded", 
+            "session_id": session_id,
+            "filename": "document.txt",
+            "text_length": len(text),
+            "source_url": url
+        }
+    
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=400, detail=f"Error retrieving the URL: {str(e)}")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error processing the URL: {str(e)}") 

backend/tasks/create_bench.py

@@ -140,6 +140,9 @@ class CreateBenchTask:
                         self._add_log(f"[ERROR] {line}")
                     elif "WARNING" in line:
                         self._add_log(f"[WARN] {line}")
+                        # Detect specific warning about no valid questions
+                        if "No valid questions produced in single_shot_question_generation" in line:
+                            self._add_log("[ERROR] Failed to generate benchmark: The document does not contain enough information to generate a meaningful benchmark. Please try with a more detailed document.")
                     else:
                         # Detect completed stages
                         if "Completed stage:" in line:

backend/tasks/get_available_model_provider.py

@@ -68,7 +68,7 @@ def test_provider(model_name: str, provider: str, verbose: bool = False) -> bool
         except Exception as e:
             if verbose:
                 error_message = str(e)
-                logger.error(f"Error with provider {provider}: {error_message}")
+                logger.warning(f"Error with provider {provider}: {error_message}")
                 
                 # Log specific error types if we can identify them
                 if "status_code=429" in error_message:
@@ -78,12 +78,12 @@ def test_provider(model_name: str, provider: str, verbose: bool = False) -> bool
                 elif "status_code=503" in error_message:
                     logger.warning(f"Provider {provider} service unavailable. Model may be loading or provider is down.")
                 elif "timed out" in error_message.lower():
-                    logger.error(f"Timeout error with provider {provider} - request timed out after 10 seconds")
+                    logger.warning(f"Timeout error with provider {provider} - request timed out after 10 seconds")
             return False
             
     except Exception as e:
         if verbose:
-            logger.error(f"Error in test_provider: {str(e)}")
+            logger.warning(f"Error in test_provider: {str(e)}")
         return False
 
 def get_available_model_provider(model_name, verbose=False):
@@ -127,6 +127,9 @@ def get_available_model_provider(model_name, verbose=False):
             if test_provider(model_name, provider, verbose):
                 return provider
                 
+        # If we've tried all providers and none worked, log this but don't raise an exception
+        if verbose:
+            logger.error(f"No available providers for {model_name}")
         return None
         
     except Exception as e:
@@ -147,12 +150,19 @@ if __name__ == "__main__":
     ]
 
     providers = []
+    unavailable_models = []
 
     for model in models:
         provider = get_available_model_provider(model, verbose=True)
-        providers.append((model, provider))
+        if provider:
+            providers.append((model, provider))
+        else:
+            unavailable_models.append(model)
 
     for model, provider in providers:
         print(f"Model: {model}, Provider: {provider}")
-
+        
+    if unavailable_models:
+        print(f"Models with no available providers: {', '.join(unavailable_models)}")
+    
     print(f"Total Providers {len(providers)}: {providers}")

frontend/src/components/Benchmark/CreateForm.jsx

@@ -1,4 +1,4 @@
-import React, { useState, useRef } from "react";
+import React, { useState, useEffect } from "react";
 import {
   Box,
   Paper,
@@ -14,6 +14,8 @@ import {
   DialogTitle,
   DialogContent,
   DialogActions,
+  TextField,
+  Divider,
 } from "@mui/material";
 import { alpha } from "@mui/material/styles";
 import CloudUploadIcon from "@mui/icons-material/CloudUpload";
@@ -25,9 +27,11 @@ import MenuBookIcon from "@mui/icons-material/MenuBook";
 import DownloadIcon from "@mui/icons-material/Download";
 import VisibilityIcon from "@mui/icons-material/Visibility";
 import CloseIcon from "@mui/icons-material/Close";
+import LinkIcon from "@mui/icons-material/Link";
 import { useThemeMode } from "../../hooks/useThemeMode";
 import getTheme from "../../config/theme";
 import API_CONFIG from "../../config/api";
+import { useDocumentSelection } from "./hooks/useDocumentSelection";
 
 /**
  * Component for creating a new benchmark, including file upload and generation initiation
@@ -39,20 +43,45 @@ import API_CONFIG from "../../config/api";
 function CreateForm({ onStartGeneration }) {
   const { mode } = useThemeMode();
   const theme = getTheme(mode);
-  const [isDragging, setIsDragging] = useState(false);
-  const [uploadStatus, setUploadStatus] = useState(null);
-  const [isLoading, setIsLoading] = useState(false);
-  const [sessionId, setSessionId] = useState(null);
+
+  // États pour la visualisation du document
   const [openSnackbar, setOpenSnackbar] = useState(false);
-  const [selectedDocument, setSelectedDocument] = useState(null);
-  const [isDefaultDocument, setIsDefaultDocument] = useState(false);
   const [isDownloading, setIsDownloading] = useState(false);
   const [documentContent, setDocumentContent] = useState("");
   const [openContentModal, setOpenContentModal] = useState(false);
   const [isLoadingContent, setIsLoadingContent] = useState(false);
   const [modalDocument, setModalDocument] = useState(null);
-  const fileInputRef = useRef(null);
 
+  // Utiliser le hook personnalisé pour la gestion des documents
+  const {
+    isDragging,
+    isLoading,
+    sessionId,
+    selectedDocument,
+    isDefaultDocument,
+    urlInput,
+    urlSelected,
+    uploadStatus,
+    fileInputRef,
+    handleDragOver,
+    handleDragLeave,
+    handleClick,
+    handleFileChange,
+    handleDrop,
+    handleDefaultDocClick,
+    handleGenerateClick,
+    handleUrlInputChange,
+    setUploadStatus,
+  } = useDocumentSelection(onStartGeneration);
+
+  // Afficher le snackbar quand uploadStatus change
+  useEffect(() => {
+    if (uploadStatus) {
+      setOpenSnackbar(true);
+    }
+  }, [uploadStatus]);
+
+  // Liste des documents par défaut
   const defaultDocuments = [
     {
       id: "the-bitter-lesson",
@@ -78,143 +107,45 @@ function CreateForm({ onStartGeneration }) {
     setOpenSnackbar(false);
   };
 
-  const handleDragOver = (e) => {
-    e.preventDefault();
-    setIsDragging(true);
-  };
-
-  const handleDragLeave = () => {
-    setIsDragging(false);
-  };
-
-  const handleClick = () => {
-    fileInputRef.current.click();
-  };
-
-  const handleFileChange = (e) => {
-    const file = e.target.files[0];
-    if (!file) return;
-
-    // Check if it's a PDF, TXT, HTML or MD
-    if (
-      !file.name.endsWith(".pdf") &&
-      !file.name.endsWith(".txt") &&
-      !file.name.endsWith(".html") &&
-      !file.name.endsWith(".md")
-    ) {
-      setUploadStatus({
-        success: false,
-        message: "Only PDF, TXT, HTML and MD files are accepted",
-      });
-      setOpenSnackbar(true);
-      return;
-    }
-
-    // Check file size limit (3MB = 3145728 bytes)
-    if (file.size > 1048576 * 2) {
-      setUploadStatus({
-        success: false,
-        message: "File size exceeds the 2MB limit",
-      });
-      setOpenSnackbar(true);
-      return;
-    }
-
-    handleFileUpload(file);
-  };
-
-  const handleFileUpload = async (file) => {
-    setIsLoading(true);
-    setUploadStatus(null);
-    setIsDefaultDocument(false);
-    setSelectedDocument(null);
+  const handleViewDocument = async (doc) => {
+    setIsLoadingContent(true);
 
     try {
-      const formData = new FormData();
-      formData.append("file", file);
-
-      const response = await fetch(`${API_CONFIG.BASE_URL}/upload`, {
-        method: "POST",
-        body: formData,
-      });
-
-      const result = await response.json();
-
-      if (response.ok) {
-        setUploadStatus({
-          success: true,
-          message: "File uploaded successfully",
-        });
-        setOpenSnackbar(true);
-        setSessionId(result.session_id);
-        setSelectedDocument({ name: file.name });
+      let extension = "";
+      if (doc.id === "the-bitter-lesson") {
+        extension = "html";
+      } else if (doc.id === "hurricane-faq") {
+        extension = "md";
       } else {
-        setUploadStatus({
-          success: false,
-          message: result.detail || "Upload failed",
-        });
-        setOpenSnackbar(true);
+        extension = "txt";
       }
-    } catch (error) {
-      setUploadStatus({
-        success: false,
-        message: "Server connection error",
-      });
-      setOpenSnackbar(true);
-    } finally {
-      setIsLoading(false);
-    }
-  };
 
-  const handleDrop = async (e) => {
-    e.preventDefault();
-    setIsDragging(false);
-
-    const file = e.dataTransfer.files[0];
-    if (!file) {
-      setUploadStatus({ success: false, message: "No file detected" });
-      setOpenSnackbar(true);
-      return;
-    }
+      // Mettre à jour l'état du document pour la modale
+      setModalDocument(doc);
 
-    // Check if it's a PDF, TXT, HTML or MD
-    if (
-      !file.name.endsWith(".pdf") &&
-      !file.name.endsWith(".txt") &&
-      !file.name.endsWith(".html") &&
-      !file.name.endsWith(".md")
-    ) {
-      setUploadStatus({
-        success: false,
-        message: "Only PDF, TXT, HTML and MD files are accepted",
-      });
-      setOpenSnackbar(true);
-      return;
-    }
+      const response = await fetch(`/${doc.id}.${extension}`);
+      const text = await response.text();
 
-    // Check file size limit (3MB = 3145728 bytes)
-    if (file.size > 1048576 * 3) {
+      setDocumentContent(text);
+      setOpenContentModal(true);
+    } catch (error) {
+      console.error("Error loading document content:", error);
       setUploadStatus({
         success: false,
-        message: "File size exceeds the 3MB limit",
+        message: "Error loading document content",
       });
-      setOpenSnackbar(true);
-      return;
+    } finally {
+      setIsLoadingContent(false);
     }
-
-    handleFileUpload(file);
-  };
-
-  const handleDefaultDocClick = (doc) => {
-    setSelectedDocument(doc);
-    setSessionId(doc.id);
-    setIsDefaultDocument(true);
   };
 
-  const handleGenerateClick = () => {
-    if (onStartGeneration && sessionId) {
-      onStartGeneration(sessionId, isDefaultDocument);
-    }
+  const handleCloseContentModal = () => {
+    setOpenContentModal(false);
+    // Réinitialiser après la fermeture de la modale
+    setTimeout(() => {
+      setDocumentContent("");
+      setModalDocument(null);
+    }, 300);
   };
 
   const handleDownloadDocument = async (doc) => {
@@ -247,54 +178,11 @@ function CreateForm({ onStartGeneration }) {
         success: false,
         message: "Error downloading document",
       });
-      setOpenSnackbar(true);
     } finally {
       setIsDownloading(false);
     }
   };
 
-  const handleViewDocument = async (doc) => {
-    setIsLoadingContent(true);
-
-    try {
-      let extension = "";
-      if (doc.id === "the-bitter-lesson") {
-        extension = "html";
-      } else if (doc.id === "hurricane-faq") {
-        extension = "md";
-      } else {
-        extension = "txt";
-      }
-
-      // Mettre à jour l'état du document pour la modale
-      setModalDocument(doc);
-
-      const response = await fetch(`/${doc.id}.${extension}`);
-      const text = await response.text();
-
-      setDocumentContent(text);
-      setOpenContentModal(true);
-    } catch (error) {
-      console.error("Error loading document content:", error);
-      setUploadStatus({
-        success: false,
-        message: "Error loading document content",
-      });
-      setOpenSnackbar(true);
-    } finally {
-      setIsLoadingContent(false);
-    }
-  };
-
-  const handleCloseContentModal = () => {
-    setOpenContentModal(false);
-    // Réinitialiser après la fermeture de la modale
-    setTimeout(() => {
-      setDocumentContent("");
-      setModalDocument(null);
-    }, 300);
-  };
-
   return (
     <Box sx={{ mt: -2 }}>
       <Typography
@@ -303,10 +191,11 @@ function CreateForm({ onStartGeneration }) {
         align="center"
         sx={{ mb: 2, color: "text.secondary" }}
       >
-        To create a benchmark, choose a sample document
+        To create a benchmark, choose a sample document or upload your own
+        file/URL
       </Typography>
 
-      <Grid container spacing={2} sx={{ mb: 2 }}>
+      <Grid container spacing={2} sx={{ mb: 0 }}>
         {defaultDocuments.map((doc) => (
           <Grid item xs={12} sm={4} key={doc.id}>
             <Box
@@ -379,81 +268,149 @@ function CreateForm({ onStartGeneration }) {
         ))}
       </Grid>
 
-      <Typography
-        variant="subtitle1"
-        component="div"
-        align="center"
-        sx={{ mb: 2, color: "text.secondary" }}
-      >
-        Or upload your own ...
-      </Typography>
-
       <Box
         sx={{
-          p: 4,
-          mt: 2,
-          mb: 2,
-          borderRadius: 1.5,
-          border:
-            selectedDocument?.name && !isDefaultDocument
-              ? `2px solid ${theme.palette.primary.main}`
-              : isDragging
-              ? `2px dashed ${theme.palette.primary.main}`
-              : `2px dashed ${theme.palette.divider}`,
-          backgroundColor: isDragging
-            ? alpha(theme.palette.action.hover, 0.5)
-            : "transparent",
           display: "flex",
-          flexDirection: "column",
-          alignItems: "center",
-          justifyContent: "center",
-          minHeight: 180,
-          cursor: "pointer",
-          transition: "all 0.3s ease",
+          flexDirection: { xs: "column", md: "row" },
+          gap: 2,
+          mb: 2,
         }}
-        onDragOver={handleDragOver}
-        onDragLeave={handleDragLeave}
-        onDrop={handleDrop}
-        onClick={handleClick}
       >
-        <input
-          type="file"
-          ref={fileInputRef}
-          onChange={handleFileChange}
-          accept=".pdf,.txt,.html,.md"
-          style={{ display: "none" }}
-        />
-        {selectedDocument?.name && !isDefaultDocument ? (
-          <>
-            <InsertDriveFileIcon
-              sx={{ fontSize: 50, color: "primary.main", mb: 1 }}
-            />
-            <Typography variant="h6" component="div" gutterBottom>
-              {selectedDocument.name}
-            </Typography>
-            <Typography variant="body2" color="text.secondary">
-              Click to upload a different file
-            </Typography>
-          </>
-        ) : (
-          <>
-            {isLoading ? (
-              <CircularProgress size={50} sx={{ mb: 1 }} />
-            ) : (
-              <CloudUploadIcon
+        {/* Zone de glisser-déposer pour les fichiers */}
+        <Box
+          sx={{
+            flex: 1,
+            p: 4,
+            mt: 2,
+            borderRadius: 1.5,
+            border:
+              selectedDocument?.name && !isDefaultDocument && !urlSelected
+                ? `2px solid ${theme.palette.primary.main}`
+                : isDragging
+                ? `2px dashed ${theme.palette.primary.main}`
+                : `2px dashed ${theme.palette.divider}`,
+            backgroundColor: isDragging
+              ? alpha(theme.palette.action.hover, 0.5)
+              : "transparent",
+            display: "flex",
+            flexDirection: "column",
+            alignItems: "center",
+            justifyContent: "center",
+            minHeight: 180,
+            cursor: "pointer",
+            transition: "all 0.3s ease",
+          }}
+          onDragOver={handleDragOver}
+          onDragLeave={handleDragLeave}
+          onDrop={handleDrop}
+          onClick={handleClick}
+        >
+          <input
+            type="file"
+            ref={fileInputRef}
+            onChange={handleFileChange}
+            accept=".pdf,.txt,.html,.md"
+            style={{ display: "none" }}
+          />
+          {selectedDocument?.name && !isDefaultDocument && !urlSelected ? (
+            <>
+              <InsertDriveFileIcon
                 sx={{ fontSize: 50, color: "primary.main", mb: 1 }}
               />
-            )}
-            <Typography variant="h6" component="div" gutterBottom>
-              {isLoading
-                ? "Uploading your file..."
-                : "Drag and drop your file here or click to browse"}
-            </Typography>
-            <Typography variant="body2" color="text.secondary">
-              Accepted formats: PDF, TXT, HTML, MD
-            </Typography>
-          </>
-        )}
+              <Typography variant="h6" component="div" gutterBottom>
+                {selectedDocument.name}
+              </Typography>
+              <Typography variant="body2" color="text.secondary">
+                Click to upload a different file
+              </Typography>
+            </>
+          ) : (
+            <>
+              {isLoading && !urlSelected ? (
+                <CircularProgress size={50} sx={{ mb: 1 }} />
+              ) : (
+                <CloudUploadIcon
+                  sx={{ fontSize: 50, color: "primary.main", mb: 1 }}
+                />
+              )}
+              <Typography variant="h6" component="div" gutterBottom>
+                {isLoading && !urlSelected
+                  ? "Uploading your file..."
+                  : "Drag and drop your file here or click to browse"}
+              </Typography>
+              <Typography variant="body2" color="text.secondary">
+                Accepted formats: PDF, TXT, HTML, MD
+              </Typography>
+            </>
+          )}
+        </Box>
+
+        {/* Champ d'entrée URL */}
+        <Box
+          sx={{
+            flex: 1,
+            p: 4,
+            mt: 2,
+            borderRadius: 1.5,
+            border: urlSelected
+              ? `2px solid ${theme.palette.primary.main}`
+              : `2px dashed ${theme.palette.divider}`,
+            display: "flex",
+            flexDirection: "column",
+            alignItems: "center",
+            justifyContent: "center",
+            minHeight: 180,
+            transition: "all 0.3s ease",
+          }}
+        >
+          {selectedDocument?.name && urlSelected ? (
+            <>
+              <LinkIcon sx={{ fontSize: 50, color: "primary.main", mb: 1 }} />
+              <Typography
+                variant="h6"
+                component="div"
+                gutterBottom
+                noWrap
+                sx={{
+                  maxWidth: "100%",
+                  textOverflow: "ellipsis",
+                  overflow: "hidden",
+                }}
+              >
+                {selectedDocument?.domain || "URL processed"}
+              </Typography>
+              <TextField
+                fullWidth
+                variant="outlined"
+                label="Enter a new URL"
+                placeholder="https://example.com"
+                value={urlInput}
+                onChange={handleUrlInputChange}
+                disabled={isLoading}
+              />
+            </>
+          ) : (
+            <>
+              {isLoading && urlSelected ? (
+                <CircularProgress size={50} sx={{ mb: 1 }} />
+              ) : (
+                <LinkIcon sx={{ fontSize: 50, color: "primary.main", mb: 1 }} />
+              )}
+              <Typography variant="h6" component="div" gutterBottom>
+                {isLoading && urlSelected ? "Processing URL..." : "Enter a URL"}
+              </Typography>
+              <TextField
+                fullWidth
+                variant="outlined"
+                label="Website URL"
+                placeholder="https://example.com"
+                value={urlInput}
+                onChange={handleUrlInputChange}
+                disabled={isLoading}
+              />
+            </>
+          )}
+        </Box>
       </Box>
 
       <Box sx={{ display: "flex", justifyContent: "center" }}>
@@ -465,19 +422,27 @@ function CreateForm({ onStartGeneration }) {
           disabled={!sessionId}
           sx={{ mt: 2 }}
         >
-          Generate Benchmark
+          {!sessionId
+            ? "Select a document first"
+            : isDefaultDocument
+            ? `Generate Benchmark from "${selectedDocument?.name}"`
+            : urlSelected
+            ? "Generate Benchmark from URL"
+            : "Generate Benchmark from File"}
         </Button>
       </Box>
 
       <Snackbar
         open={openSnackbar}
-        autoHideDuration={6000}
+        autoHideDuration={5000}
         onClose={handleCloseSnackbar}
         anchorOrigin={{ vertical: "bottom", horizontal: "right" }}
       >
         <Alert
           onClose={handleCloseSnackbar}
           severity={uploadStatus?.success ? "success" : "error"}
+          variant="outlined"
+          elevation={1}
           sx={{ width: "100%" }}
         >
           {uploadStatus?.message}

frontend/src/components/Benchmark/DefaultDocumentCard.jsx

@@ -0,0 +1,104 @@
+import React from "react";
+import {
+  Box,
+  Typography,
+  CircularProgress,
+  IconButton,
+  Tooltip,
+} from "@mui/material";
+import { alpha } from "@mui/material/styles";
+import VisibilityIcon from "@mui/icons-material/Visibility";
+
+/**
+ * Composant pour afficher une carte de document par défaut
+ * @param {Object} props - Propriétés du composant
+ * @returns {JSX.Element} - Carte de document par défaut
+ */
+const DefaultDocumentCard = ({
+  doc,
+  theme,
+  selectedDocument,
+  isDefaultDocument,
+  urlSelected,
+  isLoadingContent,
+  handleDefaultDocClick,
+  handleViewDocument,
+  resetSelection,
+}) => {
+  // Vérifier si cette carte est actuellement sélectionnée
+  const isSelected = selectedDocument?.id === doc.id;
+
+  // Vérifier si une autre méthode est active (URL ou fichier)
+  const otherMethodActive =
+    (selectedDocument && !isDefaultDocument) || (urlSelected && !isSelected);
+
+  return (
+    <Box
+      elevation={2}
+      sx={{
+        p: 2,
+        display: "flex",
+        flexDirection: "column",
+        borderRadius: 1.5,
+        alignItems: "center",
+        cursor: "pointer",
+        transition: "all 0.2s ease",
+        height: "100%",
+        position: "relative",
+        border: isSelected
+          ? `2px solid ${theme.palette.primary.main}`
+          : `2px solid ${theme.palette.divider}`,
+        "&:hover": {
+          transform: "translateY(-2px)",
+          boxShadow: 3,
+        },
+      }}
+      onClick={() => handleDefaultDocClick(doc)}
+    >
+      <Tooltip title="View content">
+        <IconButton
+          onClick={(e) => {
+            e.stopPropagation();
+            handleViewDocument(doc);
+          }}
+          sx={{
+            position: "absolute",
+            top: 4,
+            right: 4,
+            color: "text.secondary",
+            opacity: 0.4,
+            "&:hover": {
+              opacity: 0.8,
+              backgroundColor: alpha(theme.palette.primary.main, 0.05),
+            },
+            padding: 0.3,
+            "& .MuiSvgIcon-root": {
+              fontSize: 16,
+            },
+          }}
+          disabled={isLoadingContent}
+        >
+          {isLoadingContent && selectedDocument?.id === doc.id ? (
+            <CircularProgress size={14} />
+          ) : (
+            <VisibilityIcon />
+          )}
+        </IconButton>
+      </Tooltip>
+      <Box sx={{ color: "primary.main", mb: 1 }}>{doc.icon}</Box>
+      <Typography variant="subtitle1" component="div" gutterBottom>
+        {doc.name}
+      </Typography>
+      <Typography
+        variant="body2"
+        color="text.secondary"
+        align="center"
+        sx={{ flexGrow: 1 }}
+      >
+        {doc.description}
+      </Typography>
+    </Box>
+  );
+};
+
+export default DefaultDocumentCard;

frontend/src/components/Benchmark/DocumentViewerDialog.jsx

@@ -0,0 +1,129 @@
+import React from "react";
+import {
+  Dialog,
+  DialogTitle,
+  DialogContent,
+  Box,
+  Typography,
+  CircularProgress,
+  IconButton,
+  Tooltip,
+} from "@mui/material";
+import DownloadIcon from "@mui/icons-material/Download";
+import CloseIcon from "@mui/icons-material/Close";
+
+/**
+ * Composant pour afficher un document dans une boîte de dialogue
+ * @param {Object} props - Propriétés du composant
+ * @returns {JSX.Element} - Dialogue pour visualiser un document
+ */
+const DocumentViewerDialog = ({
+  open,
+  onClose,
+  document,
+  content,
+  isLoading,
+  isDownloading,
+  handleDownload,
+}) => {
+  // Déterminer le type de document
+  const getDocumentType = () => {
+    if (!document) return "";
+
+    if (document.id === "the-bitter-lesson") return "HTML";
+    if (document.id === "hurricane-faq") return "Markdown";
+    return "Text";
+  };
+
+  return (
+    <Dialog
+      open={open}
+      onClose={onClose}
+      maxWidth="md"
+      fullWidth
+      aria-labelledby="document-content-dialog-title"
+    >
+      <DialogTitle id="document-content-dialog-title">
+        <Box
+          sx={{
+            display: "flex",
+            justifyContent: "space-between",
+            alignItems: "flex-start",
+          }}
+        >
+          <Box>
+            {document && (
+              <Typography variant="h6" sx={{ fontWeight: 600 }}>
+                {document.name}
+              </Typography>
+            )}
+            <Typography variant="body2" color="text.secondary">
+              {document && getDocumentType()}
+            </Typography>
+          </Box>
+          <Box sx={{ display: "flex", gap: 1 }}>
+            {document && (
+              <Tooltip title="Download document">
+                <IconButton
+                  edge="end"
+                  color="inherit"
+                  onClick={() => handleDownload(document)}
+                  disabled={isDownloading}
+                  aria-label="download"
+                  sx={{
+                    color: "text.secondary",
+                    opacity: 0.4,
+                    "&:hover": {
+                      opacity: 0.8,
+                    },
+                  }}
+                >
+                  {isDownloading ? (
+                    <CircularProgress size={20} />
+                  ) : (
+                    <DownloadIcon />
+                  )}
+                </IconButton>
+              </Tooltip>
+            )}
+            <IconButton
+              edge="end"
+              color="inherit"
+              onClick={onClose}
+              aria-label="close"
+            >
+              <CloseIcon />
+            </IconButton>
+          </Box>
+        </Box>
+      </DialogTitle>
+      <DialogContent
+        dividers
+        sx={{
+          padding: 0,
+        }}
+      >
+        {isLoading ? (
+          <Box sx={{ display: "flex", justifyContent: "center", my: 4 }}>
+            <CircularProgress />
+          </Box>
+        ) : (
+          <Box
+            sx={{
+              maxHeight: "60vh",
+              overflow: "auto",
+              whiteSpace: "pre-wrap",
+              fontFamily: "monospace",
+              fontSize: "0.875rem",
+              p: 2.5,
+            }}
+          >
+            {content}
+          </Box>
+        )}
+      </DialogContent>
+    </Dialog>
+  );
+};
+
+export default DocumentViewerDialog;

frontend/src/components/Benchmark/Generator.jsx

@@ -244,6 +244,25 @@ const Generator = ({ sessionId, isDefaultDocument, onComplete }) => {
     );
 
     if (hasError) {
+      // Vérifier d'abord le cas spécifique du document avec info insuffisante
+      const insufficientInfoMessage = generationLogs.find((log) =>
+        log.includes(
+          "Failed to generate benchmark: The document does not contain enough information"
+        )
+      );
+
+      if (insufficientInfoMessage) {
+        setError(
+          "Your document doesn't contain enough information to generate a benchmark. Please try with a more comprehensive document that includes richer content."
+        );
+        notifyGenerationComplete(
+          false,
+          null,
+          "Insufficient document information"
+        );
+        return;
+      }
+
       const errorMessage =
         generationLogs.find(
           (log) =>

frontend/src/components/Benchmark/hooks/useBenchmarkLogs.js

@@ -19,6 +19,21 @@ export const useBenchmarkLogs = (sessionId, isDefault, onComplete) => {
   const [generationComplete, setGenerationComplete] = useState(false);
 
   const checkForErrors = (logs) => {
+    // Check for document with insufficient information error
+    const hasInsufficientInfoError = logs.some((log) =>
+      log.includes(
+        "Failed to generate benchmark: The document does not contain enough information"
+      )
+    );
+
+    if (hasInsufficientInfoError) {
+      return {
+        hasError: true,
+        error:
+          "Your document doesn't contain enough information to generate a benchmark. Please try with a more comprehensive document that includes richer content.",
+      };
+    }
+
     // Check for rate limiting errors
     const hasRateLimitError = logs.some(
       (log) =>
@@ -65,6 +80,22 @@ export const useBenchmarkLogs = (sessionId, isDefault, onComplete) => {
       };
     }
 
+    // Check for insufficient content errors
+    const hasInsufficientContentError = logs.some(
+      (log) =>
+        log.includes(
+          "No valid questions produced in single_shot_question_generation"
+        ) || log.includes("No parseable JSON found")
+    );
+
+    if (hasInsufficientContentError) {
+      return {
+        hasError: true,
+        error:
+          "The document does not contain enough information to generate a meaningful benchmark. Please try with a more detailed document.",
+      };
+    }
+
     return { hasError: false };
   };
 

frontend/src/components/Benchmark/hooks/useDocumentSelection.js

@@ -0,0 +1,322 @@
+import { useState, useRef } from "react";
+import API_CONFIG from "../../../config/api";
+
+/**
+ * Hook personnalisé pour gérer la sélection et l'upload de documents
+ * @returns {Object} - Méthodes et états pour la gestion des documents
+ */
+export const useDocumentSelection = (onStartGeneration) => {
+  const [isDragging, setIsDragging] = useState(false);
+  const [uploadStatus, setUploadStatus] = useState(null);
+  const [isLoading, setIsLoading] = useState(false);
+  const [sessionId, setSessionId] = useState(null);
+  const [selectedDocument, setSelectedDocument] = useState(null);
+  const [isDefaultDocument, setIsDefaultDocument] = useState(false);
+  const [urlInput, setUrlInput] = useState("");
+  const [urlSelected, setUrlSelected] = useState(false);
+  const fileInputRef = useRef(null);
+
+  const handleDragOver = (e) => {
+    e.preventDefault();
+    setIsDragging(true);
+  };
+
+  const handleDragLeave = () => {
+    setIsDragging(false);
+  };
+
+  const handleClick = () => {
+    // Lors d'un clic sur la zone de fichier, seulement ouvrir le dialogue de sélection
+    // mais ne pas réinitialiser les autres options avant que l'utilisateur ne sélectionne réellement un fichier
+    fileInputRef.current.click();
+  };
+
+  const handleFileChange = (e) => {
+    const file = e.target.files[0];
+    if (!file) return;
+
+    // Maintenant que l'utilisateur a choisi un fichier, réinitialiser les autres options
+    setUrlInput("");
+    setUrlSelected(false);
+    setIsDefaultDocument(false);
+    setSessionId(null); // Reset session ID until upload completes
+
+    // Check if it's a PDF, TXT, HTML or MD
+    if (
+      !file.name.endsWith(".pdf") &&
+      !file.name.endsWith(".txt") &&
+      !file.name.endsWith(".html") &&
+      !file.name.endsWith(".md")
+    ) {
+      setUploadStatus({
+        success: false,
+        message: "Only PDF, TXT, HTML and MD files are accepted",
+      });
+      return { success: false, error: "Invalid file format" };
+    }
+
+    // Check file size limit (3MB = 3145728 bytes)
+    if (file.size > 1048576 * 2) {
+      setUploadStatus({
+        success: false,
+        message: "File size exceeds the 2MB limit",
+      });
+      return { success: false, error: "File too large" };
+    }
+
+    handleFileUpload(file);
+    return { success: true };
+  };
+
+  const handleFileUpload = async (file) => {
+    setIsLoading(true);
+    setUploadStatus(null);
+
+    // Réinitialiser les sélections précédentes
+    setSelectedDocument(null);
+
+    try {
+      const formData = new FormData();
+      formData.append("file", file);
+
+      const response = await fetch(`${API_CONFIG.BASE_URL}/upload`, {
+        method: "POST",
+        body: formData,
+      });
+
+      const result = await response.json();
+
+      if (response.ok) {
+        setUploadStatus({
+          success: true,
+          message: "File uploaded successfully",
+        });
+        setSessionId(result.session_id);
+        setSelectedDocument({ name: file.name });
+        // Fichier uploadé avec succès, donc on désactive les autres options
+        setIsDefaultDocument(false);
+        setUrlSelected(false);
+        return { success: true };
+      } else {
+        setUploadStatus({
+          success: false,
+          message: result.detail || "Upload failed",
+        });
+        return { success: false, error: result.detail || "Upload failed" };
+      }
+    } catch (error) {
+      setUploadStatus({
+        success: false,
+        message: "Server connection error",
+      });
+      return { success: false, error: "Server connection error" };
+    } finally {
+      setIsLoading(false);
+    }
+  };
+
+  const handleDrop = async (e) => {
+    e.preventDefault();
+    setIsDragging(false);
+
+    // Réinitialiser les autres options
+    setUrlInput("");
+    setUrlSelected(false);
+    setIsDefaultDocument(false);
+    fileInputRef.current.value = "";
+    setSessionId(null); // Reset session ID until upload completes
+
+    const file = e.dataTransfer.files[0];
+    if (!file) {
+      setUploadStatus({
+        success: false,
+        message: "No file detected",
+      });
+      return { success: false, error: "No file detected" };
+    }
+
+    // Check if it's a PDF, TXT, HTML or MD
+    if (
+      !file.name.endsWith(".pdf") &&
+      !file.name.endsWith(".txt") &&
+      !file.name.endsWith(".html") &&
+      !file.name.endsWith(".md")
+    ) {
+      setUploadStatus({
+        success: false,
+        message: "Only PDF, TXT, HTML and MD files are accepted",
+      });
+      return { success: false, error: "Invalid file format" };
+    }
+
+    // Check file size limit (3MB = 3145728 bytes)
+    if (file.size > 1048576 * 3) {
+      setUploadStatus({
+        success: false,
+        message: "File size exceeds the 3MB limit",
+      });
+      return { success: false, error: "File too large" };
+    }
+
+    handleFileUpload(file);
+    return { success: true };
+  };
+
+  const handleDefaultDocClick = (doc) => {
+    // Réinitialiser les autres options
+    setUrlInput("");
+    setUrlSelected(false);
+    fileInputRef.current.value = "";
+
+    // Set the selected document
+    setSelectedDocument(doc);
+    if (doc) {
+      setSessionId(doc.id);
+      setIsDefaultDocument(true);
+    } else {
+      // Si on désélectionne
+      setIsDefaultDocument(false);
+      setSessionId(null);
+    }
+  };
+
+  const handleGenerateClick = () => {
+    if (onStartGeneration && sessionId) {
+      onStartGeneration(sessionId, isDefaultDocument);
+    } else if (!sessionId) {
+      setUploadStatus({
+        success: false,
+        message: "Please select or upload a document first",
+      });
+    }
+  };
+
+  const handleUrlInputChange = async (e) => {
+    const url = e.target.value;
+    setUrlInput(url);
+
+    // Ne pas réinitialiser les autres options tant que l'URL n'est pas valide
+    // Vérification simple d'URL valide avec regex
+    const urlRegex =
+      /^(https?:\/\/)?([\da-z.-]+)\.([a-z.]{2,6})([/\w .-]*)*\/?$/;
+
+    if (url && urlRegex.test(url)) {
+      // URL valide, maintenant réinitialiser les autres options
+      setIsDefaultDocument(false);
+      fileInputRef.current.value = "";
+
+      setUrlSelected(true);
+
+      // Si l'URL semble valide et a une longueur suffisante, on la traite
+      if (url.length > 10) {
+        await handleUrlUpload(url);
+      }
+    } else if (url && url.length > 5) {
+      // Si l'URL n'est pas valide mais que l'utilisateur a saisi quelque chose
+      setUrlSelected(false);
+      setUploadStatus({
+        success: false,
+        message: "Please enter a valid URL",
+      });
+    } else {
+      // Si le champ est vide ou presque vide, réinitialiser complètement
+      setUrlSelected(false);
+
+      // Si on avait déjà une URL sélectionnée avant et qu'on vide le champ
+      // il faut aussi réinitialiser le document sélectionné et la session
+      if (urlSelected) {
+        setSelectedDocument(null);
+        setSessionId(null);
+      }
+    }
+  };
+
+  const handleUrlUpload = async (url) => {
+    setIsLoading(true);
+    setUploadStatus(null);
+
+    // Réinitialiser les sélections précédentes
+    setSelectedDocument(null);
+    setIsDefaultDocument(false);
+
+    try {
+      // Création d'un FormData pour envoyer l'URL
+      const formData = new FormData();
+      formData.append("url", url);
+
+      const response = await fetch(`${API_CONFIG.BASE_URL}/upload-url`, {
+        method: "POST",
+        body: formData,
+      });
+
+      const result = await response.json();
+
+      if (response.ok) {
+        setUploadStatus({
+          success: true,
+          message: "Content from URL uploaded successfully",
+        });
+        setSessionId(result.session_id);
+
+        // Extraire le domaine racine de l'URL
+        let domain = url;
+        try {
+          // Utiliser URL API pour extraire le hostname
+          const urlObj = new URL(
+            url.startsWith("http") ? url : `https://${url}`
+          );
+          domain = urlObj.hostname;
+        } catch (e) {
+          console.log("Error parsing URL, using original:", e);
+        }
+
+        setSelectedDocument({ name: url, domain: domain });
+        setUrlSelected(true);
+        return { success: true };
+      } else {
+        setUploadStatus({
+          success: false,
+          message: result.detail || "URL processing failed",
+        });
+        return {
+          success: false,
+          error: result.detail || "URL processing failed",
+        };
+      }
+    } catch (error) {
+      setUploadStatus({
+        success: false,
+        message: "Server connection error",
+      });
+      return { success: false, error: "Server connection error" };
+    } finally {
+      setIsLoading(false);
+    }
+  };
+
+  return {
+    // États
+    isDragging,
+    isLoading,
+    sessionId,
+    selectedDocument,
+    isDefaultDocument,
+    urlInput,
+    urlSelected,
+    uploadStatus,
+    fileInputRef,
+
+    // Gestionnaires d'événements
+    handleDragOver,
+    handleDragLeave,
+    handleClick,
+    handleFileChange,
+    handleDrop,
+    handleDefaultDocClick,
+    handleGenerateClick,
+    handleUrlInputChange,
+
+    // Setters pour les mises à jour externes
+    setUploadStatus,
+  };
+};