...

app/main_WS_FS_pure_test.py

@@ -604,51 +604,34 @@ with gr.Blocks(theme=theme, css_paths=css_path) as demo:
             error_para.add_run(html.unescape(error_text))
 
 
-    def generate_docx(html_content: str) -> str:
-        # Предварительная обработка
-        html_content = re.sub(r"\*\*(.+?)\*\*", r"<p><strong>\1</strong></p>", html_content)
-        soup = BeautifulSoup(html_content, 'html.parser')
-
-        # Создаем документ
+    def generate_docx(html: str) -> str:
+        # Упрощаем HTML перед обработкой
+        html = re.sub(r'class="[^"]+"', '', html)  # Удаляем все классы
+        html = re.sub(r'<br/?>', '\n', html)  # Заменяем <br> на переносы
+        html = re.sub(r'<strong>(.*?)</strong>', r'\1', html)  # Убираем bold
+
+        # Удаляем проблемные атрибуты
+        soup = BeautifulSoup(html, 'html.parser')
+        for tag in soup.find_all(True):
+            tag.attrs = {}
+
+        # Конвертируем
         doc = Document()
-        section = doc.sections[0]
-        section.orientation = WD_ORIENT.LANDSCAPE
-        section.page_width, section.page_height = section.page_height, section.page_width
-
-        # Инициализация конвертера
-        converter = HtmlToDocx()
-        converter.table_style = "Table Grid"
-
-        # Обработка основных блоков
-        header = soup.find('section', class_='header-block')
-        tables = soup.find_all('table')
-        materials = soup.find('div', class_='reference-block')
-
-        # Конвертация шапки
-        if header:
-            safe_convert(converter, header, doc, "в шапке")
-
-        # Конвертация таблиц
-        for table in tables:
-            # Упрощаем структуру таблицы
-            for tr in table.find_all('tr'):
-                for td in tr.find_all('td'):
-                    td.attrs = {}
-                    td.name = 'td'  # Принудительно устанавливаем тег
-
-            safe_convert(converter, table, doc, "в таблице")
-
-        # Конвертация дополнительных материалов
-        if materials:
-            safe_convert(converter, materials, doc, "в материалах")
-
-        # Сохранение файла
-        temp_dir = tempfile.gettempdir()
-        filename = f"lesson_plan_{datetime.now():%Y%m%d%H%M%S}.docx"
-        filepath = os.path.join(temp_dir, filename)
-        doc.save(filepath)
-
-        return filepath
+        parser = HtmlToDocx()
+
+        try:
+            parser.add_html_to_document(str(soup), doc)
+        except Exception as e:
+            # При ошибке вставляем сырой текст
+            doc.add_paragraph("ОШИБКА КОНВЕРТАЦИИ:")
+            doc.add_paragraph(soup.get_text())
+
+        # Сохраняем файл
+        tmp = tempfile.gettempdir()
+        fname = f"plan_{datetime.now():%Y%m%d%H%M%S}.docx"
+        path = os.path.join(tmp, fname)
+        doc.save(path)
+        return path
 
 
     ################### СТРИМИНГ