Update app.py

app.py

@@ -1,109 +1,53 @@
 import gradio as gr
-from docling.datamodel.base_models import InputFormat
-from docling.document_converter import DocumentConverter
-import tempfile
-import io
-import re
-import os
-import language_tool_python
-import html
-tool = language_tool_python.LanguageTool('de-DE')
-bad_string = """<br>&lt;!<span style="color: red; text-decoration: underline;" title="Möglicher Tippfehler gefunden. (Vorschläge: - -image)">-- image</span> <span style="color: red; text-decoration: underline;" title="Möchten Sie einen Pfeil verwenden? (Vorschläge: →, ⇾, ⇉)">--&gt;</span><br>"""
-def check_spelling(text_input: str) -> str:
-    """Prüft den Text mit LanguageTool und markiert Fundstellen in HTML."""
-    if tool is None:
-        return "❌ **Fehler:** LanguageTool konnte nicht geladen werden."
-    text = text_input or ""
-    if not text.strip():
-        return "*Bitte lade eine Datei hoch oder füge Text ein...*"
-    if text.startswith("❌") or text.startswith("Fehler"):
-        return text
-    matches = tool.check(text)
-    if not matches:
-        return "<p style='color: green;'>✅ Keine Fehler gefunden!</p>"
-    # HTML-Aufbau für die Anzeige der Fehler
-    parts = []
-    text_len = len(text)
-    last_idx = text_len
-    for match in reversed(matches):
-        start = match.offset
-        end = match.offset + match.error_length
-        parts.append(html.escape(text[end:last_idx]))
-        word = text[start:end]
-        suggestions = ", ".join(match.replacements[:3]) if match.replacements else "keine Vorschläge"
-        error_html = (
-            f'<span style="color: red; text-decoration: underline;" '
-            f'title="{html.escape(match.message)} (Vorschläge: {html.escape(suggestions)})">'
-            f'{html.escape(word)}</span>'
-        )
-        parts.append(error_html)
-        last_idx = start
-    parts.append(html.escape(text[:last_idx]))
-    html_texte = "".join(reversed(parts)).replace("\n\n", "<br><br>")
-    html_text = html_texte.replace("\n", "<br>")
-    html_result = f"<div><strong>⚠️ {len(matches)} Fehler gefunden</strong><br><br>{html_text}</div>"
-    return html_result
-
-def process_pdf(pdf_file):
-    if pdf_file is None:
-        return None, "Bitte laden Sie ein PDF hoch."
-    try:
-        converter = DocumentConverter()
-        result = converter.convert(pdf_file.name)
-        extracted_text = result.document.export_to_markdown()
-        return extracted_text.replace(bad_string, "")
-    except Exception as e:
-        return f"Fehler bei der Verarbeitung: {str(e)}"
-
-def process_text(text_input):
-    """Verarbeitet direkt eingegebenen Text und prüft die Rechtschreibung."""
-    if not text_input or not text_input.strip():
-        return "*Bitte Text eingeben...*"
-    return check_spelling(text_input)
-
-def process_pdf_with_spelling(pdf_file):
-    """Verarbeitet PDF und führt anschließend Rechtschreibprüfung durch."""
-    extracted = process_pdf(pdf_file)
-    if extracted is None or str(extracted).startswith("Fehler"):
-        return extracted
-    #return extracted
-    return check_spelling(extracted)
-
-# Gradio Interface
-with gr.Blocks(title="Smart PDF OCR (Column-Aware)") as demo:
-    gr.Markdown("# 📑 Smart PDF Layout-Aware OCR")
-
-    with gr.Tabs():
-        # --- Tab 1: PDF Upload ---
-        with gr.TabItem("📄 PDF hochladen"):
-            with gr.Row():
-                pdf_input = gr.File(label="PDF hochladen", file_types=[".pdf"])
-
-            with gr.Row():
-                text_output = gr.Markdown(label="Extrahierter Text (Markdown Format)")
-
-            submit_btn = gr.Button("Dokument analysieren", variant="primary")
-            submit_btn.click(fn=process_pdf_with_spelling, inputs=pdf_input, outputs=text_output)
-
-        # --- Tab 2: Text direkt eingeben ---
-        with gr.TabItem("✏️ Text eingeben"):
-            with gr.Row():
-                text_input = gr.Textbox(
-                    label="Text eingeben",
-                    placeholder="Geben Sie hier Ihren deutschen Text ein...",
-                    lines=10,
-                    max_lines=30,
-                )
-
-            with gr.Row():
-                spell_output = gr.HTML(label="Rechtschreibprüfung")
-
-            with gr.Row():
-                check_btn = gr.Button("Rechtschreibung prüfen", variant="primary")
-                clear_btn = gr.Button("Leeren", variant="secondary")
-
-            check_btn.click(fn=process_text, inputs=text_input, outputs=spell_output)
-            clear_btn.click(fn=lambda: ("", ""), inputs=None, outputs=[text_input, spell_output])
-
+from spellchecker import SpellChecker
+
+# 1. Logik: Rechtschreibprüfung initialisieren (Deutsch)
+spell = SpellChecker(language='de')
+
+def process_text(text):
+    if not text or text.strip() == "":
+        return "Bitte gib einen Text ein."
+    
+    words = text.split()
+    misspelled = spell.unknown([w.strip('.,!?:;') for w in words])
+    
+    corrected_words = []
+    for word in words:
+        clean = word.strip('.,!?:;')
+        # Wenn das Wort falsch geschrieben ist, Vorschlag in Klammern zeigen
+        if clean.lower() in [m.lower() for m in misspelled]:
+            correction = spell.correction(clean)
+            corrected_words.append(f"❌ {word} ({correction}?)")
+        else:
+            corrected_words.append(word)
+            
+    return " ".join(corrected_words)
+
+# 2. UI: Gradio Interface Aufbau
+with gr.Blocks() as demo:
+    gr.Markdown("# 📝 Deutsche Rechtschreibprüfung")
+    
+    with gr.Column():
+        text_input = gr.Textbox(lines=5, label="Dein Text", placeholder="Schreibe hier etwas auf Deutsch...")
+        spell_output = gr.Textbox(label="Ergebnis / Korrekturvorschläge")
+
+        with gr.Row():
+            check_btn = gr.Button("Rechtschreibung prüfen", variant="primary")
+            clear_btn = gr.Button("Leeren", variant="secondary")
+
+    # 3. Events: Verknüpfung der Buttons mit der Logik
+    check_btn.click(
+        fn=process_text, 
+        inputs=text_input, 
+        outputs=spell_output
+    )
+    
+    clear_btn.click(
+        fn=lambda: ("", ""), 
+        inputs=None, 
+        outputs=[text_input, spell_output]
+    )
+
+# App starten
 if __name__ == "__main__":
     demo.launch()