Update app.py

app.py

@@ -8,7 +8,7 @@ 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
 
@@ -29,20 +29,18 @@ class KDChecker:
         self.cabinet_list = []
         self.known_docs = ["Э3", "В4", "ПЭ3", "ВО", "ТЭ5", "СБ", "С5", "ОЛ", "Э1", "Э4", "Э7", "Д3", "Э6"]
         
-        # === РЕГИСТРАЦИЯ ШРИФТА (Исправление квадратов) ===
-        self.font_name = 'Helvetica' # Запасной вариант
+        # === ШРИФТЫ ===
+        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}")
+                print(f"⚠️ Шрифт не найден (ищем {font_path})")
         except Exception as e:
-            print(f"❌ Ошибка загрузки шрифта: {e}")
+            print(f"❌ Ошибка шрифта: {e}")
 
     def load_excel_db(self, excel_file):
         if excel_file is None:
@@ -51,10 +49,9 @@ class KDChecker:
         excel_path = excel_file if isinstance(excel_file, str) else getattr(excel_file, 'name', str(excel_file))
         
         if not os.path.exists(excel_path):
-            return f"❌ Файл не найден: {excel_path}", gr.update(choices=[], value=None)
+            return f"❌ Файл не найден", gr.update(choices=[], value=None)
 
         all_data = []
-        sheets_info = []
 
         try:
             xls = pd.read_excel(excel_path, sheet_name=None, header=None, engine='openpyxl')
@@ -64,6 +61,7 @@ class KDChecker:
                 cab_col = -1
                 rem_col = -1
 
+                # Ищем заголовки
                 for i in range(min(20, len(df_raw))):
                     row_vals = [str(x).lower().strip() for x in df_raw.iloc[i].values]
                     for idx, val in enumerate(row_vals):
@@ -75,8 +73,9 @@ class KDChecker:
                         header_row = i
                         break
 
-                if header_row != -1 and cab_col < len(df_raw.columns) and rem_col < len(df_raw.columns):
+                if header_row != -1:
                     df = pd.read_excel(excel_path, sheet_name=sheet_name, header=header_row, engine='openpyxl')
+                    # Проверяем индексы еще раз
                     if cab_col < len(df.columns) and rem_col < len(df.columns):
                         df_sub = df.iloc[:, [cab_col, rem_col]]
                         df_sub.columns = ["Cabinet", "Remark"]
@@ -86,24 +85,18 @@ class KDChecker:
                             lambda x: x.strip().replace(" ", "").replace("\n", "").replace("\r", "")
                         )
                         all_data.append(df_sub)
-                        sheets_info.append(f"'{sheet_name}': {len(df_sub)}")
 
             if not all_data:
-                return "❌ Не найдены колонки 'Шкаф' и 'Примечание'", gr.update(choices=[], value=None)
+                return "❌ Структура Excel не распознана", gr.update(choices=[], value=None)
 
             self.excel_db = pd.concat(all_data, ignore_index=True)
             self.cabinet_list = sorted(self.excel_db["Cabinet"].unique().tolist())
             
-            msg = f"✅ База загружена!\n\n"
-            msg += f"📊 Записей: {len(self.excel_db)}\n"
-            msg += f"🗄️ Шкафов: {len(self.cabinet_list)}\n"
-            msg += f"📋 Листы: {', '.join(sheets_info)}"
-            
+            msg = f"✅ База загружена!\nВсего записей: {len(self.excel_db)}\nШкафов: {len(self.cabinet_list)}"
             return msg, gr.update(choices=self.cabinet_list, value=None)
 
         except Exception as e:
-            import traceback
-            return f"❌ Ошибка: {e}\n\n{traceback.format_exc()}", gr.update(choices=[], value=None)
+            return f"❌ Ошибка Excel: {e}", gr.update(choices=[], value=None)
 
     def extract_text(self, pdf_path):
         try:
@@ -127,7 +120,7 @@ class KDChecker:
     def determine_doc_type(self, filename):
         fname = filename.upper()
         doc_map = {
-            "С2": ["С2"], "ПЭ3": ["ПЭ3", "ПЕРЕЧЕНЬ"], "Э3": ["Э3", "СХЕМА ЭЛЕКТРИЧЕСКАЯ"],
+            "С2": ["С2"], "ПЭ3": ["ПЭ3", "ПЕРЕЧЕНЬ"], "��3": ["Э3", "СХЕМА"],
             "Э4": ["Э4"], "В4": ["В4", "СПЕЦИФИКАЦИЯ"], "ВО": ["ВО", "Э7", "ГАБАРИТ"],
             "ТЭ5": ["ТЭ5", "ТАБЛИЦА"], "СБ": ["СБ"], "С5": ["С5"], "ОЛ": ["ОЛ"],
             "Э1": ["Э1"], "Э6": ["Э6", "ЗАЗЕМЛЕНИЯ"], "Д3": ["Д3", "МОНТАЖ"]
@@ -138,8 +131,7 @@ class KDChecker:
         return "UNKNOWN"
 
     def get_remarks(self, cabinet_key, is_clean=True):
-        if self.excel_db.empty:
-            return {}
+        if self.excel_db.empty: return {}
         
         if is_clean:
             target = cabinet_key.replace(" ", "")
@@ -148,9 +140,6 @@ class KDChecker:
             mask = self.excel_db['Cabinet'] == cabinet_key
 
         rows = self.excel_db[mask]
-        if rows.empty:
-            return {}
-        
         parsed = {}
         for remark in rows['Remark']:
             text = str(remark)
@@ -158,8 +147,7 @@ class KDChecker:
             items = re.split(r'(?:^|\n)\s*(?=\d+[\.\)])', text)
             
             for item in items:
-                if len(item) < 3:
-                    continue
+                if len(item) < 3: continue
                 clean_item = item.strip()
                 no_num = re.sub(r'^\d+[\.\)]\s*', '', clean_item)
                 pattern = r'^(?:Документ\s+|В\s+)?([А-ЯA-Z0-9\s,\(\)\-]+?)(?:[\.\:\-]|\s+)(.*)'
@@ -167,7 +155,6 @@ class KDChecker:
                 
                 docs = []
                 final_text = clean_item
-                
                 if match:
                     potential = match.group(1).upper()
                     cleaned = potential.replace("(", " ").replace(")", " ").replace(",", " ")
@@ -177,26 +164,17 @@ class KDChecker:
                         docs = valid
                         final_text = match.group(2).strip()
                 
-                if not docs:
-                    docs = ["ALL"]
-                
+                if not docs: docs = ["ALL"]
                 for d in docs:
-                    if d not in parsed:
-                        parsed[d] = []
+                    if d not in parsed: parsed[d] = []
                     parsed[d].append(final_text)
-        
         return parsed
 
     def check_files(self, files, manual_cabinet):
-        if not files:
-            return "❌ Загрузите PDF файлы", None
-        if self.excel_db.empty:
-            return "❌ Сначала загрузите Excel!", None
+        if not files: return "⚠️ Загрузите PDF", None
+        if self.excel_db.empty: return "⚠️ Загрузите базу Excel", None
 
-        file_paths = []
-        for f in files:
-            fp = f if isinstance(f, str) else getattr(f, 'name', str(f))
-            file_paths.append(fp)
+        file_paths = [f if isinstance(f, str) else getattr(f, 'name', str(f)) for f in files]
 
         detected_cabinet = "Не определен"
         method = ""
@@ -225,15 +203,14 @@ class KDChecker:
                             detected_cabinet = cab
                             method = "name"
                             break
-                    if method:
-                        break
+                    if method: break
 
         if detected_cabinet == "Не определен":
-            return "⚠️ Шкаф не определён. Выберите вручную.", None
+            return "⚠️ Шкаф не определён. Выберите вручную в списке слева.", None
 
         remarks = self.get_remarks(detected_cabinet, is_clean=(method == "number"))
         if not remarks:
-            return f"⚠️ Для '{detected_cabinet}' замечаний нет.", None
+            return f"✅ Для '{detected_cabinet}' замечаний не найдено.", None
 
         checklist = {}
         for fp in file_paths:
@@ -248,34 +225,25 @@ class KDChecker:
         try:
             pdf_path = self.create_pdf(title, checklist)
         except Exception as e:
-            import traceback
-            traceback.print_exc() # Пишем в консоль для отладки
-            return f"❌ Ошибка генерации PDF: {e}", None
+            return f"❌ Ошибка PDF: {e}", None
 
         total = sum(len(v) for v in checklist.values())
-        method_name = {"manual": "Ручной", "number": "По номеру", "name": "По названию"}.get(method, "?")
-
-        return f"✅ Готово!\n\n📂 {detected_cabinet}\n🔍 {method_name}\n📄 Файлов: {len(file_paths)}\n🚩 Замечаний: {total}", pdf_path
+        return f"✅ Готово!\nШкаф: {detected_cabinet}\nМетод: {method}\nЗамечаний: {total}", pdf_path
 
     def create_pdf(self, cabinet, data):
         path = os.path.join(tempfile.gettempdir(), "CheckList.pdf")
         c = canvas.Canvas(path, pagesize=A4)
         w, h = A4
-        
-        # Используем наш зарегистрированный шрифт
-        font = self.font_name 
+        font = self.font_name
 
         y = h - 50
         c.setFont(font, 16)
         c.drawString(50, y, "CHECK-LIST")
         y -= 25
         c.setFont(font, 12)
-        
-        # Обрезаем очень длинные названия шкафов для заголовка
         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'))
+        c.drawString(450, y, datetime.now().strftime('%d.%m.%Y'))
         y -= 20
         c.line(50, y, w - 50, y)
         y -= 30
@@ -285,12 +253,13 @@ class KDChecker:
             c.save()
             return path
 
+        cb_counter = 0 # Счетчик для уникальных имен чекбоксов
+
         for fname, tasks in data.items():
             if y < 100:
                 c.showPage()
                 y = h - 50
-                # Важно: на новой странице тоже нужно ставить шрифт
-                c.setFont(font, 12) 
+                c.setFont(font, 12)
                 
             c.setFillColor(colors.darkblue)
             c.setFont(font, 11)
@@ -300,33 +269,43 @@ class KDChecker:
             c.setFont(font, 10)
 
             for task in tasks:
+                cb_counter += 1
                 if y < 60:
                     c.showPage()
                     y = h - 50
                     c.setFont(font, 10)
                 
-                # Рисуем квадратик чекбокса
-                c.rect(50, y - 2, 8, 8, stroke=1, fill=0)
+                # === ИСПРАВЛЕНИЕ ЧЕКБОКСОВ ===
+                # Вместо rect рисуем настоящую форму
+                c.acroForm.checkbox(
+                    name=f'CB_{cb_counter}',
+                    x=50, 
+                    y=y-4, 
+                    buttonStyle='check',
+                    borderColor=colors.black,
+                    fillColor=colors.white,
+                    textColor=colors.black,
+                    forceBorder=True,
+                    size=10
+                )
                 
-                # Перенос строк
+                # Текст задачи
                 words = task.replace('\n', ' ').split()
                 lines, line = [], ""
                 for word in words:
-                    # Примерная ширина строки. Arial чуть шире Helvetica, запас 80
-                    if len(line) + len(word) < 80: 
+                    if len(line) + len(word) < 80:
                         line += word + " "
                     else:
                         lines.append(line.strip())
                         line = word + " "
-                if line:
-                    lines.append(line.strip())
+                if line: lines.append(line.strip())
 
                 for ln in lines:
                     if y < 40:
                         c.showPage()
                         y = h - 50
                         c.setFont(font, 10)
-                    c.drawString(65, y, ln)
+                    c.drawString(70, y, ln) # Отступ 70, чтобы не налезть на чекбокс
                     y -= 12
                 y -= 5
             y -= 15
@@ -338,39 +317,47 @@ class KDChecker:
 # ========== ИНТЕРФЕЙС ==========
 checker = KDChecker()
 
-# Используем CSS для улучшения вида
-css = """
-.file-upload { height: 100px !important; min-height: 100px !important; }
+# CSS для компактности и оранжевой кнопки
+custom_css = """
+.compact_file { height: 120px !important; min-height: 120px !important; }
+.orange_btn { background: #FF7F27 !important; border: none !important; color: white !important; font-size: 16px !important;}
+.gradio-container { max-width: 1400px !important; }
+footer { display: none !important; }
 """
 
-with gr.Blocks(title="КД Checker", theme=gr.themes.Soft(), css=css) as app:
-    gr.Markdown("# ✅ Генератор чек-листов КД")
+with gr.Blocks(title="КД Checker", theme=gr.themes.Soft(), css=custom_css) as app:
+    gr.Markdown("## ✅ Локальный генератор чек-листов КД")
     
     with gr.Row():
-        # Левая колонка - База
-        with gr.Column(scale=1): 
+        # --- ЛЕВАЯ КОЛОНКА (БАЗА) ---
+        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)
+            db_input = gr.File(label="Загрузить Excel (.xlsx)", file_types=[".xlsx", ".xls"], 
+                               type="filepath", elem_classes="compact_file")
             
-        # Правая колонка - Чертежи
+            # Поля управления базой под файлом
+            with gr.Group():
+                cabinet_dd = gr.Dropdown(label="Или выберите шкаф-аналог вручную", choices=[], interactive=True)
+                db_status = gr.Textbox(label="Статус базы", lines=2, interactive=False, placeholder="Ожидание загрузки...")
+
+        # --- ПРАВАЯ КОЛОНКА (ЧЕРТЕЖИ) ---
         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("### 2. Документация")
+            pdf_input = gr.File(label="Чертежи (PDF)", file_count="multiple", file_types=[".pdf"], 
+                                type="filepath", elem_classes="compact_file")
+            
+            # Кнопка под файлами
+            gr.Markdown("") # Отступ
+            run_btn = gr.Button("Сформировать чек-лист", variant="primary", elem_classes="orange_btn")
 
-    # Нижняя секция - Результат
-    gr.Markdown("### 3. Результат")
+    # --- НИЖНЯЯ СЕКЦИЯ (РЕЗУЛЬТАТ) ---
     with gr.Row():
         with gr.Column(scale=1):
-            result_txt = gr.Textbox(label="Лог проверки", lines=5)
+            result_txt = gr.Textbox(label="Результат", lines=6)
         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])