fixes_v5

app.py

@@ -1,9 +1,8 @@
-import language_tool_python
-import openai
 import streamlit as st
+import language_tool_python
 import subprocess
-from pathlib import Path
 import html
+import openai
 
 is_java_installed = False
 
@@ -19,7 +18,7 @@ def install_java():
 
 @st.cache_resource
 def load_assets():
-    openai.api_key = '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'
+    openai.api_key = 'sk-proj-WY9cBkzPHS9iZq_PruWf9_t1D3BlbkFJaNjZSiuy2VBtUX6zzR6dmauyN1OB5vCrxwHv0dLmDl6bXQt5JlbyzDW7qBa7PZM-GLJpEqBqQA'
     install_java()
     tool = language_tool_python.LanguageTool('ru-RU', language_tool_download_version="6.1")
     return tool
@@ -51,36 +50,48 @@ def main():
             .error-highlight {
                 background-color: #ffe066;
                 border-bottom: 2px dotted #ff9900;
+                cursor: pointer;
                 color: black !important;
             }
+            .error-highlight:hover {
+                background-color: #ffd700;
+            }
         </style>
     """, unsafe_allow_html=True)
 
     st.title('Проверка орфографии')
     
-    text_input = st.text_area("Введите текст для проверки:", height=200)
+    # Сохраняем текст в session_state
+    text_input = st.text_area("Введите текст для проверки:", 
+                             value=st.session_state.get('text_to_check', ''),
+                             height=200)
     text = text_input.replace("\n", " ")
     tool = load_assets()
 
     if st.button('Проверить текст'):
+        st.session_state.text_to_check = text_input
         if not text.strip():
             st.warning("Введите текст для проверки")
         else:
             errors = check_text(text, tool)
-            st.session_state.errors = sorted(errors, key=lambda x: x['start'])
-            st.session_state.show_all_errors = True  # Показывать все ошибки по умолчанию
+            st.session_state.errors = errors
+            st.session_state.sorted_errors = sorted(errors, key=lambda x: x['start'])
+            st.session_state.selected_error = None  # Сброс выбранной ошибки
 
+    # Используем сохраненные ошибки при наличии
     errors = st.session_state.get('errors', [])
-    
+    sorted_errors = st.session_state.get('sorted_errors', [])
+
     if errors:
-        # Выводим текст с подсветкой ошибок
+        # Генерация HTML с кликабельными ошибками
         highlighted = []
         last_pos = 0
         
-        for error in errors:
+        for index, error in enumerate(sorted_errors):
             highlighted.append(html.escape(text[last_pos:error['start']]))
             highlighted.append(
-                f'<span class="error-highlight" title="{html.escape(error["message"])}">'
+                f'<span class="error-highlight" onclick="window.parent.postMessage({{index: {index}}}, \'*\')" '
+                f'title="Кликните для подробностей">'
                 f'{html.escape(text[error["start"]:error["end"]])}'
                 f'</span>'
             )
@@ -88,30 +99,55 @@ def main():
         
         highlighted.append(html.escape(text[last_pos:]))
         
+        html_content = f"""
+        <div style="
+            background: white;
+            padding: 20px;
+            border-radius: 8px;
+            border: 1px solid #ddd;
+            white-space: pre-wrap;
+            font-family: monospace;
+            color: #000000;
+        ">
+            {''.join(highlighted)}
+        </div>
+        """
+        
         st.markdown("### Результат проверки:")
-        st.markdown(
-            f'<div style="background: white; padding: 20px; border-radius: 8px; color: black;">'
-            f'{"".join(highlighted)}'
-            f'</div>',
-            unsafe_allow_html=True
-        )
+        st.markdown(html_content, unsafe_allow_html=True)
 
-        # Выбор ошибки через радио-кнопки
-        st.markdown("### Выберите ошибку для подробного описания:")
-        selected_error = st.radio(
-            "Список ошибок:",
-            options=[f"Ошибка {i+1} (позиция {e['start']}-{e['end']})" for i, e in enumerate(errors)],
-            index=None,
-            key="error_selector"
-        )
+        # JavaScript для обработки кликов
+        st.markdown("""
+        <script>
+        window.addEventListener('message', function(event) {
+            if (event.data.index !== undefined) {
+                const index = event.data.index;
+                // Отправляем индекс в Streamlit
+                const streamlitDoc = window.parent.document;
+                const iframe = streamlitDoc.querySelector('iframe[data-testid="stIFrame"]');
+                if (iframe) {
+                    iframe.contentWindow.postMessage({type: 'streamlit:setComponentValue', value: index}, '*');
+                }
+            }
+        });
+        </script>
+        """, unsafe_allow_html=True)
 
-        # Показываем описание выбранной ошибки
-        if selected_error:
-            error_index = int(selected_error.split()[1]) - 1  # "Ошибка 1" → индекс 0
-            error = errors[error_index]
-            st.markdown(f"**Описание:** {error['message']}")
+        # Создаем компонент для получения индекса ошибки
+        clicked_error = st.session_state.get('clicked_error', None)
+        
+        # Обработка кликов через custom component
+        if clicked_error is not None:
+            st.session_state.selected_error = clicked_error
+        
+        # Обработка выбранной ошибки
+        if 'selected_error' in st.session_state and st.session_state.selected_error is not None:
+            error = sorted_errors[st.session_state.selected_error]
+            st.markdown("### Описание ошибки:")
+            st.markdown(f"**Ошибка {st.session_state.selected_error + 1}**: {error['message']}")
         else:
-            st.info("Кликните на ошибку в списке выше, чтобы увидеть её описание.")
+            st.markdown("### Найденные ошибки:")
+            st.markdown("🔍 **Кликните на выделенный текст, чтобы увидеть описание ошибки**")
 
 if __name__ == "__main__":
-    main()
+    main()