Update app.py

app.py

@@ -4,6 +4,11 @@ import os
 from PIL import Image
 import tempfile
 import PIL.Image
+import time
+import ssl
+from urllib.parse import urlparse
+import yt_dlp
+import requests
 
 app = Flask(__name__)
 
@@ -23,14 +28,10 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
-
 @app.route('/')
 def generale():
     return render_template("generale.html")
 
-
-
-
 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
@@ -83,8 +84,7 @@ def upload_and_process_file(file_path):
                 time.sleep(retry_delay * (attempt + 1))
             else:
                 raise
-                
-                
+
 
 def is_youtube_url(url):
     """Vérifie si l'URL est une URL YouTube"""
@@ -141,12 +141,14 @@ def allowed_file(filename):
     ALLOWED_EXTENSIONS = {'pdf', 'png', 'jpg', 'jpeg', 'gif', 'mp4', 'mov', 'avi'}
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+
 @app.route('/submit', methods=['POST'])
 def submit_question():
     question = request.form.get('question')
     urls = request.form.getlist('urls')
     files = request.files.getlist('files')
-    
+    web_search = request.form.get('webSearch') == 'true' # Get the boolean value of the web search mode
+
     print("URLs reçues:", urls)
     content = [question]
     temp_files = []
@@ -189,7 +191,11 @@ def submit_question():
             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."
         )
-        response = model.generate_content(content, request_options={"timeout": 600})
+        if web_search:
+            response = model.generate_content(contents=content, tools='google_search_retrieval', request_options={"timeout": 600})
+        else:
+            response = model.generate_content(content, request_options={"timeout": 600})
+
         return jsonify({"response": response.text})
 
     except Exception as e:
@@ -200,6 +206,4 @@ def submit_question():
             try:
                 os.unlink(temp_file)
             except Exception as e:
-                print(f"Erreur lors de la suppression du fichier temporaire {temp_file}: {e}") 
-
-
+                print(f"Erreur lors de la suppression du fichier temporaire {temp_file}: {e}")