Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# Modelle vorbereiten
+# Modelle laden
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 headline_model = pipeline("text2text-generation", model="google/flan-t5-base")
 
@@ -10,19 +10,28 @@ def simplify_and_structure(text):
         return "❗ Bitte gib einen längeren Text ein."
 
     try:
-        # 1. Zusammenfassung erzeugen
+        # Zusammenfassung erzeugen
         summary = summarizer(text, max_length=150, min_length=40, do_sample=False)[0]['summary_text']
 
-        # 2. Überschrift generieren mit Prompt
+        # Überschrift generieren
         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
+        # Strukturierte Ausgabe
         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:\n{str(e)}"
 
-# Gradio UI
+# Gradio App
 with gr.Blocks() as demo:
-    gr.Markdown("## 📚 Vereinfachung mit Überschrift – Für Mittelschüler erklärt")
+    gr.Markdown("## 📚 Textvereinfachung & Überschrift (Mittelschulniveau)")
+
+    input_text = gr.Textbox(label="📝 Ursprünglicher Text", lines=12)
+    output_markdown = gr.Textbox(label="🧠 Ausgabe", lines=12, interactive=False)
+
+    run_button = gr.Button("Vereinfachen & Überschrift erzeugen")
+
+    run_button.click(fn=simplify_and_structure, inputs=input_text, outputs=output_markdown)
+
+demo.launch()