Upload 2 files

converter.py

@@ -1,5 +1,8 @@
 import re
 import json
+import io
+import requests
+import base64
 from pathlib import Path
 from typing import List, Dict, Any, Optional, Tuple
 import markdown
@@ -203,7 +206,10 @@ class MarkdownToDocxConverter:
                         language = cls.replace('language-', '')
                         break
                 code_text = code_element.get_text()
-                self.add_code_block(code_text, language)
+                if language == 'mermaid':
+                    self.add_mermaid_diagram(code_text)
+                else:
+                    self.add_code_block(code_text, language)
             else:
                 self.add_code_block(element.get_text())
         elif element.name == 'code' and parent_paragraph:
@@ -241,6 +247,32 @@ class MarkdownToDocxConverter:
             for child in element.children:
                 self.process_html_element(child, parent_paragraph)
                 
+    def add_mermaid_diagram(self, code: str):
+        """Рендеринг и вставка диаграммы Mermaid"""
+        try:
+            # Кодируем код диаграммы в base64
+            graphbytes = code.encode("ascii")
+            base64_bytes = base64.b64encode(graphbytes)
+            base64_string = base64_bytes.decode("ascii")
+            
+            # Формируем URL для запроса
+            url = f"https://mermaid.ink/img/{base64_string}"
+            
+            # Выполняем запрос и получаем изображение
+            response = requests.get(url)
+            response.raise_for_status()  # Проверка на ошибки HTTP
+            
+            # Вставляем изображение в документ
+            image_stream = io.BytesIO(response.content)
+            self.doc.add_picture(image_stream, width=Inches(6))
+            
+        except requests.exceptions.RequestException as e:
+            print(f"Ошибка при рендеринге диаграммы Mermaid: {e}")
+            self.doc.add_paragraph(f"[Ошибка рендеринга диаграммы: {e}]", style='Body Text')
+        except Exception as e:
+            print(f"Неожиданная ошибка при обработке Mermaid: {e}")
+            self.doc.add_paragraph(f"[Неожиданная ошибка: {e}]", style='Body Text')
+
     def add_table(self, table_element):
         """Добавление таблицы в документ"""
         rows = table_element.find_all('tr')

requirements.txt

@@ -1,4 +1,4 @@
-python-docx
-gradio
 markdown
-beautifulsoup4
+python-docx
+beautifulsoup4
+requests