Update app.py

app.py

@@ -1,24 +1,26 @@
-from flask import Flask, render_template, request, redirect, url_for, flash
+from flask import Flask, render_template, request, redirect, url_for, flash, jsonify
 from flask_sqlalchemy import SQLAlchemy
-from sqlalchemy import func
+from sqlalchemy import func  # Import func
 from bleach import clean
 from bs4 import BeautifulSoup
 import os
+# Import pour Gemini
 from google import genai
 from google.genai import types
 
-# Clé API Gemini
-api_key = "AIzaSyBzv8YdOqVfNlskpAJNvQTPS_AZ-P3t19k"
+cle = os.environ.get("TOKEN")
+# Clé API Google pour Gemini
+GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "AIzaSyBzv8YdOqVfNlskpAJNvQTPS_AZ-P3t19k")
+
 
-# Configuration de Flask
 app = Flask(__name__)
-app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get("TOKEN")  # Assurez-vous que cette variable est définie
+app.config['SQLALCHEMY_DATABASE_URI'] = cle
 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 ---
+# --- Modèles (adaptés de models.py, pour la cohérence) ---
 
 class Matiere(db.Model):
     """Modèle représentant une matière."""
@@ -37,7 +39,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}>"
@@ -53,10 +55,11 @@ class Texte(db.Model):
     def __repr__(self):
         return f"<Texte {self.titre}>"
 
+
 # --- Fonctions utilitaires ---
 
 def sanitize_html(html_content):
-    """Assainit le contenu HTML."""
+    """Assainit le contenu HTML (comme dans ton admin.py)."""
     allowed_tags = [
         'a', 'abbr', 'acronym', 'b', 'blockquote', 'br', 'code', 'div', 'em',
         'h1', 'h2', 'h3', 'h4', 'h5', 'h6', 'hr', 'i', 'img', 'li', 'ol', 'p',
@@ -75,14 +78,41 @@ def sanitize_html(html_content):
             tag.decompose()
     return str(soup)
 
+def generate_content_from_youtube(youtube_url, prompt, model_name):
+    """Génère du contenu à partir d'une vidéo YouTube en utilisant l'API Gemini."""
+    try:
+        # Configuration du client Gemini
+        genai.configure(api_key=GEMINI_API_KEY)
+        client = genai.Client()
+        
+        # Préparation de la requête
+        contents = types.Content(
+            parts=[
+                types.Part(text=prompt),
+                types.Part(
+                    file_data=types.FileData(file_uri=youtube_url)
+                )
+            ]
+        )
+        
+        # Génération de la réponse
+        response = client.models.generate_content(
+            model=model_name,
+            contents=contents
+        )
+        
+        return {"success": True, "text": response.text}
+    except Exception as e:
+        return {"success": False, "error": str(e)}
+
 # --- Routes ---
 
 @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:
@@ -95,7 +125,6 @@ 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')
@@ -108,59 +137,28 @@ 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 sous_categorie_id and youtube_url and prompt and 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)
                 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 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')
+                    flash(f"Erreur lors de l'ajout : {e}", 'danger')
+
 
-        # 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}')
+            matiere.nom = request.form.get(f'edit_nom_matiere_{matiere_id}')
             try:
                 db.session.commit()
                 flash(f"Matière '{matiere.nom}' modifiée avec succès.", 'success')
@@ -168,7 +166,6 @@ 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)
@@ -181,7 +178,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)
@@ -190,6 +187,7 @@ 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')
@@ -197,7 +195,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)
@@ -209,7 +207,6 @@ 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)
@@ -218,10 +215,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)
@@ -231,17 +228,31 @@ 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 des données pour l'affichage
+    # Récupération de toutes les 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()
     return render_template('gere.html', matieres=matieres, sous_categories=sous_categories, textes=textes)
 
+@app.route('/generate-content', methods=['POST'])
+def generate_content():
+    """Route API pour générer du contenu avec Gemini."""
+    data = request.json
+    youtube_url = data.get('youtube_url')
+    prompt = data.get('prompt')
+    model = data.get('model')
+    
+    if not youtube_url or not prompt or not model:
+        return jsonify({'success': False, 'error': 'Paramètres manquants'})
+    
+    result = generate_content_from_youtube(youtube_url, prompt, model)
+    return jsonify(result)
+
 if __name__ == '__main__':
     with app.app_context():
         db.create_all()  # Crée les tables si elles n'existent pas
-    app.run(debug=True)
+    app.run(debug=True)  # Port différent pour éviter les conflits