new key and other

app.py

@@ -66,7 +66,7 @@ def install_java():
 
 @st.cache_resource
 def load_assets():
-    openai.api_key = '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'
+    openai.api_key = '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'
     install_java()
     tool = language_tool_python.LanguageTool('ru-RU',
                                          language_tool_download_version="6.1")

search_errors_logic.py

@@ -4,14 +4,15 @@ import difflib
 
 prompt_fix_text_gpt = """
 Исправь ошибки в данном тексте. Жаргонизмы считаются ошибками, приемлим только официальный стиль.
+Обязательно проверяй, не заменена ли где-то буква "ё" на букву "е". Это считается грубой ошибкой.
 Верни только текст, никаких комментариев от себя не оставляй
 
 ### Пример 1:
 input text:
-досвиданя кошак
+досвиданя кошак, это твое
 
 Твой ответ:
-До свидания, кошка
+До свидания, кошка, это твоё
 
 
 Теперь твоя очередь:
@@ -111,8 +112,8 @@ def get_gpt_response_openai(inp):
 
 def get_gpt_response_vsegpt(inp):
     client = OpenAI(
-        api_key='sk-or-vv-44ca318747a45e810149bf769e9dfbe5f42046695875efd7c9121cca590d6906',
-        base_url='https://api.vsegpt.ru/v1'
+        api_key='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',
+        base_url='https://openrouter.ai/api/v1'
     )
     response = client.chat.completions.create(
         model="gpt-4o-mini",
@@ -160,7 +161,7 @@ def get_piece_of_text_bounds(s, start, end):
     return start, end
 
 
-def add_comments_to_text(text, errors):
+def add_comments_to_text(text, errors, add_errors=False):
     errors = sorted(errors, key=lambda x: x['end'])
 
     shift = 0
@@ -168,21 +169,24 @@ def add_comments_to_text(text, errors):
         error['start'] += shift
         error['end'] += shift
         inp = f"({i})"
+        if add_errors:
+            inp = inp[:-1] + ' - ' + error['message'] + ')'
         text = text[:error['end']] + inp + text[error['end']:]
         shift += len(inp)
 
     return text.replace("\n", " ")
 
 
-def check_text(text, tool, mode):
+def check_text(text, tool, mode="chat_gpt", add_errors=False):
     if mode == "tool":
-        return check_text_with_tool(text, tool)
+        return check_text_with_tool(text, tool, add_errors=add_errors)
     else:
-        return check_text_chat_gpt(text)
+        return check_text_chat_gpt(text, add_errors=add_errors)
 
 
-def check_text_chat_gpt(text, *args, **kwargs):
-    fixed_text = get_gpt_response(prompt_fix_text_gpt.format(text), "vsegpt")
+def check_text_chat_gpt(text, fixed_text=None, add_errors=False, *args, **kwargs):
+    if fixed_text is None:
+        fixed_text = get_gpt_response(prompt_fix_text_gpt.format(text), "vsegpt")
     changes = find_corrected_positions(text, fixed_text)
     errors = []
     for change in changes:
@@ -195,11 +199,11 @@ def check_text_chat_gpt(text, *args, **kwargs):
             'message': get_gpt_response(inp, client_name="vsegpt"),
         })
 
-    text_with_comments = add_comments_to_text(text, errors)
+    text_with_comments = add_comments_to_text(text, errors, add_errors=add_errors)
     return text_with_comments, errors
 
 
-def check_text_with_tool(text, tool):
+def check_text_with_tool(text, tool, add_errors=False):
     matches = tool.check(text)
     errors = []
     for match in matches:
@@ -212,5 +216,5 @@ def check_text_with_tool(text, tool):
         }
         errors.append(error_info)
     
-    text_with_comments = add_comments_to_text(text, errors)
+    text_with_comments = add_comments_to_text(text, errors, add_errors=add_errors)
     return text_with_comments, errors