Update app.py

app.py

@@ -7,81 +7,113 @@ model_name = "Moustapha91/bart_large_poetique-v02"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Fonction de génération ajustée
-def generate_summary(input_text):
+# Fonction de génération
+def generate_poem(prompt, temperature, top_k, top_p, num_beams, repetition_penalty, max_length):
+    """
+    Génère un texte poétique à partir d'un prompt et de paramètres personnalisés.
+    """
+    # Tokenisation du prompt
     inputs = tokenizer(
-        input_text,
+        prompt,
         padding="max_length",
-        max_length=512,  # Ajustez si nécessaire
+        max_length=512,
         return_tensors="pt",
         truncation=True,
     )
     input_ids = inputs.input_ids.to(model.device)
     attention_mask = inputs.attention_mask.to(model.device)
 
-    # Génération avec des paramètres personnalisés
+    # Génération avec les paramètres spécifiés par l'utilisateur
     outputs = model.generate(
         input_ids=input_ids,
         attention_mask=attention_mask,
-        max_length=512,
-        num_beams=4,
-        temperature=1.0,
+        max_length=max_length,
+        temperature=temperature,
+        top_k=top_k,
+        top_p=top_p,
+        num_beams=num_beams,
+        repetition_penalty=repetition_penalty,
         early_stopping=True,
-        repetition_penalty=6.0,
-        no_repeat_ngram_size=5,
+        no_repeat_ngram_size=3,  # Empêche les répétitions courtes
     )
-    output_str = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return output_str
-
-# Liste d'exemples enrichis
-examples = [
-    "L'amour, cet élan mystérieux qui unit les âmes, est une lumière douce dans les ténèbres du quotidien.",
-    "Les tirailleurs sénégalais, ces soldats venus d'Afrique, ont marqué l'histoire par leur courage et leur sacrifice. Recrutés à partir de 1857 sous l'autorité coloniale française, ils provenaient de nombreux pays d'Afrique de l'Ouest.",
-    "Il était une fois un garçon plein de rêves et d'espoir. Porté par une vision audacieuse, il rêvait d'unifier toute l'Afrique sous un même idéal de paix et de solidarité.",
-    "L'eau, source de vie, est l'élément essentiel qui relie tous les êtres vivants. Elle coule dans les rivières, traverse les montagnes et nourrit la terre.",
-    "Dans un petit village au cœur de l'Afrique, une vieille femme racontait les légendes des ancêtres à la lumière des étoiles, transmettant ainsi leur sagesse aux générations futures."
-]
-
-# Personnalisation de la description et de l'interface
-title = "📝 Générateur de Texte Poétique et Résumé"
-description = """
-Bienvenue dans notre application interactive de génération de texte poétique ou résumé ! 🖋️  
-Utilisez ce modèle pré-entraîné pour transformer des idées en textes élégants et expressifs.  
-Ajoutez un paragraphe dans la zone ci-dessous et voyez la magie opérer !
-"""
-about = """
-### À propos
-Ce projet utilise le modèle **BART** de Facebook, connu pour ses capacités avancées en résumé et génération de texte.  
-Créé pour inspirer, éduquer et honorer l'héritage de la littérature et des récits africains.  
-Vous pouvez tester avec les exemples ci-dessus ou entrer vos propres idées.
+    # Décodage du texte généré
+    output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return output_text
 
----
-
-📚 **Exemples suggérés :**
-- Un texte sur l'amour.  
-- Une introduction sur les tirailleurs sénégalais.  
-- Un récit d'unité et d'espoir en Afrique.
-"""
+# Interface Gradio
+def gradio_interface():
+    # Définir les composants Gradio
+    title = "✨ Générateur de Texte Poétique"
+    description = (
+        "Cette interface interactive vous permet de générer des textes poétiques "
+        "en entrant un 'prompt' et en ajustant les paramètres de génération."
+    )
+    examples = [
+        # Chaque exemple correspond à [prompt, temperature, top_k, top_p, num_beams, repetition_penalty, max_length]
+        ["Dans la clarté d'une nuit étoilée, un fleuve murmure des secrets oubliés.", 1.0, 50, 0.9, 4, 1.5, 150],
+        ["L'amour est un feu doux, éclairant les âmes perdues dans l'ombre.", 1.2, 40, 0.85, 5, 2.0, 100],
+        ["Un enfant rêve d'unifier un continent par des chants de liberté.", 0.8, 30, 0.95, 6, 1.8, 200],
+    ]
 
-# Interface Gradio améliorée
-interface = gr.Interface(
-    fn=generate_summary,
-    inputs=gr.Textbox(
-        lines=10,
-        placeholder="Entrez un texte à transformer en poème ou résumé ici...",
-        label="🖊️ Texte à transformer"
-    ),
-    outputs=gr.Textbox(
-        lines=10,
-        label="✨ Texte généré"
-    ),
-    examples=examples,
-    title=title,
-    description=description,
-    theme="compact",  # Thème compact et moderne
-    article=about,  # Section 'À propos'
-    allow_flagging="never",  # Désactive le flagging si inutile
-)
+    interface = gr.Interface(
+        fn=generate_poem,
+        inputs=[
+            gr.Textbox(
+                label="Prompt",
+                placeholder="Entrez votre prompt ici...",
+                lines=2,
+            ),
+            gr.Slider(
+                label="Température (Créativité)",
+                minimum=0.5,
+                maximum=1.5,
+                value=1.0,
+                step=0.1,
+            ),
+            gr.Slider(
+                label="Top-k (Mots possibles)",
+                minimum=0,
+                maximum=100,
+                value=50,
+                step=1,
+            ),
+            gr.Slider(
+                label="Top-p (Nucleus Sampling)",
+                minimum=0.5,
+                maximum=1.0,
+                value=0.9,
+                step=0.1,
+            ),
+            gr.Slider(
+                label="Nombre de faisceaux (Beam Search)",
+                minimum=1,
+                maximum=10,
+                value=4,
+                step=1,
+            ),
+            gr.Slider(
+                label="Pénalité de répétition",
+                minimum=1.0,
+                maximum=5.0,
+                value=1.5,
+                step=0.1,
+            ),
+            gr.Slider(
+                label="Longueur maximale",
+                minimum=50,
+                maximum=512,
+                value=150,
+                step=10,
+            ),
+        ],
+        outputs=gr.Textbox(label="Texte Poétique Généré", lines=10),
+        title=title,
+        description=description,
+        examples=examples,
+    )
+    return interface
 
-# Lancer l'application
-interface.launch()
+# Lancer l'application Gradio
+if __name__ == "__main__":
+    interface = gradio_interface()
+    interface.launch()