adding_chat_gptv1

app.py

@@ -1,4 +1,5 @@
 import language_tool_python
+import openai
 import streamlit as st
 import subprocess
 from pathlib import Path
@@ -22,12 +23,23 @@ def install_java():
 
 @st.cache_resource
 def load_assets():
+    openai.api_key = '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'
     install_java()
     tool = language_tool_python.LanguageTool('ru-RU',
                                          language_tool_download_version="6.1")
     return tool
 
 
+def generate_gpt_comment(message, context):
+    response = openai.ChatCompletion.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "user", "content": f"Прокомментируй ошибку: {message} в данном тексте '{context}'. Не больше одного предложения, на русском."}
+        ]
+    )
+    return response.choices[0].message['content']
+
+
 def check_text(text, tool):
     matches = tool.check(text)
     errors = []
@@ -35,8 +47,7 @@ def check_text(text, tool):
         error_info = {
             'start': match.offset,
             'end': match.offset + match.errorLength,
-            'message': match.message,
-            'context': match.context
+            'message': generate_gpt_comment(match.message, match.context),
         }
         errors.append(error_info)
     return errors