from flask import Flask, render_template, request, redirect, url_for, flash from flask_sqlalchemy import SQLAlchemy from sqlalchemy import func from bleach import clean from bs4 import BeautifulSoup import os 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'] = 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 --- class Matiere(db.Model): """Modèle représentant une matière.""" __tablename__ = 'matiere' id = db.Column(db.Integer, primary_key=True) nom = db.Column(db.String(64), unique=True, nullable=False) sous_categories = db.relationship('SousCategorie', backref='matiere', lazy='dynamic', cascade="all, delete-orphan") def __repr__(self): return f"" class SousCategorie(db.Model): """Modèle représentant une sous-catégorie.""" __tablename__ = 'sous_categorie' id = db.Column(db.Integer, primary_key=True) 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'),) def __repr__(self): return f"" class Texte(db.Model): """Modèle représentant un texte.""" __tablename__ = 'texte' id = db.Column(db.Integer, primary_key=True) titre = db.Column(db.String(128), nullable=False) contenu = db.Column(db.Text, nullable=False) sous_categorie_id = db.Column(db.Integer, db.ForeignKey('sous_categorie.id'), nullable=False) def __repr__(self): return f"" # --- Fonctions utilitaires --- def sanitize_html(html_content): """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', 'pre', 'span', 'strong', 'table', 'tbody', 'td', 'th', 'thead', 'tr', 'ul' ] allowed_attributes = { '*': ['class', 'id', 'style'], 'a': ['href', 'rel', 'target', 'title'], 'img': ['alt', 'src', 'width', 'height'], 'table': ['border', 'cellpadding', 'cellspacing'] } cleaned_html = clean(html_content, tags=allowed_tags, attributes=allowed_attributes, strip=True) soup = BeautifulSoup(cleaned_html, 'html.parser') for tag in soup.find_all(): if not tag.contents and tag.name not in ['br', 'hr', 'img']: tag.decompose() return str(soup) # --- Routes --- @app.route('/', methods=['GET', 'POST']) def gere(): if request.method == 'POST': action = request.form.get('action') # Ajouter une matière if action == 'add_matiere': nom_matiere = request.form.get('nom_matiere') if nom_matiere: nouvelle_matiere = Matiere(nom=nom_matiere) db.session.add(nouvelle_matiere) try: db.session.commit() flash(f"Matière '{nom_matiere}' ajoutée avec succès.", 'success') except Exception as e: 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') if nom_sous_categorie and matiere_id: nouvelle_sous_categorie = SousCategorie(nom=nom_sous_categorie, matiere_id=matiere_id) db.session.add(nouvelle_sous_categorie) try: db.session.commit() flash(f"Sous-catégorie '{nom_sous_categorie}' ajoutée avec succès.", 'success') 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') 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: 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') # 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}') try: db.session.commit() flash(f"Matière '{matiere.nom}' modifiée avec succès.", 'success') except Exception as e: 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) sous_categorie.nom = request.form.get(f'edit_nom_sous_categorie_{sous_categorie_id}') sous_categorie.matiere_id = request.form.get(f'edit_matiere_id_{sous_categorie_id}') try: db.session.commit() flash(f"Sous-catégorie '{sous_categorie.nom}' modifiée avec succès.", 'success') except Exception as e: 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) texte.titre = request.form.get(f'edit_titre_texte_{texte_id}') contenu = request.form.get(f'edit_contenu_texte_{texte_id}') 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') except Exception as e: 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) db.session.delete(matiere) try: db.session.commit() flash(f"Matière '{matiere.nom}' supprimée avec succès.", 'success') except Exception as e: 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) db.session.delete(sous_categorie) try: 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') # Supprimer un texte elif action.startswith('delete_texte_'): texte_id = int(action.split('_')[-1]) texte = Texte.query.get_or_404(texte_id) db.session.delete(texte) try: db.session.commit() flash(f"Texte '{texte.titre}' supprimé avec succès.", 'success') except Exception as e: db.session.rollback() flash(f"Erreur : {e}", 'danger') return redirect(url_for('gere')) # 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() return render_template('gere.html', matieres=matieres, sous_categories=sous_categories, textes=textes) if __name__ == '__main__': with app.app_context(): db.create_all() # Crée les tables si elles n'existent pas app.run(debug=True)