Update app.py

app.py

@@ -4,11 +4,11 @@ import os
 from PIL import Image
 import tempfile
 import PIL.Image
+import requests
 import time
 import ssl
-import requests
 from urllib.parse import urlparse
-
+import yt_dlp
 
 app = Flask(__name__)
 
@@ -28,6 +28,26 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
+# Configuration de la recherche Google
+google_search = genai.Tool(
+    function_declarations=[
+        genai.FunctionDeclaration(
+            name="google_search",
+            description="A search engine. Useful for when you need to answer questions about current events. Input should be a search query.",
+            parameters={
+                "type": "object",
+                "properties": {
+                    "query": {
+                        "type": "string",
+                        "description": "The search query.",
+                    },
+                },
+                "required": ["query"],
+            },
+        ),
+    ]
+)
+
 @app.route('/')
 def generale():
     return render_template("generale.html")
@@ -36,19 +56,19 @@ def upload_and_process_file(file_path):
     """Upload et traite un fichier avec l'API Gemini avec gestion des erreurs améliorée"""
     max_retries = 3
     retry_delay = 2  # secondes
-
+    
     for attempt in range(max_retries):
         try:
             print(f"Tentative d'upload {attempt + 1}/{max_retries} pour {file_path}")
-
+            
             # Vérification du fichier
             if not os.path.exists(file_path):
                 raise FileNotFoundError(f"Le fichier {file_path} n'existe pas")
-
+                
             file_size = os.path.getsize(file_path)
             if file_size == 0:
                 raise ValueError(f"Le fichier {file_path} est vide")
-
+                
             # Upload du fichier
             uploaded_file = genai.upload_file(path=file_path)
             print(f"Upload réussi: {uploaded_file.uri}")
@@ -56,18 +76,18 @@ def upload_and_process_file(file_path):
             # Attente du traitement
             timeout = 300  # 5 minutes
             start_time = time.time()
-
+            
             while uploaded_file.state.name == "PROCESSING":
                 if time.time() - start_time > timeout:
                     raise TimeoutError("Timeout pendant le traitement du fichier")
-
+                
                 print(f"En attente du traitement... Temps écoulé: {int(time.time() - start_time)}s")
                 time.sleep(10)
                 uploaded_file = genai.get_file(uploaded_file.name)
 
             if uploaded_file.state.name == "FAILED":
                 raise ValueError(f"Échec du traitement: {uploaded_file.state.name}")
-
+                
             print(f"Traitement terminé avec succès: {uploaded_file.uri}")
             return uploaded_file
 
@@ -77,13 +97,15 @@ def upload_and_process_file(file_path):
                 time.sleep(retry_delay * (attempt + 1))
             else:
                 raise
-
+                
         except Exception as e:
             print(f"Erreur lors de l'upload (tentative {attempt + 1}): {e}")
             if attempt < max_retries - 1:
                 time.sleep(retry_delay * (attempt + 1))
             else:
                 raise
+                
+                
 
 def is_youtube_url(url):
     """Vérifie si l'URL est une URL YouTube"""
@@ -101,23 +123,23 @@ def download_youtube_video(url):
                 'no_warnings': True,
                 'extract_flat': False
             }
-
+            
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 print(f"Téléchargement de la vidéo: {url}")
                 info = ydl.extract_info(url, download=True)
                 video_path = os.path.join(temp_dir, ydl.prepare_filename(info))
-
+                
                 if not os.path.exists(video_path):
                     raise FileNotFoundError(f"La vidéo n'a pas été téléchargée correctement: {video_path}")
-
+                
                 # Copie vers un fichier temporaire permanent
                 temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(video_path)[1])
                 with open(video_path, 'rb') as f:
                     temp_file.write(f.read())
-
+                
                 print(f"Vidéo téléchargée avec succès: {temp_file.name}")
                 return temp_file.name
-
+                
     except Exception as e:
         print(f"Erreur lors du téléchargement de la vidéo: {e}")
         return None
@@ -145,12 +167,12 @@ def submit_question():
     question = request.form.get('question')
     urls = request.form.getlist('urls')
     files = request.files.getlist('files')
-    use_web = request.form.get('use_web') == 'true'  # Get web mode selection
-
+    web_search_enabled = request.form.get('web_search') == 'true'  # Récupère l'état de la recherche web
+    
     print("URLs reçues:", urls)
     content = [question]
     temp_files = []
-
+    
     try:
         # Traitement des fichiers uploadés
         for file in files:
@@ -180,22 +202,20 @@ def submit_question():
                     uploaded_file = upload_and_process_file(pdf_path)
                     content.append(uploaded_file)
 
+       
+       
+        tools = [google_search] if web_search_enabled else None
         # Génération de contenu avec Gemini
         model = genai.GenerativeModel(
             model_name="models/gemini-1.5-flash-002",
             safety_settings=safety_settings,
-            system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam."
+            system_instruction="Tu es un assistant intelligent. ton but est d'assister au mieux que tu peux. tu as été créé par Aenir et tu t'appelles Mariam.",
+            tools=tools
         )
-
-        # Use Google Search tool if web mode is selected
-        if use_web:
-            response = model.generate_content(
-                contents=question,
-                tools=[{"function_declarations": [{"name": "google_search", "description": "Tool to retrieve search results from Google Search.", "parameters": {"type": "object", "properties": {"query": {"type": "string", "description": "The query string to search for."}}, "required": ["query"]}}]}]
-)
-        else:
-            response = model.generate_content(content, request_options={"timeout": 600})
-
+        
+        response = model.generate_content(content, request_options={"timeout": 600})
+        
+        
         return jsonify({"response": response.text})
 
     except Exception as e: