...

app/test_gradio_test.py

@@ -13,7 +13,9 @@ import random
 from app.quotes import quotes
 from app.drawings import drawings
 from app.knowledge_base.textbooks import TEXTBOOKS
-# --- Настройка ---
+
+
+# НАСТРОЙКА
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 load_dotenv()
 api_key = os.getenv("API_KEY_openai")
@@ -21,7 +23,7 @@ client = OpenAI(api_key=api_key) if api_key else None
 VS_ID = os.getenv("VECTOR_STORE_ID", "")  # vector store for file_search if needed
 
 
-###### УТИЛИТЫ
+#### УТИЛИТЫ
 def upload_to_catbox(file_bytes: bytes) -> str:
     """Загружает файл на catbox.moe и возвращает URL"""
     try:
@@ -38,29 +40,88 @@ def upload_to_catbox(file_bytes: bytes) -> str:
         raise gr.Error("Не удалось загрузить изображение")
 
 
-import subprocess
+from docx.shared import Pt
+import re
 
-def generate_docx(markdown_text: str) -> str:
-    """Конвертирует markdown-текст в .docx через pandoc и возвращает путь к файлу."""
-    tmp_dir = tempfile.gettempdir()
-    md_path = os.path.join(tmp_dir, "lesson_plan.md")
-    docx_path = os.path.join(tmp_dir, f"lesson_plan_{datetime.now().strftime('%Y%m%d%H%M%S')}.docx")
+def generate_docx(text: str) -> str:
+    """Сохраняет текст с таблицами в .docx и возвращает путь"""
+    doc = Document()
+    in_table = False
+    table_rows = []
+    header = []
+    separator = []
 
-    # Сохраняем markdown во временный .md файл
-    with open(md_path, "w", encoding="utf-8") as f:
-        f.write(markdown_text)
+    for line in text.split("\n"):
+        line = line.strip()
 
-    try:
-        # Запускаем pandoc для конвертации в docx
-        subprocess.run(
-            ["pandoc", md_path, "-o", docx_path],
-            check=True
-        )
-        return docx_path
-    except subprocess.CalledProcessError as e:
-        logging.error(f"Ошибка Pandoc: {e}")
-        raise gr.Error("Ошибка при конвертации в DOCX через Pandoc")
+        # Определяем строки таблицы
+        if re.match(r'^\|.+\|$', line):
+            if not in_table:
+                in_table = True
 
+            # Очищаем и разделяем ячейки
+            cells = [cell.strip() for cell in line[1:-1].split('|')]
+
+            # Определяем тип строки: заголовок, разделитель или данные
+            if all(re.match(r'^:?-+:?$', cell) for cell in cells):
+                separator = cells
+            elif header:
+                table_rows.append(cells)
+            else:
+                header = cells
+
+        else:
+            if in_table:
+                # Создаем таблицу когда выходим из блока таблицы
+                if header and separator and table_rows:
+                    cols = len(header)
+                    table = doc.add_table(rows=1, cols=cols)
+                    table.style = 'Table Grid'
+
+                    # Заголовок
+                    hdr_row = table.rows[0]
+                    for i, cell_text in enumerate(header):
+                        hdr_row.cells[i].text = cell_text
+                        hdr_row.cells[i].paragraphs[0].runs[0].font.bold = True
+
+                    # Данные
+                    for row in table_rows:
+                        new_row = table.add_row()
+                        for i, cell_text in enumerate(row[:cols]):  # Обрезаем лишние ячейки
+                            new_row.cells[i].text = cell_text
+
+                # Сбрасываем состояние
+                in_table = False
+                header = []
+                separator = []
+                table_rows = []
+
+            # Добавляем обычный текст
+            if line:
+                p = doc.add_paragraph(line)
+                p.style = 'BodyText'
+
+    # Обрабатываем таблицу в конце текста
+    if in_table and header and table_rows:
+        cols = len(header)
+        table = doc.add_table(rows=1, cols=cols)
+        table.style = 'Table Grid'
+
+        hdr_row = table.rows[0]
+        for i, cell_text in enumerate(header):
+            hdr_row.cells[i].text = cell_text
+            hdr_row.cells[i].paragraphs[0].runs[0].font.bold = True
+
+        for row in table_rows:
+            new_row = table.add_row()
+            for i, cell_text in enumerate(row[:cols]):
+                new_row.cells[i].text = cell_text
+
+    # Сохраняем документ
+    tmp = tempfile.gettempdir()
+    path = os.path.join(tmp, f"lesson_plan_{datetime.now().strftime('%Y%m%d%H%M%S')}.docx")
+    doc.save(path)
+    return path
 
 # названия textbooks для автозаполнения поля Учебник
 import json

requirements

@@ -2,7 +2,7 @@ streamlit
 python-dotenv
 python-docx
 openai
-
+markdown
 
 
 

requirements.txt

@@ -1,6 +1,5 @@
 python-dotenv
 python-docx
-pandoc
 openai
 gradio
 requests