Update app.py

app.py

@@ -1,22 +1,24 @@
 from flask import Flask, render_template, request, redirect, url_for, flash
 from flask_sqlalchemy import SQLAlchemy
-from sqlalchemy import func  # Import func
+from sqlalchemy import func
 from bleach import clean
 from bs4 import BeautifulSoup
 import os
-cle = os.environ.get("TOKEN")
-
-
+from google import genai
+from google.genai import types
 
+# Clé API Gemini
+api_key = "AIzaSyBzv8YdOqVfNlskpAJNvQTPS_AZ-P3t19k"
 
+# Configuration de Flask
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = cle
+app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get("TOKEN")  # Assurez-vous que cette variable est définie
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 app.config['SECRET_KEY'] = 'une_cle_secrete_tres_longue'  # Clé secrète pour les messages flash
 
 db = SQLAlchemy(app)
 
-# --- Modèles (adaptés de models.py, pour la cohérence) ---
+# --- Modèles ---
 
 class Matiere(db.Model):
     """Modèle représentant une matière."""
@@ -35,7 +37,7 @@ class SousCategorie(db.Model):
     nom = db.Column(db.String(64), nullable=False)
     matiere_id = db.Column(db.Integer, db.ForeignKey('matiere.id'), nullable=False)
     textes = db.relationship('Texte', backref='sous_categorie', lazy='dynamic', cascade="all, delete-orphan")
-    __table_args__ = (db.UniqueConstraint('nom', 'matiere_id', name='_nom_matiere_uc'), {})
+    __table_args__ = (db.UniqueConstraint('nom', 'matiere_id', name='_nom_matiere_uc'),)
 
     def __repr__(self):
         return f"<SousCategorie {self.nom}>"
@@ -51,11 +53,10 @@ class Texte(db.Model):
     def __repr__(self):
         return f"<Texte {self.titre}>"
 
-
 # --- Fonctions utilitaires ---
 
 def sanitize_html(html_content):
-    """Assainit le contenu HTML (comme dans ton admin.py)."""
+    """Assainit le contenu HTML."""
     allowed_tags = [
         'a', 'abbr', 'acronym', 'b', 'blockquote', 'br', 'code', 'div', 'em',
         'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'hr', 'i', 'img', 'li', 'ol', 'p',
@@ -79,9 +80,9 @@ def sanitize_html(html_content):
 @app.route('/', methods=['GET', 'POST'])
 def gere():
     if request.method == 'POST':
-        # Traitement des actions (ajout, modification, suppression)
         action = request.form.get('action')
 
+        # Ajouter une matière
         if action == 'add_matiere':
             nom_matiere = request.form.get('nom_matiere')
             if nom_matiere:
@@ -94,6 +95,7 @@ def gere():
                     db.session.rollback()
                     flash(f"Erreur lors de l'ajout de la matière : {e}", 'danger')
 
+        # Ajouter une sous-catégorie
         elif action == 'add_sous_categorie':
             nom_sous_categorie = request.form.get('nom_sous_categorie')
             matiere_id = request.form.get('matiere_id')
@@ -106,28 +108,59 @@ def gere():
                 except Exception as e:
                     db.session.rollback()
                     flash(f"Erreur : {e}", 'danger')
+
+        # Ajouter un texte avec génération IA
         elif action == 'add_texte':
             titre_texte = request.form.get('titre_texte')
-            contenu_texte = request.form.get('contenu_texte')
             sous_categorie_id = request.form.get('sous_categorie_id')
+            youtube_url = request.form.get('youtube_url')
+            prompt = request.form.get('prompt')
+            model = request.form.get('model')
 
-            if titre_texte and contenu_texte and sous_categorie_id:
-                 # Assainir le contenu HTML
-                contenu_texte_sanitized = sanitize_html(contenu_texte)
-                nouveau_texte = Texte(titre=titre_texte, contenu=contenu_texte_sanitized, sous_categorie_id=sous_categorie_id)
-                db.session.add(nouveau_texte)
+            if titre_texte and sous_categorie_id and youtube_url and prompt and model:
                 try:
+                    # Configurer le client Gemini
+                    client = genai.Client(api_key=api_key)
+
+                    # Préparer la requête pour l'IA
+                    contents = types.Content(
+                        parts=[
+                            types.Part(text=prompt),
+                            types.Part(file_data=types.FileData(file_uri=youtube_url))
+                        ]
+                    )
+
+                    # Générer le contenu avec le modèle sélectionné
+                    response = client.models.generate_content(
+                        model=model,
+                        contents=contents
+                    )
+
+                    # Récupérer et assainir le contenu généré
+                    contenu_texte = response.text
+                    contenu_texte_sanitized = sanitize_html(contenu_texte)
+
+                    # Ajouter le texte à la base de données
+                    nouveau_texte = Texte(
+                        titre=titre_texte,
+                        contenu=contenu_texte_sanitized,
+                        sous_categorie_id=sous_categorie_id
+                    )
+                    db.session.add(nouvelle_texte)
                     db.session.commit()
                     flash(f"Texte '{titre_texte}' ajouté avec succès.", 'success')
+
                 except Exception as e:
                     db.session.rollback()
-                    flash(f"Erreur lors de l'ajout : {e}", 'danger')
-
+                    flash(f"Erreur lors de la génération ou de l'ajout du texte : {str(e)}", 'danger')
+            else:
+                flash("Tous les champs sont requis pour ajouter un texte.", 'warning')
 
+        # Modifier une matière
         elif action.startswith('edit_matiere_'):
             matiere_id = int(action.split('_')[-1])
             matiere = Matiere.query.get_or_404(matiere_id)
-            matiere.nom = request.form.get(f'edit_nom_matiere_{matiere_id}')
+            matiere.nom = request.form.get(f'edit_nom_matiere_{matiere 艾迪}')
             try:
                 db.session.commit()
                 flash(f"Matière '{matiere.nom}' modifiée avec succès.", 'success')
@@ -135,6 +168,7 @@ def gere():
                 db.session.rollback()
                 flash(f"Erreur lors de la modification : {e}", 'danger')
 
+        # Modifier une sous-catégorie
         elif action.startswith('edit_sous_categorie_'):
             sous_categorie_id = int(action.split('_')[-1])
             sous_categorie = SousCategorie.query.get_or_404(sous_categorie_id)
@@ -147,7 +181,7 @@ def gere():
                 db.session.rollback()
                 flash(f"Erreur : {e}", 'danger')
 
-
+        # Modifier un texte
         elif action.startswith('edit_texte_'):
             texte_id = int(action.split('_')[-1])
             texte = Texte.query.get_or_404(texte_id)
@@ -156,7 +190,6 @@ def gere():
             if contenu:
                 texte.contenu = sanitize_html(contenu)
             texte.sous_categorie_id = request.form.get(f'edit_sous_categorie_id_{texte_id}')
-
             try:
                 db.session.commit()
                 flash(f"Texte '{texte.titre}' modifié avec succès.", 'success')
@@ -164,7 +197,7 @@ def gere():
                 db.session.rollback()
                 flash(f"Erreur : {e}", 'danger')
 
-
+        # Supprimer une matière
         elif action.startswith('delete_matiere_'):
             matiere_id = int(action.split('_')[-1])
             matiere = Matiere.query.get_or_404(matiere_id)
@@ -176,6 +209,7 @@ def gere():
                 db.session.rollback()
                 flash(f"Erreur : {e}", 'danger')
 
+        # Supprimer une sous-catégorie
         elif action.startswith('delete_sous_categorie_'):
             sous_categorie_id = int(action.split('_')[-1])
             sous_categorie = SousCategorie.query.get_or_404(sous_categorie_id)
@@ -184,10 +218,10 @@ def gere():
                 db.session.commit()
                 flash(f"Sous-catégorie '{sous_categorie.nom}' supprimée.", 'success')
             except Exception as e:
-                 db.session.rollback()
-                 flash(f"Erreur : {e}", 'danger')
-
+                db.session.rollback()
+                flash(f"Erreur : {e}", 'danger')
 
+        # Supprimer un texte
         elif action.startswith('delete_texte_'):
             texte_id = int(action.split('_')[-1])
             texte = Texte.query.get_or_404(texte_id)
@@ -197,11 +231,11 @@ def gere():
                 flash(f"Texte '{texte.titre}' supprimé avec succès.", 'success')
             except Exception as e:
                 db.session.rollback()
-                flash(f"Erreur : {e}",'danger')
+                flash(f"Erreur : {e}", 'danger')
 
         return redirect(url_for('gere'))
 
-    # Récupération de toutes les données pour l'affichage
+    # Récupération des données pour l'affichage
     matieres = Matiere.query.order_by(func.lower(Matiere.nom)).all()
     sous_categories = SousCategorie.query.order_by(func.lower(SousCategorie.nom)).all()
     textes = Texte.query.order_by(Texte.titre).all()
@@ -210,4 +244,4 @@ def gere():
 if __name__ == '__main__':
     with app.app_context():
         db.create_all()  # Crée les tables si elles n'existent pas
-    app.run(debug=True)  # Port différent pour éviter les conflits
+    app.run(debug=True)