docx

app.py

@@ -4,6 +4,8 @@ import streamlit as st
 import subprocess
 from pathlib import Path
 import html
+import docx
+from io import BytesIO
 
 is_java_installed = False
 prompt = """
@@ -85,16 +87,24 @@ def check_text(text, tool):
     errors = sorted(errors, key=lambda x: x['end'])
 
     shift = 0
-    for i, error in enumerate(errors):
+    for i, error in enumerate(errors, 1):
         error['start'] += shift
         error['end'] += shift
-        inp = f"({i + 1})"
+        inp = f"({i})"
         text = text[:error['end']] + inp + text[error['end']:]
         shift += len(inp)
 
     return text, errors
 
 
+def extract_text_from_docx(file):
+    doc = docx.Document(BytesIO(file.getvalue()))
+    full_text = []
+    for para in doc.paragraphs:
+        full_text.append(para.text)
+    return "\n".join(full_text)
+
+
 def main():
     st.markdown("""
         <style>
@@ -123,9 +133,27 @@ def main():
         </style>
     """, unsafe_allow_html=True)
 
-
     st.title('Проверка орфографии')
-    text = st.text_area("Введите текст для проверки:", height=200).replace("\n", " ")
+    
+    # Добавляем переключатель между режимами ввода
+    input_mode = st.radio(
+        "Выберите способ ввода текста:",
+        ("Ввести текст вручную", "Загрузить Word документ"),
+        horizontal=True
+    )
+    
+    text = ""
+    if input_mode == "Загрузить Word документ":
+        uploaded_file = st.file_uploader("Загрузите Word документ", type=['docx'])
+        if uploaded_file is not None:
+            try:
+                text = extract_text_from_docx(uploaded_file)
+                st.text_area("Текст из документа:", value=text, height=200, key="docx_text")
+            except Exception as e:
+                st.error(f"Ошибка при чтении файла: {e}")
+    else:
+        text = st.text_area("Введите текст для проверки:", height=200, key="manual_text").replace("\n", " ")
+    
     tool = load_assets()
 
     if st.button('Проверить текст'):
@@ -174,7 +202,7 @@ def main():
                 
                 st.markdown("### Найденные ошибки:")
                 for i, error in enumerate(errors, 1):
-                    st.markdown(f"{i}. **Позиция {error['start']}-{error['end']}**: {error['message']}")
+                    st.markdown(f"{i}. {error['message']}")
 
 
 if __name__ == "__main__":

requirements.txt

@@ -21,6 +21,7 @@ propcache==0.3.1
 psutil==7.0.0
 pydantic==2.11.3
 pydantic_core==2.33.1
+python-docx==0.8.11
 requests==2.32.3
 setuptools==78.1.0
 sniffio==1.3.1