Update tools/tools.py

tools/tools.py

@@ -8,6 +8,11 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
+
+@spaces.GPU(duration=20)
+def initialize_gpu() :
+    return 0
+
 @tool
 def vision_tool(prompt: str, image_list: list) -> str:
     """
@@ -20,7 +25,9 @@ def vision_tool(prompt: str, image_list: list) -> str:
     """
     import io, base64, os
     from smolagents import OpenAIServerModel
-    from PIL import Image 
+    from PIL import Image
+    
+    initialize_gpu()
     
     model = OpenAIServerModel(
         model_id='gemma4:31b-cloud',
@@ -45,7 +52,6 @@ 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:
     """
@@ -59,6 +65,8 @@ def ask_youtube_full_pipeline(url: str, question: str) -> str:
     from PIL import Image
     from smolagents import OpenAIServerModel
 
+    initialize_gpu()
+    
     # 1. Extraction ID interne
     patterns = [r"v=([a-zA-Z0-9_-]{11})", r"youtu\.be/([a-zA-Z0-9_-]{11})"]
     video_id = None
@@ -109,7 +117,6 @@ 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:
     """
@@ -120,6 +127,8 @@ def read_pdf_file(file_path: str) -> str:
     """
     import os
     from pypdf import PdfReader
+
+    initialize_gpu()
     
     if not os.path.exists(file_path):
         return f"Erreur : Le fichier au chemin '{file_path}' est introuvable."
@@ -141,7 +150,6 @@ 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:
     """
@@ -152,6 +160,8 @@ def read_docx_file(file_path: str) -> str:
     """
     import docx
     import os
+
+    initialize_gpu()
     
     if not os.path.exists(file_path):
         return f"Erreur : Le fichier au chemin '{file_path}' est introuvable."
@@ -324,7 +334,6 @@ def extract_text_via_ocr(image_path: str) -> str:
     except Exception as e:
         return f"Erreur lors de l'appel API OCR : {str(e)}"
         
-@spaces.GPU(duration=20)
 @tool
 def summarize_csv_data(path: str, query: str = "") -> str:
     """