api language tool

app.py

@@ -1,29 +1,25 @@
 import gradio as gr
 from transformers import pipeline
-import language_tool_python
-
-tool = language_tool_python.LanguageTool('pl')
+import requests
 
 def correct_polish_text(text):
-    max_iterations = 3
-    for _ in range(max_iterations):
-        matches = tool.check(text)
-        if not matches:
-            break
-        corrected_parts = []
-        last_pos = 0
-        for match in matches:
-            start = match.offset
-            end = start + match.errorLength
-            corrected_parts.append(text[last_pos:start])
-            if match.replacements:
-                corrected_parts.append(match.replacements[0])
-            else:
-                corrected_parts.append(text[start:end])
-            last_pos = end
-        corrected_parts.append(text[last_pos:])
-        text = ''.join(corrected_parts)
-    return text
+    api_url = "https://api.languagetoolplus.com/v2/check"
+    params = {
+        "text": text,
+        "language": "pl",
+    }
+    response = requests.post(api_url, data=params)
+    if response.status_code == 200:
+        matches = response.json().get("matches", [])
+        corrected_text = text
+        for match in reversed(matches):
+            start = match["offset"]
+            end = start + match["length"]
+            replacement = match["replacements"][0]["value"] if match["replacements"] else text[start:end]
+            corrected_text = corrected_text[:start] + replacement + corrected_text[end:]
+        return corrected_text
+    else:
+        return text
 
 def transcribe(audio):
     pipe = pipeline(model="marcsixtysix/whisper-base-pl")