Update app.py

app.py

@@ -4,10 +4,6 @@ import os
 from PIL import Image
 import tempfile
 import PIL.Image
-import requests
-from urllib.parse import urlparse
-import ssl
-import time
 
 app = Flask(__name__)
 
@@ -27,6 +23,7 @@ safety_settings = [
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
 ]
 
+
 @app.route('/')
 def generale():
     return render_template("generale.html")
@@ -52,23 +49,26 @@ def upload_and_process_file(file_path):
                 raise ValueError(f"Le fichier {file_path} est vide")
                 
             # Upload du fichier
-            # Note: genai.upload_file n'existe pas dans l'API Gemini, tu dois utiliser des multipart
-            #       comme dans l'exemple ci-dessous pour uploader des fichiers.
-            #       Ici, je vais simuler l'upload avec un print pour l'exemple.
-            print(f"Upload réussi (simulation): {file_path}")
+            uploaded_file = genai.upload_file(path=file_path)
+            print(f"Upload réussi: {uploaded_file.uri}")
 
-            # Attente du traitement (simulation)
-            # Dans une vraie implémentation, tu devrais vérifier l'état de l'upload
+            # Attente du traitement
             timeout = 300  # 5 minutes
             start_time = time.time()
             
-            while time.time() - start_time < timeout:
-                print(f"En attente du traitement (simulation)... Temps écoulé: {int(time.time() - start_time)}s")
+            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)
-                break  # Simule la fin du traitement
+                uploaded_file = genai.get_file(uploaded_file.name)
 
-            print(f"Traitement terminé avec succès (simulation): {file_path}")
-            return {"path": file_path, "status": "success"}  # Retourne un objet similaire
+            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
 
         except ssl.SSLError as e:
             print(f"Erreur SSL lors de l'upload (tentative {attempt + 1}): {e}")
@@ -146,7 +146,6 @@ def submit_question():
     question = request.form.get('question')
     urls = request.form.getlist('urls')
     files = request.files.getlist('files')
-    web_search = request.form.get('web_search') == 'true'  # Récupère l'état de la recherche web
     
     print("URLs reçues:", urls)
     content = [question]
@@ -162,8 +161,7 @@ def submit_question():
                     if file.filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):
                         content.append(PIL.Image.open(temp_file.name))
                     else:
-                        # Simulation de l'upload pour les autres types de fichiers
-                        uploaded_file = {"path": temp_file.name, "status": "success"}
+                        uploaded_file = upload_and_process_file(temp_file.name)
                         content.append(uploaded_file)
 
         # Traitement des URLs
@@ -173,29 +171,25 @@ def submit_question():
                 video_path = download_youtube_video(url)
                 if video_path:
                     temp_files.append(video_path)
-                    # Simulation de l'upload pour les vidéos
-                    uploaded_file = {"path": video_path, "status": "success"}
+                    uploaded_file = upload_and_process_file(video_path)
                     content.append(uploaded_file)
             elif url.lower().endswith('.pdf'):
                 pdf_path = telecharger_pdf(url)
                 if pdf_path:
                     temp_files.append(pdf_path)
-                    # Simulation de l'upload pour les PDFs
-                    uploaded_file = {"path": pdf_path, "status": "success"}
+                    uploaded_file = upload_and_process_file(pdf_path)
                     content.append(uploaded_file)
+
+       
        
 
         # Génération de contenu avec Gemini
         model = genai.GenerativeModel(
-            model_name="gemini-1.5-flash",
+            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."
         )
-        # Intégration de la recherche web
-        if web_search:
-            response = model.generate_content(question,tools='google_search_retrieval',request_options={"timeout": 600})
-        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:
@@ -206,4 +200,6 @@ 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}") 
+
+