Update app.py

app.py

@@ -78,7 +78,9 @@ class KDChecker:
         text = ""
         try:
             with pdfplumber.open(pdf_path) as pdf:
-                for page in pdf.pages:
+                # Читаем только первые 2 страницы для скорости, обычно номер там
+                # Если нужно читать весь файл - уберите [:2]
+                for page in pdf.pages[:5]: 
                     text += (page.extract_text() or "") + "\n"
         except Exception as e:
             print(f"Ошибка чтения PDF {pdf_path}: {e}")
@@ -162,32 +164,30 @@ class KDChecker:
             found_by_method = "manual"
             is_manual = True
         else:
-            all_pdf_text = ""
-            for file_path in progress.tqdm(files, desc="Чтение PDF"):
-                all_pdf_text += self.extract_text(file_path) + "\n"
-
-            pdf_numbers = self.find_all_decimal_numbers(all_pdf_text)
-            print(f"Найдено номеров: {pdf_numbers}")
-
+            # === ОПТИМИЗАЦИЯ: Читаем файлы по очереди и прерываемся, если нашли ===
             db_clean_keys = set(self.excel_db["Cabinet_Clean"].tolist())
+            
+            for file_path in progress.tqdm(files, desc="Поиск номера шкафа"):
+                text = self.extract_text(file_path)
+                pdf_numbers = self.find_all_decimal_numbers(text)
+                
+                # Проверяем, есть ли найденные номера в базе
+                for cand in pdf_numbers:
+                    if cand in db_clean_keys:
+                        detected_cabinet = cand
+                        found_by_method = "number"
+                        break # Прерываем поиск по номерам в текущем файле
+                
+                if found_by_method == "number":
+                    print(f"✅ Шкаф найден в файле: {os.path.basename(file_path)}")
+                    break # Прерываем чтение остальных файлов!
 
-            for cand in pdf_numbers:
-                if cand in db_clean_keys:
-                    detected_cabinet = cand
-                    found_by_method = "number"
-                    break
-
+            # Если по номерам не нашли, пробуем по имени (тоже не читаем все подряд)
             if detected_cabinet == "Не определен":
-                unique_cabinets = self.excel_db["Cabinet"].unique()
-                for cab_name in unique_cabinets:
-                    sub_names = [n.strip() for n in cab_name.split(',')]
-                    for sub_name in sub_names:
-                        if len(sub_name) < 5: continue
-                        if sub_name.lower() in all_pdf_text.lower():
-                            detected_cabinet = cab_name
-                            found_by_method = "name"
-                            break
-                    if found_by_method == "name": break
+                # Здесь сложнее, так как имя может быть в любом файле. 
+                # Но если мы уже прочитали часть файлов и не нашли номер, 
+                # скорее всего это ручной ввод.
+                pass 
 
         print(f"Определен шкаф: {detected_cabinet}")
 
@@ -200,6 +200,7 @@ class KDChecker:
         if not remarks:
             return f"⚠️ Для шкафа '{detected_cabinet}' нет замечаний в базе.", None
 
+        # Формируем список задач (здесь уже не читаем PDF, только имена файлов)
         processed_count = 0
         for file_path in files:
             fname = os.path.basename(file_path)
@@ -233,24 +234,19 @@ class KDChecker:
         form = c.acroForm
         width, height = A4
 
-        # --- ШРИФТЫ (ДЛЯ HUGGING FACE) ---
+        # --- ШРИФТЫ ---
         font_name = 'Helvetica'
-        # На HF файл должен лежать в корне рядом с app.py
         local_font = "arial.ttf" 
 
         try:
-            # Сначала ищем локальный файл (который вы должны загрузить на HF)
             if os.path.exists(local_font):
                 pdfmetrics.registerFont(TTFont('Arial', local_font))
                 font_name = 'Arial'
-            # Затем пробуем системный (для Windows тестов)
             elif os.path.exists("C:\\Windows\\Fonts\\arial.ttf"):
                 pdfmetrics.registerFont(TTFont('Arial', "C:\\Windows\\Fonts\\arial.ttf"))
                 font_name = 'Arial'
-            else:
-                print("⚠️ Шрифт arial.ttf не найден! Русский текст может не отображаться.")
-        except Exception as e:
-            print(f"Ошибка шрифт��: {e}")
+        except:
+            pass
 
         y = height - 50
         c.setFont(font_name, 16)
@@ -316,7 +312,7 @@ class KDChecker:
         return path
 
 
-# --- НОВЫЙ ИНТЕРФЕЙС С CSS ---
+# --- ИНТЕРФЕЙС ---
 
 css = """
 .gradio-container { max-width: 95% !important; }
@@ -329,9 +325,6 @@ footer { display: none !important; }
 def create_app():
     checker = KDChecker()
     
-    # Чтобы исправить warning, убираем css и theme отсюда, если версия gradio очень новая,
-    # НО для Hugging Face обычно лучше оставить здесь для предпросмотра.
-    # Если все еще ругается - перенесем в launch.
     with gr.Blocks(title="Генератор чек-листов КД") as app:
         gr.Markdown("## ✅ Генератор чек-листов КД")
 
@@ -361,14 +354,6 @@ def create_app():
     
     return app
 
-# --- ЗАПУСК ДЛЯ HUGGING FACE ---
 if __name__ == "__main__":
     app = create_app()
-    # ДЛЯ HUGGING FACE SPACES НУЖНО ПРОСТО .launch() БЕЗ АРГУМЕНТОВ!
-    # Мы передаем CSS и Theme сюда, чтобы убрать Warning
-    app.launch(
-        server_name="0.0.0.0", # Важно для Docker контейнеров
-        server_port=7860,      # Стандартный порт HF
-        css=css,               # Передали стили сюда
-        theme=gr.themes.Soft() # Передали тему сюда
-    )
+    app.launch(server_name="0.0.0.0", server_port=7860, css=css, theme=gr.themes.Soft())