Update app.py

app.py

@@ -8,6 +8,9 @@ from datetime import datetime
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import A4
 from reportlab.lib import colors
+# === ИМПОРТЫ ДЛЯ ШРИФТОВ ===
+from reportlab.pdfbase import pdfmetrics
+from reportlab.pdfbase.ttfonts import TTFont
 
 # ========== АВТОРИЗАЦИЯ ==========
 AUTH_USERNAME = "admin"
@@ -25,6 +28,21 @@ class KDChecker:
         self.excel_db = pd.DataFrame()
         self.cabinet_list = []
         self.known_docs = ["Э3", "В4", "ПЭ3", "ВО", "ТЭ5", "СБ", "С5", "ОЛ", "Э1", "Э4", "Э7", "Д3", "Э6"]
+        
+        # === РЕГИСТРАЦИЯ ШРИФТА (Исправление квадратов) ===
+        self.font_name = 'Helvetica' # Запасной вариант
+        try:
+            # Путь к шрифту относительно папки, где лежит скрипт
+            font_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "arial.ttf")
+            
+            if os.path.exists(font_path):
+                pdfmetrics.registerFont(TTFont('Arial', font_path))
+                self.font_name = 'Arial'
+                print(f"✅ Шрифт загружен: {font_path}")
+            else:
+                print(f"⚠️ Файл шрифта не найден: {font_path}")
+        except Exception as e:
+            print(f"❌ Ошибка загрузки шрифта: {e}")
 
     def load_excel_db(self, excel_file):
         if excel_file is None:
@@ -230,7 +248,9 @@ class KDChecker:
         try:
             pdf_path = self.create_pdf(title, checklist)
         except Exception as e:
-            return f"❌ Ошибка PDF: {e}", None
+            import traceback
+            traceback.print_exc() # Пишем в консоль для отладки
+            return f"❌ Ошибка генерации PDF: {e}", None
 
         total = sum(len(v) for v in checklist.values())
         method_name = {"manual": "Ручной", "number": "По номеру", "name": "По названию"}.get(method, "?")
@@ -241,14 +261,20 @@ class KDChecker:
         path = os.path.join(tempfile.gettempdir(), "CheckList.pdf")
         c = canvas.Canvas(path, pagesize=A4)
         w, h = A4
-        font = 'Helvetica'
+        
+        # Используем наш зарегистрированный шрифт
+        font = self.font_name 
 
         y = h - 50
         c.setFont(font, 16)
         c.drawString(50, y, "CHECK-LIST")
         y -= 25
         c.setFont(font, 12)
-        c.drawString(50, y, f"Cabinet: {cabinet[:50]}")
+        
+        # Обрезаем очень длинные названия шкафов для заголовка
+        cab_title = cabinet[:60] + "..." if len(cabinet) > 60 else cabinet
+        c.drawString(50, y, f"Cabinet: {cab_title}")
+        
         c.drawString(400, y, datetime.now().strftime('%d.%m.%Y'))
         y -= 20
         c.line(50, y, w - 50, y)
@@ -263,6 +289,9 @@ class KDChecker:
             if y < 100:
                 c.showPage()
                 y = h - 50
+                # Важно: на новой странице тоже нужно ставить шрифт
+                c.setFont(font, 12) 
+                
             c.setFillColor(colors.darkblue)
             c.setFont(font, 11)
             c.drawString(50, y, f"File: {fname}")
@@ -275,12 +304,16 @@ class KDChecker:
                     c.showPage()
                     y = h - 50
                     c.setFont(font, 10)
+                
+                # Рисуем квадратик чекбокса
                 c.rect(50, y - 2, 8, 8, stroke=1, fill=0)
                 
+                # Перенос строк
                 words = task.replace('\n', ' ').split()
                 lines, line = [], ""
                 for word in words:
-                    if len(line) + len(word) < 85:
+                    # Примерная ширина строки. Arial чуть шире Helvetica, запас 80
+                    if len(line) + len(word) < 80: 
                         line += word + " "
                     else:
                         lines.append(line.strip())
@@ -305,25 +338,39 @@ class KDChecker:
 # ========== ИНТЕРФЕЙС ==========
 checker = KDChecker()
 
-with gr.Blocks(title="КД Checker", theme=gr.themes.Soft()) as app:
+# Используем CSS для улучшения вида
+css = """
+.file-upload { height: 100px !important; min-height: 100px !important; }
+"""
+
+with gr.Blocks(title="КД Checker", theme=gr.themes.Soft(), css=css) as app:
     gr.Markdown("# ✅ Генератор чек-листов КД")
     
     with gr.Row():
-        with gr.Column():
-            gr.Markdown("### 📁 База знаний")
-            db_input = gr.File(label="Excel (.xlsx)", file_types=[".xlsx", ".xls"], type="filepath")
-            cabinet_dd = gr.Dropdown(label="Шкаф вручную", choices=[], interactive=True)
-            db_status = gr.Textbox(label="Статус", lines=5)
+        # Левая колонка - База
+        with gr.Column(scale=1): 
+            gr.Markdown("### 1. База знаний")
+            # height=100 - исправляет гигантский размер
+            db_input = gr.File(label="Excel (.xlsx)", file_types=[".xlsx", ".xls"], type="filepath", height=100)
+            db_status = gr.Textbox(label="Статус базы", lines=3, interactive=False)
+            cabinet_dd = gr.Dropdown(label="Выбрать шкаф вручную (если авто не сработал)", choices=[], interactive=True)
             
-        with gr.Column():
-            gr.Markdown("### 📄 Чертежи")
-            pdf_input = gr.File(label="PDF файлы", file_count="multiple", file_types=[".pdf"], type="filepath")
-            run_btn = gr.Button("🔍 Сформировать", variant="primary")
-
+        # Правая колонка - Чертежи
+        with gr.Column(scale=1):
+            gr.Markdown("### 2. Проверка чертежей")
+            # height=100 - исправляет гигантский размер
+            pdf_input = gr.File(label="Загрузить PDF чертежи", file_count="multiple", file_types=[".pdf"], type="filepath", height=100)
+            run_btn = gr.Button("🔍 Сформировать чек-лист", variant="primary", size="lg")
+
+    # Нижняя секция - Результат
+    gr.Markdown("### 3. Результат")
     with gr.Row():
-        result_txt = gr.Textbox(label="Результат", lines=5)
-        result_pdf = gr.File(label="📥 PDF")
+        with gr.Column(scale=1):
+            result_txt = gr.Textbox(label="Лог проверки", lines=5)
+        with gr.Column(scale=1):
+            result_pdf = gr.File(label="📥 Скачать PDF")
 
+    # Логика
     db_input.change(checker.load_excel_db, [db_input], [db_status, cabinet_dd])
     run_btn.click(checker.check_files, [pdf_input, cabinet_dd], [result_txt, result_pdf])