new_logic

app.py

@@ -2,7 +2,7 @@ import language_tool_python
 import openai
 import streamlit as st
 import subprocess
-from search_errors_logic import check_text, check_text_chat_gpt
+from search_errors_logic import check_text
 import html
 import docx
 from io import BytesIO
@@ -136,7 +136,7 @@ def main():
         if not text.strip():
             st.warning("Введите текст для проверки")
         else:
-            text, errors = check_text_chat_gpt(text, tool)
+            text, errors = check_text(text, tool, mode="chat_gpt")
             if not errors:
                 st.success("Ошибок не найдено! 👍")
             else:

search_errors_logic.py

@@ -174,6 +174,13 @@ def add_comments_to_text(text, errors):
     return text.replace("\n", " ")
 
 
+def check_text(text, tool, mode):
+    if mode == "tool":
+        return check_text_with_tool(text, tool)
+    else:
+        return check_text_chat_gpt(text)
+
+
 def check_text_chat_gpt(text, *args, **kwargs):
     fixed_text = get_gpt_response(prompt_fix_text_gpt.format(text), "vsegpt")
     changes = find_corrected_positions(text, fixed_text)
@@ -191,7 +198,8 @@ def check_text_chat_gpt(text, *args, **kwargs):
     text_with_comments = add_comments_to_text(text, errors)
     return text_with_comments, errors
 
-def check_text(text, tool):
+
+def check_text_with_tool(text, tool):
     matches = tool.check(text)
     errors = []
     for match in matches: