Update app.py

app.py

@@ -1,34 +1,28 @@
 import gradio as gr
 from transformers import pipeline
 
-# Zusammenfassungsmodell laden
+# Modelle vorbereiten
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+headline_model = pipeline("text2text-generation", model="google/flan-t5-base")
 
-def simplify_text(text):
-    if not text or len(text.strip()) < 10:
-        return "Bitte gib einen längeren Text ein.", ""
+def simplify_and_structure(text):
+    if not text.strip() or len(text.strip()) < 30:
+        return "❗ Bitte gib einen längeren Text ein."
 
     try:
-        # Text vereinfachen
-        summary = summarizer(text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
-        
-        # Überschrift: erste Aussage als Platzhalter
-        headline = summary.split(".")[0].strip()
-        
-        return summary.strip(), headline
+        # 1. Zusammenfassung erzeugen
+        summary = summarizer(text, max_length=150, min_length=40, do_sample=False)[0]['summary_text']
+
+        # 2. Überschrift generieren mit Prompt
+        headline_prompt = f"Schreibe eine passende Überschrift zu diesem Text:\n{text}"
+        headline = headline_model(headline_prompt, max_length=20, do_sample=False)[0]['generated_text']
+
+        # 3. Strukturierter Output
+        output = f"## 🏷 {headline.strip()}\n\n{summary.strip()}"
+        return output
     except Exception as e:
-        return f"⚠️ Fehler bei der Verarbeitung:\n{str(e)}", ""
+        return f"⚠️ Fehler bei der Verarbeitung:\n{str(e)}"
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 📚 Vereinfachung & automatische Überschrift")
-
-    input_text = gr.Textbox(label="📝 Ursprünglicher Text", lines=10, placeholder="Füge hier deinen komplexen Text ein…")
-    output_summary = gr.Textbox(label="✏️ Vereinfachte Version", lines=6)
-    output_headline = gr.Textbox(label="🏷 Überschrift", lines=1)
-
-    simplify_button = gr.Button("Jetzt vereinfachen")
-
-    simplify_button.click(fn=simplify_text, inputs=input_text, outputs=[output_summary, output_headline])
-
-demo.launch()
+    gr.Markdown("## 📚 Vereinfachung mit Überschrift – Für Mittelschüler erklärt")