fix_v1

app.py

@@ -9,10 +9,14 @@ from io import BytesIO
 
 is_java_installed = False
 prompt = """
-Тебе дан текст и сообщение об ошибке. Твоя задача - кратко описать суть ошибки и, если необходимо, исправить её. 
+Ты должен писать комментарии об ошибках в тексте.
+Тебе дан кусок текст, в котором есть ошибка, контекст, в котором стоит это слово и сообщение об ошибке. Твоя задача - кратко описать суть ошибки и, если необходимо, исправить её. 
 Исправляй только ту ошибку, на которую указывает сообщение. Отвечай на русском языке.
 
 ### Пример 1:
+Кусок текста:
+"кросивую"
+
 Сообщение об ошибке: 
 "Возможно найдена орфографическая ошибка."
 
@@ -23,6 +27,9 @@ prompt = """
 Орфографическая ошибка в слове "кросивую" - правильно "красивую".
 
 ### Пример 2:
+Кусок текста:
+"  "
+
 Сообщение об ошибке: 
 "Повтор пробела."
 
@@ -33,11 +40,14 @@ prompt = """
 Обнаружен повтор пробела между словами.
 
 Теперь твоя очередь:
+Кусок текста:
+"{}"
+
 Сообщение об ошибке: 
-{}
+"{}"
 
 Текст: 
-{}
+"{}"
 """
 
 
@@ -63,11 +73,11 @@ def load_assets():
     return tool
 
 
-def generate_gpt_comment(message, context):
+def generate_gpt_comment(piece_or_text, message, context):
     response = openai.ChatCompletion.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "user", "content": prompt.format(message, context)}
+            {"role": "user", "content": prompt.format(piece_or_text, message, context)}
         ]
     )
     return response.choices[0].message['content']
@@ -80,7 +90,7 @@ def check_text(text, tool):
         error_info = {
             'start': match.offset,
             'end': match.offset + match.errorLength,
-            'message': generate_gpt_comment(match.message, match.context),
+            'message': generate_gpt_comment(text[match.offset:match.offset + match.errorLength], match.message, match.context),
         }
         errors.append(error_info)
     
@@ -94,7 +104,7 @@ def check_text(text, tool):
         text = text[:error['end']] + inp + text[error['end']:]
         shift += len(inp)
 
-    return text, errors
+    return text.replace("\n", " "), errors
 
 
 def extract_text_from_docx(file):
@@ -152,7 +162,7 @@ def main():
             except Exception as e:
                 st.error(f"Ошибка при чтении файла: {e}")
     else:
-        text = st.text_area("Введите текст для проверки:", height=200, key="manual_text").replace("\n", " ")
+        text = st.text_area("Введите текст для проверки:", height=200, key="manual_text")
     
     tool = load_assets()