fixes_v7

app.py

@@ -90,7 +90,7 @@ def main():
         for index, error in enumerate(sorted_errors):
             highlighted.append(html.escape(text[last_pos:error['start']]))
             highlighted.append(
-                f'<span class="error-highlight" onclick="window.parent.postMessage({{index: {index}}}, \'*\')" '
+                f'<span class="error-highlight" onclick="handleClick({index})" '
                 f'title="Кликните для подробностей">'
                 f'{html.escape(text[error["start"]:error["end"]])}'
                 f'</span>'
@@ -99,6 +99,9 @@ def main():
         
         highlighted.append(html.escape(text[last_pos:]))
         
+        # Создаем кастомный компонент
+        clicked_index = st.session_state.get("clicked_index", None)
+        
         html_content = f"""
         <div style="
             background: white;
@@ -111,43 +114,44 @@ def main():
         ">
             {''.join(highlighted)}
         </div>
+        <script>
+            function handleClick(index) {{
+                window.parent.postMessage({{
+                    type: 'streamlit:setComponentValue',
+                    value: index
+                }}, '*');
+            }}
+        </script>
         """
         
-        st.markdown("### Результат проверки:")
-        st.markdown(html_content, unsafe_allow_html=True)
+        # Рендерим через кастомный компонент
+        st.components.v1.html(html_content, height=300, key="highlighter")
+        
+        # Обрабатываем клик
+        if st.session_state.get("clicked_index") is not None:
+            selected_index = st.session_state.clicked_index
+            error = sorted_errors[selected_index]
+            st.markdown("### Описание ошибки:")
+            st.markdown(f"**Ошибка {selected_index + 1}**: {error['message']}")
+            # Сбрасываем состояние после показа
+            st.session_state.clicked_index = None
+        else:
+            st.markdown("### Найденные ошибки:")
+            st.markdown("🔍 **Кликните на выделенный текст, чтобы увидеть описание ошибки**")
 
-        # 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}, '*');
-                }
+        window.addEventListener("message", (event) => {
+            if (event.data.type === 'streamlit:setComponentValue') {
+                window.parent.postMessage({
+                    type: 'streamlit:setComponentValue',
+                    value: event.data.value
+                }, '*');
             }
         });
         </script>
         """, unsafe_allow_html=True)
 
-        # Создаем компонент для получения индекса ошибки
-        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.markdown("### Найденные ошибки:")
-            st.markdown("🔍 **Кликните на выделенный текст, чтобы увидеть описание ошибки**")
-
 if __name__ == "__main__":
     main()