Update tools/tools.py

tools/tools.py

@@ -1,12 +1,12 @@
 import os
 from smolagents import tool
 import spaces
+from dotenv import load_dotenv
 # ----------------------------------------------------------------------------
 # SECTION 1: Vision & Multimodal
 # ----------------------------------------------------------------------------
 
-
-
+load_dotenv()
 
 @tool
 def vision_tool(prompt: str, image_list: list) -> str:
@@ -45,6 +45,7 @@ def vision_tool(prompt: str, image_list: list) -> str:
 # ----------------------------------------------------------------------------
 # SECTION 2: YouTube Pipeline (Tout-en-un pour la robustesse)
 # ----------------------------------------------------------------------------
+@spaces.GPU(duration=20)
 @tool
 def ask_youtube_full_pipeline(url: str, question: str) -> str:
     """
@@ -108,6 +109,7 @@ def ask_youtube_full_pipeline(url: str, question: str) -> str:
 # ----------------------------------------------------------------------------
 # SECTION 3: Fichiers & Audio
 # ----------------------------------------------------------------------------
+@spaces.GPU(duration=10)
 @tool
 def read_pdf_file(file_path: str) -> str:
     """
@@ -139,6 +141,7 @@ def read_pdf_file(file_path: str) -> str:
     except Exception as e:
         return f"Erreur lors de la lecture du fichier PDF : {str(e)}"
 
+@spaces.GPU(duration=10)
 @tool
 def read_docx_file(file_path: str) -> str:
     """