Update app.py

app.py

@@ -12,18 +12,22 @@ from werkzeug.utils import secure_filename
 import markdown # Pour convertir la réponse Markdown en HTML
 
 # --- Configuration Initiale ---
+
 load_dotenv() # Charge les variables depuis .env
 
 app = Flask(__name__)
+
 # Clé secrète FORTEMENT recommandée pour la sécurité des sessions
 app.config['SECRET_KEY'] = os.getenv('FLASK_SECRET_KEY', 'une-cle-secrete-tres-difficile-a-deviner')
 
 # Configuration pour les uploads
 UPLOAD_FOLDER = 'temp'
+
 # Extensions autorisées (incluant vidéo)
 ALLOWED_EXTENSIONS = {'txt', 'pdf', 'png', 'jpg', 'jpeg', 'mp4', 'mov', 'avi', 'mkv', 'webm'}
 VIDEO_EXTENSIONS = {'mp4', 'mov', 'avi', 'mkv', 'webm'} # Pour identifier les vidéos
 app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+
 # Augmenter la limite pour les vidéos (ex: 100MB) - Ajustez si nécessaire
 app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024
 
@@ -33,16 +37,16 @@ print(f"Dossier d'upload configuré : {os.path.abspath(UPLOAD_FOLDER)}")
 
 # --- Configuration de l'API Gemini ---
 # Utilisez les noms de modèles les plus récents auxquels vous avez accès
-MODEL_FLASH = 'gemini-2.0-flash'
-MODEL_PRO = 'gemini-2.5-pro-03-25' # Pro est souvent nécessaire/meilleur pour la vidéo
+MODEL_FLASH = 'gemini-1.5-flash-latest'
+MODEL_PRO = 'gemini-1.5-pro-latest' # Pro est souvent nécessaire/meilleur pour la vidéo
 
 # Instruction système pour le modèle
 SYSTEM_INSTRUCTION = "Tu es un assistant intelligent et amical nommé Mariam. Tu assistes les utilisateurs au mieux de tes capacités, y compris dans l'analyse de texte, d'images et de vidéos (via upload ou lien YouTube). Tu as été créé par Aenir."
 
 # Paramètres de sécurité (ajuster si nécessaire, BLOCK_NONE est très permissif)
 SAFETY_SETTINGS = [ {"category": c, "threshold": "BLOCK_NONE"} for c in [
-    "HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_HATE_SPEECH",
-    "HARM_CATEGORY_SEXUALLY_EXPLICIT", "HARM_CATEGORY_DANGEROUS_CONTENT"]]
+"HARM_CATEGORY_HARASSMENT", "HARM_CATEGORY_HATE_SPEECH",
+"HARM_CATEGORY_SEXUALLY_EXPLICIT", "HARM_CATEGORY_DANGEROUS_CONTENT"]]
 
 GEMINI_CONFIGURED = False
 gemini_client = None # Client API pour les opérations sur les fichiers (upload vidéo)
@@ -55,7 +59,7 @@ try:
     # Initialise le client pour les opérations sur les fichiers (upload vidéo avec polling)
     gemini_client = genai.Client(api_key=gemini_api_key)
     # Configure également l'espace de noms global pour GenerativeModel, etc.
-    
+
 
     # Vérifie si les modèles requis sont disponibles
     print("Vérification des modèles Gemini disponibles...")
@@ -69,6 +73,7 @@ try:
         missing = [model for model in required_models if model not in models_list]
         raise ValueError(f"Les modèles Gemini requis suivants sont manquants: {missing}")
 
+
 except Exception as e:
     print(f"ERREUR Critique lors de la configuration initiale de Gemini : {e}")
     print("L'application fonctionnera sans les fonctionnalités IA.")
@@ -94,13 +99,14 @@ def is_youtube_url(url):
         return False
     # Regex simple pour les formats courants d'URL YouTube
     youtube_regex = re.compile(
-        r'(https?://)?(www\.)?' # Protocole et www optionnels
-        r'(youtube|youtu|youtube-nocookie)\.(com|be)/' # Domaines youtube.com, youtu.be, etc.
-        r'(watch\?v=|embed/|v/|.+\?v=)?' # Différents chemins possibles
+        r'(https?://)?(www.)?' # Protocole et www optionnels
+        r'(youtube|youtu|youtube-nocookie).(com|be)/' # Domaines youtube.com, youtu.be, etc.
+        r'(watch?v=|embed/|v/|.+?v=)?' # Différents chemins possibles
         r'([^&=%\?]{11})') # L'ID vidéo de 11 caractères
     return youtube_regex.match(url) is not None
 
 # --- Fonction d'Upload Vidéo avec Polling ---
+
 def upload_video_with_polling(filepath, mime_type, max_wait_seconds=300, poll_interval=10):
     """
     Upload une vidéo via client.files.upload et attend son traitement.
@@ -118,7 +124,7 @@ def upload_video_with_polling(filepath, mime_type, max_wait_seconds=300, poll_in
 
         start_time = time.time()
         # Boucle de polling tant que l'état est "PROCESSING"
-        while video_file.state == genai_types.FileState.PROCESSING:
+        while video_file.state == genai.types.FileState.PROCESSING:
             elapsed_time = time.time() - start_time
             # Vérifie le timeout
             if elapsed_time > max_wait_seconds:
@@ -130,11 +136,11 @@ def upload_video_with_polling(filepath, mime_type, max_wait_seconds=300, poll_in
             video_file = gemini_client.files.get(name=video_file.name)
 
         # Vérifie l'état final après la boucle
-        if video_file.state == genai_types.FileState.FAILED:
+        if video_file.state == genai.types.FileState.FAILED:
             print(f"ERREUR: Le traitement de la vidéo a échoué. État: {video_file.state.name}")
             raise ValueError("Le traitement de la vidéo a échoué côté serveur.")
 
-        if video_file.state == genai_types.FileState.ACTIVE:
+        if video_file.state == genai.types.FileState.ACTIVE:
             print(f"Traitement vidéo terminé avec succès: {video_file.uri}")
             return video_file # Retourne l'objet fichier SDK réussi
 
@@ -155,6 +161,7 @@ def upload_video_with_polling(filepath, mime_type, max_wait_seconds=300, poll_in
         raise # Relance l'exception originale pour qu'elle soit gérée par l'appelant
 
 # --- Fonctions de Recherche Web (inchangées - implémentez si nécessaire) ---
+
 def perform_web_search(query):
     """Effectue une recherche web via l'API Serper (Exemple)."""
     serper_api_key = os.getenv("SERPER_API_KEY")
@@ -171,14 +178,14 @@ def format_search_results(data):
     """Met en forme les résultats de recherche (Exemple)."""
     if not data: return "Aucun résultat de recherche web pertinent."
     # ... (votre implémentation du formatage) ...
-    results = ["**Résultats Web:**"]
+    results = ["Résultats Web:"]
     if data.get('organic'):
         for item in data['organic'][:3]:
             results.append(f"- {item.get('title', '')}: {item.get('snippet', '')}")
-    return "\n".join(results)
-
+        return "\n".join(results)
 
 # --- Préparation Historique (inchangé) ---
+
 def prepare_gemini_history(chat_history):
     """Convertit l'historique de session pour l'API Gemini (texte seulement)."""
     gemini_history = []
@@ -273,12 +280,12 @@ def chat_api():
                 # Appel bloquant qui attend le traitement
                 processed_media_file = upload_video_with_polling(filepath, mime_type)
                 # Crée le Part Gemini à partir de l'objet File retourné
-                uploaded_media_part = genai_types.Part(file_data=processed_media_file)
+                uploaded_media_part = genai.types.Part(file_data=processed_media_file)
             else:
                 print("  Traitement FICHIER standard...")
                 # Utilise l'upload global plus simple pour les non-vidéos
                 processed_media_file = genai.upload_file(path=filepath, mime_type=mime_type)
-                uploaded_media_part = genai_types.Part(file_data=processed_media_file)
+                uploaded_media_part = genai.types.Part(file_data=processed_media_file)
             print(f"  Part Média ({media_type}) créé: {processed_media_file.uri}")
 
         # 2. Sinon, traiter l'URL YouTube si fournie et valide
@@ -296,8 +303,8 @@ def chat_api():
                 youtube_uri = youtube_url # Utilise l'URL validée comme URI
                 # Crée un Part FileData directement à partir de l'URI
                 # On peut spécifier un mime_type générique, Gemini gère les liens YT
-                uploaded_media_part = genai_types.Part(
-                    file_data=genai_types.FileData(file_uri=youtube_uri, mime_type="video/mp4")
+                uploaded_media_part = genai.types.Part(
+                    file_data=genai.types.FileData(file_uri=youtube_uri, mime_type="video/mp4")
                 )
                 print(f"  Part YouTube créé pour: {youtube_uri}")
                 # Ajoute un prompt par défaut si l'utilisateur n'en a pas mis
@@ -350,22 +357,22 @@ def chat_api():
                 # Construit un prompt enrichi
                 final_prompt_for_gemini = f"""Basé sur la question suivante et les informations web ci-dessous, fournis une réponse complète.
 
-Question Originale:
-"{prompt}"
+    Question Originale:
+    "{prompt}"
 
-Informations Web Pertinentes:
---- DEBUT RESULTATS WEB ---
-{formatted_results}
---- FIN RESULTATS WEB ---
+    Informations Web Pertinentes:
+    --- DEBUT RESULTATS WEB ---
+    {formatted_results}
+    --- FIN RESULTATS WEB ---
 
-Réponse:"""
+    Réponse:"""
                 print("  Prompt enrichi avec les résultats web.")
             else:
                 print("  Aucun résultat de recherche web trouvé ou pertinent.")
 
         # Ajouter la partie texte (originale ou enrichie) s'il y a du texte
         if final_prompt_for_gemini:
-             current_gemini_parts.append(genai_types.Part(text=final_prompt_for_gemini))
+             current_gemini_parts.append(genai.types.Part(text=final_prompt_for_gemini))
 
         # Vérification de sécurité : il doit y avoir au moins une partie (média ou texte)
         if not current_gemini_parts:
@@ -384,36 +391,35 @@ Réponse:"""
         print(f"  Modèle sélectionné: {selected_model_name}")
 
         # Crée l'instance du modèle spécifique pour cette requête
-            active_model = gemini_client.models.generate_content(
-        model_name=selected_model_name,
-        contents = contents_for_gemini
-        config = types.GenerateContentConfig(
-            system_instruction=SYSTEM_INSTRUCTION,
-    safety_settings=[
-
-        types.SafetySetting(
-          category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
-          threshold=types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-        ),
-        types.SafetySetting(
-          category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
-          threshold=types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-        ),
-        types.SafetySetting(
-          category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
-          threshold=types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
-        ),
-        types.SafetySetting(
-          category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
-          threshold=types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+        active_model = gemini_client.models.generate_content(
+            model_name=selected_model_name,
+            contents = contents_for_gemini, # Virgule ajoutée ici
+            config = genai.types.GenerateContentConfig(
+                system_instruction=SYSTEM_INSTRUCTION,
+                safety_settings=[
+
+                    genai.types.SafetySetting(
+                      category=genai.types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                      threshold=genai.types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+                    ),
+                    genai.types.SafetySetting(
+                      category=genai.types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                      threshold=genai.types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+                    ),
+                    genai.types.SafetySetting(
+                      category=genai.types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                      threshold=genai.types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+                    ),
+                    genai.types.SafetySetting(
+                      category=genai.types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                      threshold=genai.types.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+                    )
+                ]
+            ),
         )
-    ]
-),
-    )
 
         print(f"  Envoi de la requête à {selected_model_name} ({len(contents_for_gemini)} messages/tours)...")
-        
-        #(non-streamé pour correspondre au code précédent)
+        # Appel API (non-streamé pour correspondre au code précédent)
         response = active_model
 
         # --- Traitement de la Réponse ---
@@ -480,7 +486,6 @@ Réponse:"""
             except OSError as e:
                 print(f"  ERREUR lors de la suppression du fichier temporaire '{filepath_to_delete}': {e}")
 
-
 @app.route('/clear', methods=['POST'])
 def clear_chat():
     """Efface l'historique de chat dans la session."""
@@ -498,8 +503,8 @@ def clear_chat():
         flash("Conversation effacée.", "info")
         return redirect(url_for('root')) # Redirige vers la page d'accueil
 
-
 # --- Démarrage de l'application Flask ---
+
 if __name__ == '__main__':
     print("Démarrage du serveur Flask...")
     # Utiliser un port différent si 5000 est déjà pris (ex: 5001)