Update app.py

app.py

@@ -11,6 +11,10 @@ from reportlab.pdfbase import pdfmetrics
 from reportlab.pdfbase.ttfonts import TTFont
 from reportlab.lib import colors
 
+# --- ОТЛАДОЧНАЯ ПЕЧАТЬ ---
+def debug_log(message):
+    print(f"[{datetime.now().strftime('%H:%M:%S')}] {message}")
+
 class KDChecker:
     def __init__(self):
         self.excel_db = pd.DataFrame()
@@ -18,7 +22,7 @@ class KDChecker:
         self.known_docs = ["Э3", "В4", "ПЭ3", "ВО", "ТЭ5", "СБ", "С5", "ОЛ", "Э1", "Э4", "Э7", "Д3", "Э6"]
 
     def load_excel_db(self, excel_path):
-        # Если файл не выбран
+        debug_log(f"Начало загрузки Excel: {excel_path}")
         if excel_path is None:
             return "Файл не выбран", gr.update(choices=[], value=None)
 
@@ -26,8 +30,8 @@ class KDChecker:
         sheets_log = []
 
         try:
-            # Читаем Excel напрямую по пути
             xls = pd.read_excel(excel_path, sheet_name=None, header=None)
+            debug_log(f"Excel прочитан, листов: {len(xls)}")
 
             for sheet_name, df_raw in xls.items():
                 header_row_index = -1
@@ -65,17 +69,19 @@ class KDChecker:
                     sheets_log.append(f"Лист '{sheet_name}': заголовки не найдены")
 
             if not all_data:
+                debug_log("Ошибка: данные не найдены")
                 return "❌ Ошибка: Не найдены заголовки 'Шкаф' и 'Примечание'.", 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())
 
+            debug_log(f"Успех. Всего строк: {len(self.excel_db)}")
             msg = f"✅ База знаний загружена успешно!\nВсего записей: {len(self.excel_db)}\nОбработаны листы: {', '.join(sheets_log)}"
             return msg, gr.update(choices=self.cabinet_list, value=None, interactive=True)
 
         except Exception as e:
-            # Выводим ошибку в интерфейс, чтобы не гадать
-            return f"❌ Критическая ошибка чтения Excel: {str(e)}", gr.update(choices=[], value=None)
+            debug_log(f"Exception Excel: {e}")
+            return f"❌ Ошибка чтения Excel: {e}", gr.update(choices=[], value=None)
 
     def extract_text(self, pdf_path):
         try:
@@ -84,7 +90,8 @@ class KDChecker:
                 for page in pdf.pages:
                     full_text += (page.extract_text() or "") + "\n"
             return full_text
-        except:
+        except Exception as e:
+            debug_log(f"Ошибка чтения PDF {pdf_path}: {e}")
             return ""
 
     def find_all_decimal_numbers(self, text):
@@ -115,7 +122,6 @@ class KDChecker:
 
     def get_remarks(self, cabinet_key, is_clean_key=True):
         if self.excel_db.empty: return {}
-
         if is_clean_key:
             target = cabinet_key.replace(" ", "")
             mask = self.excel_db['Cabinet_Clean'].str.contains(re.escape(target), case=False, na=False)
@@ -124,46 +130,35 @@ class KDChecker:
 
         rows = self.excel_db[mask]
         if rows.empty: return {}
-
         parsed = {}
-
         for remark_cell in rows['Remark']:
             cell_text = str(remark_cell)
             cell_text = re.sub(r'(\d+)\.([А-ЯA-Z])', r'\1. \2', cell_text)
-
             items = re.split(r'(?:^|\n)\s*(?=\d+[\.\)])', cell_text)
-
             for item in items:
                 if len(item) < 3: continue
                 clean_item = item.strip()
                 clean_item_no_num = re.sub(r'^\d+[\.\)]\s*', '', clean_item)
-
                 doc_pattern = r'^(?:Документ\s+|В\s+)?([А-ЯA-Z0-9\s,\(\)\-]+?)(?:[\.\:\-]|\s+)(.*)'
                 match = re.match(doc_pattern, clean_item_no_num, re.IGNORECASE | re.DOTALL)
-
                 detected_docs = []
                 final_text = clean_item
-
                 if match:
                     potential_docs_str = match.group(1).upper()
                     cleaned_codes = potential_docs_str.replace("(", " ").replace(")", " ").replace(",", " ")
                     parts = cleaned_codes.split()
-
                     valid_parts = [p for p in parts if p in self.known_docs]
                     if valid_parts:
                         detected_docs = valid_parts
                         final_text = match.group(2).strip()
-
-                if not detected_docs:
-                    detected_docs = ["ALL"]
-
+                if not detected_docs: detected_docs = ["ALL"]
                 for doc in detected_docs:
                     if doc not in parsed: parsed[doc] = []
                     parsed[doc].append(final_text)
-
         return parsed
 
     def check_files(self, files, manual_cabinet):
+        debug_log("Начало проверки файлов")
         if not files: return "Файлы не загр��жены", None
         if self.excel_db.empty: return "Сначала загрузите Excel базу!", None
 
@@ -176,10 +171,13 @@ class KDChecker:
             detected_cabinet = manual_cabinet
             found_by_method = "manual"
             is_manual = True
+            debug_log(f"Выбран ручной шкаф: {detected_cabinet}")
         else:
             all_pdf_text = ""
             for file_path in files:
                 all_pdf_text += self.extract_text(file_path) + "\n"
+            
+            debug_log(f"Текст извлечен, длина: {len(all_pdf_text)}")
 
             pdf_numbers = self.find_all_decimal_numbers(all_pdf_text)
             db_clean_keys = set(self.excel_db["Cabinet_Clean"].tolist())
@@ -189,7 +187,7 @@ class KDChecker:
                     detected_cabinet = cand
                     found_by_method = "number"
                     break
-
+            
             if detected_cabinet == "Не определен":
                 unique_cabinets = self.excel_db["Cabinet"].unique()
                 for cab_name in unique_cabinets:
@@ -215,11 +213,9 @@ class KDChecker:
         for file_path in files:
             fname = os.path.basename(file_path)
             dtype = self.determine_doc_type(fname)
-
             tasks = []
             if dtype in remarks: tasks.extend(remarks[dtype])
             if "ALL" in remarks and dtype != "С2": tasks.extend(remarks["ALL"])
-
             if tasks:
                 checklist[fname] = list(dict.fromkeys(tasks))
             processed_count += 1
@@ -227,11 +223,16 @@ class KDChecker:
         pdf_title = detected_cabinet
         if is_manual: pdf_title += " (Выбор вручную)"
 
-        pdf = self.create_pdf(pdf_title, checklist)
-        total = sum(len(v) for v in checklist.values())
+        debug_log("Формирование PDF...")
+        try:
+            pdf = self.create_pdf(pdf_title, checklist)
+            debug_log("PDF успешно создан")
+        except Exception as e:
+            debug_log(f"ОШИБКА создания PDF: {e}")
+            return f"❌ Ошибка создания PDF (возможно, нет шрифта arial.ttf?): {e}", None
 
-        method_str = "Ручной выбор" if is_manual else (
-            "По децимальному номеру" if is_clean_search else "По наименованию")
+        total = sum(len(v) for v in checklist.values())
+        method_str = "Ручной выбор" if is_manual else ("По децимальному номеру" if is_clean_search else "По наименованию")
 
         return f"✅ Готово!\n\n📂 Шкаф: {detected_cabinet}\n🔍 Метод: {method_str}\n📄 Обработано файлов: {processed_count}\n🚩 Всего замечаний: {total}", pdf
 
@@ -242,28 +243,51 @@ class KDChecker:
         form = c.acroForm
         width, height = A4
 
-        font_name = 'Helvetica'
-        font_path = "arial.ttf"
-
-        possible_paths = ["C:\\Windows\\Fonts\\arial.ttf", "arial.ttf",
-                          "/usr/share/fonts/truetype/msttcorefonts/Arial.ttf"]
+        # --- ПОИСК ШРИФТА ---
+        font_name = 'Helvetica' # По умолчанию (но не умеет кириллицу)
+        font_path = "arial.ttf" # Ищем в текущей папке
 
-        for p in possible_paths:
-            if os.path.exists(p):
-                try:
-                    pdfmetrics.registerFont(TTFont('Arial', p))
-                    font_name = 'Arial'
-                    break
-                except:
-                    continue
+        if os.path.exists(font_path):
+            try:
+                pdfmetrics.registerFont(TTFont('Arial', font_path))
+                font_name = 'Arial'
+            except Exception as e:
+                debug_log(f"Ошибка регистрации шрифта: {e}")
+        else:
+            debug_log("⚠️ Файл arial.ttf не найден! Кирилица может сломаться.")
+            # Попытаемся найти системный, если локального нет (на HF не сработает, но для локалки полезно)
+            possible_paths = ["/usr/share/fonts/truetype/msttcorefonts/Arial.ttf"]
+            for p in possible_paths:
+                if os.path.exists(p):
+                    try:
+                        pdfmetrics.registerFont(TTFont('Arial', p))
+                        font_name = 'Arial'
+                        break
+                    except: continue
 
         y = height - 50
-        c.setFont(font_name, 16)
-        c.drawString(50, y, f"ЧЕК-ЛИСТ ПРОВЕРКИ КД")
+        
+        # Если шрифт не Arial, кириллица вызовет ошибку. Пробуем перехватить.
+        try:
+            c.setFont(font_name, 16)
+            c.drawString(50, y, f"ЧЕК-ЛИСТ ПРОВЕРКИ КД")
+        except:
+            # Если упало на кирилли��е, пишем транслитом
+            c.setFont("Helvetica", 16)
+            c.drawString(50, y, "CHECK-LIST PROVERKI KD (Font Missing)")
+
         y -= 25
         c.setFont(font_name, 12)
         disp_cab = cabinet[:60] + "..." if len(cabinet) > 60 else cabinet
-        c.drawString(50, y, f"Шкаф: {disp_cab}")
+        
+        # Безопасный вывод текста
+        try:
+            c.drawString(50, y, f"Шкаф: {disp_cab}")
+        except:
+            c.setFont("Helvetica", 12)
+            c.drawString(50, y, f"Cabinet: {disp_cab} (Cyrillic Error)")
+            c.setFont(font_name, 12) # Возвращаем
+
         c.drawString(400, y, f"Дата: {datetime.now().strftime('%d.%m.%Y')}")
         y -= 20
         c.line(50, y, width - 50, y)
@@ -280,23 +304,25 @@ class KDChecker:
 
             c.setFillColor(colors.darkblue)
             c.setFont(font_name, 11)
-            c.drawString(50, y, f"Файл: {filename}")
+            try:
+                c.drawString(50, y, f"Файл: {filename}")
+            except:
+                c.setFont("Helvetica", 11)
+                c.drawString(50, y, f"File: {filename}")
+                c.setFont(font_name, 11)
+
             c.setFillColor(colors.black)
             y -= 15
             c.setFont(font_name, 10)
 
             for task in tasks:
                 paragraphs = task.split('\n')
-
                 if y < 80:
-                    c.showPage();
-                    y = height - 50;
-                    c.setFont(font_name, 10)
+                    c.showPage(); y = height - 50; c.setFont(font_name, 10)
 
                 c.rect(50, y - 10, 10, 10)
                 form.checkbox(name=f"cb_{cb_id}", x=50, y=y - 10, size=10, buttonStyle='check', forceBorder=True)
                 cb_id += 1
-
                 text_start_y = y - 2
 
                 for paragraph in paragraphs:
@@ -313,14 +339,15 @@ class KDChecker:
 
                     for l in lines:
                         if text_start_y < 40:
-                            c.showPage();
-                            text_start_y = height - 50;
-                            c.setFont(font_name, 10)
-                        c.drawString(65, text_start_y, l.strip())
+                            c.showPage(); text_start_y = height - 50; c.setFont(font_name, 10)
+                        try:
+                            c.drawString(65, text_start_y, l.strip())
+                        except:
+                             c.setFont("Helvetica", 10)
+                             c.drawString(65, text_start_y, "Error printing line (font missing)")
+                             c.setFont(font_name, 10)
                         text_start_y -= 12
-
                 y = text_start_y - 8
-
             y -= 10
             c.setStrokeColor(colors.lightgrey)
             c.line(50, y, width - 50, y)
@@ -330,7 +357,6 @@ class KDChecker:
         c.save()
         return path
 
-
 def create_app():
     checker = KDChecker()
     with gr.Blocks(title="Генератор чек-листов КД") as app:
@@ -340,16 +366,15 @@ def create_app():
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### 1. База знаний")
-                # ВАЖНО: Добавлен type="filepath", чтобы не зависало
+                # type="filepath" чтобы не зависало
                 db_in = gr.File(label="Загрузить Excel (.xlsx)", type="filepath")
                 manual_cab = gr.Dropdown(label="Или выберите шкаф-аналог вручную", choices=[], interactive=True)
                 db_out = gr.Textbox(label="Статус загрузки", lines=8, max_lines=30)
-
                 db_in.upload(checker.load_excel_db, inputs=[db_in], outputs=[db_out, manual_cab])
 
             with gr.Column():
                 gr.Markdown("### 2. Документация (PDF)")
-                # ВАЖНО: Добавлен type="filepath", чтобы не зависало
+                # type="filepath" чтобы не зависало
                 files_in = gr.File(label="Загрузить чертежи", file_count="multiple", type="filepath")
                 btn = gr.Button("Сформировать чек-лист", variant="primary")
 
@@ -360,7 +385,6 @@ def create_app():
         btn.click(checker.check_files, inputs=[files_in, manual_cab], outputs=[res_txt, res_pdf])
     return app
 
-
 app = create_app()
 
 if __name__ == "__main__":