Update app.py

app.py

@@ -1,73 +1,25 @@
 from transformers import pipeline
 import gradio as gr
 
-# Charger un modèle français performant pour le résumé
-summarizer = pipeline("summarization", model="moussaKam/barthez-orangesum-title")
+# Charger le modèle de résumé (BART CNN)
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 def summarize_text(text):
-    if not text or not text.strip():
-        return "Veuillez entrer un texte à résumer."
-    try:
-        # Limiter pour éviter les erreurs de longueur
-        summary = summarizer(text[:2000], max_length=120, min_length=40, do_sample=False)
-        return summary[0]["summary_text"]
-    except Exception as e:
-        return "Erreur lors du résumé. Veuillez réessayer avec un texte plus court."
-
-# Couleurs Chariow-style (tu peux les ajuster)
-PRIMARY_COLOR = "#2563eb"  # bleu moderne
-BACKGROUND_COLOR = "#f9fafb"
-TEXT_COLOR = "#111827"
-
-with gr.Blocks(
-    title="Résumeur IA – Chariow TechWorks",
-    theme=gr.themes.Soft(primary_hue="blue", neutral_hue="slate").set(
-        block_background_fill=BACKGROUND_COLOR,
-        body_background_fill=BACKGROUND_COLOR,
-        button_primary_background_fill=PRIMARY_COLOR,
-        button_primary_background_fill_hover="#1d4ed8",
-        button_primary_text_color="white",
-        block_label_text_color=TEXT_COLOR,
-        block_title_text_color=TEXT_COLOR,
-    ),
-    css="""
-    footer { display: none !important; }
-    .gradio-container { max-width: 800px; margin: auto; padding: 20px; }
-    h1 { text-align: center; font-weight: 700; margin-bottom: 8px; }
-    .subtitle { text-align: center; color: #4b5563; margin-bottom: 24px; }
-    """
-) as demo:
-    gr.Markdown(
-        
-        <div style="text-align: center; margin-bottom: 20px;">
-            <h1>📝 Résumeur IA</h1>
-            <p class="subtitle">Par <strong>Chariow TechWorks</strong> — Automatisation intelligente pour les équipes non techniques</p>
-        </div>
-    )
-    gr.Markdown("➡️ Collez un texte (email, note, article…) et obtenez un **résumé clair, court et en bon français**.")
-
-    with gr.Row():
-        with gr.Column():
-            input_text = gr.Textbox(
-                label="📄 Votre texte",
-                placeholder="Exemple : « Dans le cadre de notre transition vers des outils plus automatisés, nous souhaitons simplifier la gestion… »",
-                lines=10,
-                max_lines=15
-            )
-            with gr.Row():
-                clear_btn = gr.ClearButton(value="🗑️ Effacer", components=[input_text])
-                submit_btn = gr.Button("🚀 Résumer", variant="primary")
-
-        with gr.Column():
-            output_text = gr.Textbox(
-                label="✨ Résumé généré",
-                lines=5,
-                interactive=False,
-                elem_classes=["output-box"]
-            )
-
-    # Lier le bouton et la fonction
-    submit_btn.click(fn=summarize_text, inputs=input_text, outputs=output_text)
-    input_text.submit(fn=summarize_text, inputs=input_text, outputs=output_text)  # Entrée + Entrée = résumer
+    if len(text.strip()) == 0:
+        return "Veuillez entrer du texte à résumer."
+    # Le modèle a une limite de ~1024 tokens ; on tronque si nécessaire
+    max_input_length = 1024
+    inputs = summarizer.tokenizer(text, return_tensors="pt", truncation=True, max_length=max_input_length)
+    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+    return summary[0]['summary_text']
+
+# Interface Gradio
+with gr.Blocks(title="Résumeur IA – Chariow TechWorks") as demo:
+    gr.Markdown("## 📝 Résumeur Automatique")
+    gr.Markdown("Collez un texte long (article, email, note...) et obtenez un résumé clair et concis.")
+    input_text = gr.Textbox(label="Texte à résumer", lines=10)
+    output_text = gr.Textbox(label="Résumé", lines=5)
+    btn = gr.Button("Résumer")
+    btn.click(fn=summarize_text, inputs=input_text, outputs=output_text)
 
 demo.launch()