...

app/main_WS_FS_pure_test.py

@@ -55,7 +55,6 @@ def generate_presigned_url(bucket_name, object_key, expiration=3600):
         return None
 
 
-from bs4 import BeautifulSoup
 
 def force_links_new_tab(html: str) -> str:
     """
@@ -575,52 +574,79 @@ with gr.Blocks(theme=theme, css_paths=css_path) as demo:
     from htmldocx import HtmlToDocx
     from docx import Document
     from docx.enum.section import WD_ORIENT
+    import traceback
+
+
+    def safe_convert(parser, html_part, doc, section_name):
+        try:
+            parser.add_html_to_document(str(html_part), doc)
+        except Exception as e:
+            print(f"Ошибка конвертации {section_name}: {str(e)}")
+            error_msg = doc.add_paragraph()
+            error_msg.add_run(f"[ОШИБКА КОНВЕРТАЦИИ {section_name.upper()}]: ").bold = True
+            error_msg.add_run(str(html_part))
 
 
     def preprocess_html(html: str) -> str:
-        # Переводим **ЗАГОЛОВОК** в <h3>ЗАГОЛОВОК</h3>
         return re.sub(r"\*\*(.+?)\*\*", r"<h3>\1</h3>", html)
 
 
     def generate_docx(html: str) -> str:
-        # 0. Убираем Markdown‑заголовки
+        # Предварительная обработка HTML
         html = preprocess_html(html)
-
-        # 1. Оборачиваем и парсим
         soup = BeautifulSoup(f"<body>{html}</body>", "html.parser")
 
-        # 2. (Ваша существующая обработка <span.cell-*> и стилей — без изменений)
+        # Обработка стилей
         for span in soup.select("span.cell-label"):
             span["style"] = "display:block;font-size:9pt;margin-bottom:0.2em"
         for span in soup.select("span.cell-value"):
             span["style"] = "display:block;font-size:12pt"
 
-        # 3. Заголовки h1–h6 → <p style=...>
-        import re as _re
-        for h in soup.find_all(_re.compile(r"^h[1-6]$")):
-            p = soup.new_tag("p")
-            p["style"] = "font-size:12pt;font-weight:bold;margin-bottom:0.2em"
-            p.string = h.get_text()
-            h.replace_with(p)
-
-        clean_html = str(soup)
-
-        # 4. Создаём документ в альбомной ориентации
+        # Создаем базовый документ
         doc = Document()
         sec = doc.sections[-1]
         sec.orientation = WD_ORIENT.LANDSCAPE
         sec.page_width, sec.page_height = sec.page_height, sec.page_width
 
-        # 5. Конвертируем HTML → DOCX
+        # Инициализация парсера
         parser = HtmlToDocx()
         parser.table_style = "TableGrid"
-        parser.add_html_to_document(clean_html, doc)
 
-        # 6. Сохраняем и возвращаем путь
+        # Извлекаем и конвертируем основные блоки
+        header = soup.find('section', class_='header-block')
+        plan_tables = soup.find_all('table', {'border': '1'})
+        materials = soup.find('div', class_='reference-block')
+
+        # Конвертация шапки
+        if header:
+            safe_convert(parser, header, doc, "шапки")
+        else:
+            doc.add_paragraph("Отсутствует блок шапки")
+
+        # Конвертация этапов плана
+        if plan_tables:
+            for table in plan_tables:
+                safe_convert(parser, table, doc, "таблицы плана")
+        else:
+            doc.add_paragraph("Отсутствуют таблицы плана")
+
+        # Конвертация дополнительных материалов
+        if materials:
+            safe_convert(parser, materials, doc, "дополнительных материалов")
+        else:
+            doc.add_paragraph("Отсутствуют дополнительные материалы")
+
+        # Сохранение документа
         tmp = tempfile.gettempdir()
         fname = f"lesson_plan_{datetime.now():%Y%m%d%H%M%S}.docx"
         path = os.path.join(tmp, fname)
-        doc.save(path)
+
+        try:
+            doc.save(path)
+        except Exception as e:
+            print(f"Критическая ошибка при сохранении: {str(e)}")
+            raise
+
         return path