Ed5 commited on
Commit
418e0a8
·
verified ·
1 Parent(s): dbad057

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +181 -287
app.py CHANGED
@@ -18,7 +18,8 @@ AUTH_PASSWORD = "12345"
18
 
19
  print("===== Application Startup =====")
20
  print(f"Time: {datetime.now()}")
21
- print(f"Auth enabled: Username={AUTH_USERNAME}")
 
22
  print("================================")
23
 
24
 
@@ -27,150 +28,111 @@ class KDChecker:
27
  self.excel_db = pd.DataFrame()
28
  self.cabinet_list = []
29
  self.known_docs = ["Э3", "В4", "ПЭ3", "ВО", "ТЭ5", "СБ", "С5", "ОЛ", "Э1", "Э4", "Э7", "Д3", "Э6"]
30
- print("[KDChecker] Инициализация завершена")
31
 
32
  def load_excel_db(self, excel_file):
33
  """Загрузка Excel базы знаний"""
34
- log = [f"[{datetime.now().strftime('%H:%M:%S')}] Старт загрузки Excel"]
35
 
36
- # Проверка входных данных
37
  if excel_file is None:
38
- log.append("ERROR: excel_file is None")
39
- return "\n".join(log), gr.update(choices=[], value=None)
40
 
41
- # Определяем путь к файлу (gradio может передать разные типы)
42
  if isinstance(excel_file, str):
43
  excel_path = excel_file
44
- elif hasattr(excel_file, 'name'):
45
- excel_path = excel_file.name
46
  else:
47
- excel_path = str(excel_file)
48
 
49
- log.append(f"Тип входных данных: {type(excel_file)}")
50
- log.append(f"Путь к файлу: {excel_path}")
51
 
52
- # Проверка существования файла
53
  if not os.path.exists(excel_path):
54
- log.append(f"ERROR: Файл не найден по пути: {excel_path}")
55
- return "\n".join(log), gr.update(choices=[], value=None)
56
 
57
  try:
58
- size = os.path.getsize(excel_path)
59
- log.append(f"Размер файла: {size} байт")
60
  except Exception as e:
61
- log.append(f"ERROR при получении размера: {e}")
62
- return "\n".join(log), gr.update(choices=[], value=None)
63
-
64
- # Проверка openpyxl
65
- try:
66
- import openpyxl
67
- log.append(f"openpyxl version: {openpyxl.__version__}")
68
- except ImportError as e:
69
- log.append(f"ERROR: openpyxl не установлен: {e}")
70
- return "\n".join(log), gr.update(choices=[], value=None)
71
 
72
  all_data = []
73
- sheets_log = []
74
 
75
  try:
76
- log.append("Чтение Excel файла...")
77
  xls = pd.read_excel(excel_path, sheet_name=None, header=None, engine='openpyxl')
78
- log.append(f"Найдено листов: {len(xls)}")
79
- log.append(f"Имена листов: {list(xls.keys())}")
80
 
81
  for sheet_name, df_raw in xls.items():
82
- log.append(f"--- Обработка листа: '{sheet_name}' ({len(df_raw)} строк) ---")
83
-
84
- header_row_index = -1
85
- cab_col_idx = -1
86
- rem_col_idx = -1
87
 
88
- # Поиск заголовков в первых 20 строках
89
  for i in range(min(20, len(df_raw))):
90
- row_values = [str(x).lower().strip() for x in df_raw.iloc[i].values]
91
- c_idx = -1
92
- r_idx = -1
93
- for idx, val in enumerate(row_values):
94
  if "шкаф" in val or "cabinet" in val:
95
- c_idx = idx
96
  if "примечание" in val or "remark" in val:
97
- r_idx = idx
98
- if c_idx != -1 and r_idx != -1:
99
- header_row_index = i
100
- cab_col_idx = c_idx
101
- rem_col_idx = r_idx
102
- log.append(f" Заголовки найдены в строке {i}: шкаф={c_idx}, примечание={r_idx}")
103
  break
104
 
105
- if header_row_index != -1:
106
- df = pd.read_excel(excel_path, sheet_name=sheet_name, header=header_row_index, engine='openpyxl')
107
- log.append(f" Колонок в листе: {len(df.columns)}")
108
-
109
- if cab_col_idx < len(df.columns) and rem_col_idx < len(df.columns):
110
- df_subset = df.iloc[:, [cab_col_idx, rem_col_idx]]
111
- df_subset.columns = ["Cabinet", "Remark"]
112
- df_subset["Cabinet"] = df_subset["Cabinet"].ffill()
113
- df_subset = df_subset.dropna(subset=["Remark"]).astype(str)
114
- df_subset["Cabinet_Clean"] = df_subset["Cabinet"].apply(
115
  lambda x: x.strip().replace(" ", "").replace("\n", "").replace("\r", "")
116
  )
117
- all_data.append(df_subset)
118
- sheets_log.append(f"'{sheet_name}': {len(df_subset)} записей")
119
- log.append(f" ✓ Добавлено {len(df_subset)} записей")
120
- else:
121
- sheets_log.append(f"'{sheet_name}': ошибка индексов")
122
- log.append(f" ✗ Ошибка индексов колонок")
123
- else:
124
- sheets_log.append(f"'{sheet_name}': заголовки не найдены")
125
- log.append(f" ✗ Заголовки 'шкаф'/'примечание' не найдены")
126
 
127
  if not all_data:
128
- log.append("ИТОГ: Не найдены данные ни на одном листе!")
129
- return "\n".join(log), gr.update(choices=[], value=None)
130
 
131
  self.excel_db = pd.concat(all_data, ignore_index=True)
132
  self.cabinet_list = sorted(self.excel_db["Cabinet"].unique().tolist())
133
 
134
- log.append(f"ИТОГ: Загружено {len(self.excel_db)} записей")
135
- log.append(f"Уникальных шкафов: {len(self.cabinet_list)}")
136
-
137
- result_msg = f" База загружена успешно!\n\n"
138
- result_msg += f"📊 Всего записей: {len(self.excel_db)}\n"
139
- result_msg += f"🗄️ Шкафов: {len(self.cabinet_list)}\n"
140
- result_msg += f"📋 Листы: {', '.join(sheets_log)}"
141
 
142
- return result_msg, gr.update(choices=self.cabinet_list, value=None)
143
 
144
  except Exception as e:
145
- log.append(f"КРИТИЧЕСКАЯ ОШИБКА: {str(e)}")
146
  import traceback
147
  log.append(traceback.format_exc())
148
- return "\n".join(log), gr.update(choices=[], value=None)
149
 
150
  def extract_text(self, pdf_path):
151
- """Извлечение текста из PDF"""
152
  try:
153
- full_text = ""
154
  with pdfplumber.open(pdf_path) as pdf:
155
  for page in pdf.pages:
156
- full_text += (page.extract_text() or "") + "\n"
157
- return full_text
158
- except Exception as e:
159
- print(f"[extract_text] Error: {e}")
160
  return ""
161
 
162
- def find_all_decimal_numbers(self, text):
163
- """Поиск децимальных номеров в тексте"""
164
  pattern = r"(РЛТ|ЛДАР|ВНАР|ШТМ)[\s\.]*\d{1}[\s\.]*\d{3}[\s\.]*[А-ЯA-Z]{1,4}[\s\.]*\d{3}(-[\d]+)?"
165
  matches = []
166
- for match in re.finditer(pattern, text):
167
- clean_num = match.group(0).replace(" ", "").replace("\n", "")
168
- if clean_num not in matches:
169
- matches.append(clean_num)
170
  return matches
171
 
172
  def determine_doc_type(self, filename):
173
- """Определение типа документа по имени файла"""
174
  fname = filename.upper()
175
  if "С2" in fname: return "С2"
176
  if "ПЭ3" in fname or "ПЕРЕЧЕНЬ" in fname: return "ПЭ3"
@@ -187,12 +149,11 @@ class KDChecker:
187
  if "Д3" in fname or "МОНТАЖ" in fname: return "Д3"
188
  return "UNKNOWN"
189
 
190
- def get_remarks(self, cabinet_key, is_clean_key=True):
191
- """Получение замечаний для шкафа"""
192
  if self.excel_db.empty:
193
  return {}
194
 
195
- if is_clean_key:
196
  target = cabinet_key.replace(" ", "")
197
  mask = self.excel_db['Cabinet_Clean'].str.contains(re.escape(target), case=False, na=False)
198
  else:
@@ -203,107 +164,100 @@ class KDChecker:
203
  return {}
204
 
205
  parsed = {}
206
- for remark_cell in rows['Remark']:
207
- cell_text = str(remark_cell)
208
- cell_text = re.sub(r'(\d+)\.([А-ЯA-Z])', r'\1. \2', cell_text)
209
- items = re.split(r'(?:^|\n)\s*(?=\d+[\.\)])', cell_text)
210
 
211
  for item in items:
212
  if len(item) < 3:
213
  continue
214
  clean_item = item.strip()
215
- clean_item_no_num = re.sub(r'^\d+[\.\)]\s*', '', clean_item)
216
- doc_pattern = r'^(?:Документ\s+|В\s+)?([А-ЯA-Z0-9\s,\(\)\-]+?)(?:[\.\:\-]|\s+)(.*)'
217
- match = re.match(doc_pattern, clean_item_no_num, re.IGNORECASE | re.DOTALL)
218
 
219
- detected_docs = []
220
  final_text = clean_item
221
 
222
  if match:
223
- potential_docs_str = match.group(1).upper()
224
- cleaned_codes = potential_docs_str.replace("(", " ").replace(")", " ").replace(",", " ")
225
- parts = cleaned_codes.split()
226
- valid_parts = [p for p in parts if p in self.known_docs]
227
- if valid_parts:
228
- detected_docs = valid_parts
229
  final_text = match.group(2).strip()
230
 
231
- if not detected_docs:
232
- detected_docs = ["ALL"]
233
 
234
- for doc in detected_docs:
235
- if doc not in parsed:
236
- parsed[doc] = []
237
- parsed[doc].append(final_text)
238
 
239
  return parsed
240
 
241
  def check_files(self, files, manual_cabinet):
242
- """Проверка PDF файлов"""
243
  if not files:
244
- return "❌ Файлы не загружены", None
245
  if self.excel_db.empty:
246
  return "❌ Сначала загрузите Excel базу!", None
247
 
248
- checklist = {}
249
- detected_cabinet = "Не определен"
250
- found_by_method = ""
251
- is_manual = False
252
-
253
- # Обработка входных файлов
254
  file_paths = []
255
  for f in files:
256
  if isinstance(f, str):
257
  file_paths.append(f)
258
- elif hasattr(f, 'name'):
259
- file_paths.append(f.name)
260
  else:
261
- file_paths.append(str(f))
 
 
 
 
 
262
 
263
  if manual_cabinet and manual_cabinet.strip():
264
  detected_cabinet = manual_cabinet
265
- found_by_method = "manual"
266
  is_manual = True
267
  else:
268
- all_pdf_text = ""
269
- for file_path in file_paths:
270
- all_pdf_text += self.extract_text(file_path) + "\n"
271
 
272
- pdf_numbers = self.find_all_decimal_numbers(all_pdf_text)
273
- db_clean_keys = set(self.excel_db["Cabinet_Clean"].tolist())
274
 
275
- for cand in pdf_numbers:
276
- if cand in db_clean_keys:
277
- detected_cabinet = cand
278
- found_by_method = "number"
279
  break
280
 
281
  if detected_cabinet == "Не определен":
282
- unique_cabinets = self.excel_db["Cabinet"].unique()
283
- for cab_name in unique_cabinets:
284
- sub_names = [n.strip() for n in cab_name.split(',')]
285
- for sub_name in sub_names:
286
- if len(sub_name) < 5:
287
- continue
288
- if sub_name.lower() in all_pdf_text.lower():
289
- detected_cabinet = cab_name
290
- found_by_method = "name"
291
  break
292
- if found_by_method == "name":
293
  break
294
 
295
  if detected_cabinet == "Не определен":
296
- return "⚠️ Шкаф не опознан автоматически.\nВыберите шкаф из списка вручную.", None
297
 
298
- is_clean_search = (found_by_method == "number")
299
- remarks = self.get_remarks(detected_cabinet, is_clean_key=is_clean_search)
300
 
301
  if not remarks:
302
- return f"⚠️ Шкаф '{detected_cabinet}' найден, но замечаний в базе нет.", None
303
 
304
- processed_count = 0
305
- for file_path in file_paths:
306
- fname = os.path.basename(file_path)
307
  dtype = self.determine_doc_type(fname)
308
 
309
  tasks = []
@@ -314,72 +268,49 @@ class KDChecker:
314
 
315
  if tasks:
316
  checklist[fname] = list(dict.fromkeys(tasks))
317
- processed_count += 1
318
 
319
- pdf_title = detected_cabinet
320
  if is_manual:
321
- pdf_title += " (Ручной выбор)"
322
 
323
  try:
324
- pdf_path = self.create_pdf(pdf_title, checklist)
325
  except Exception as e:
326
- return f"❌ Ошибка создания PDF: {e}", None
327
 
328
  total = sum(len(v) for v in checklist.values())
329
- method_str = "Ручной выбор" if is_manual else (
330
- "По децимальному номеру" if is_clean_search else "По наименованию")
331
 
332
- result = f"✅ Чек-лист сформирован!\n\n"
333
  result += f"📂 Шкаф: {detected_cabinet}\n"
334
- result += f"🔍 Метод определения: {method_str}\n"
335
- result += f"📄 Обработано файлов: {processed_count}\n"
336
- result += f"🚩 Всего замечаний: {total}"
337
 
338
  return result, pdf_path
339
 
340
  def create_pdf(self, cabinet, data):
341
- """Создание PDF чек-листа"""
342
- fname = "CheckList_Result.pdf"
343
- path = os.path.join(tempfile.gettempdir(), fname)
344
  c = canvas.Canvas(path, pagesize=A4)
345
  width, height = A4
346
-
347
- # Шрифт
348
- font_name = 'Helvetica'
349
- font_path = "arial.ttf"
350
- if os.path.exists(font_path):
351
- try:
352
- pdfmetrics.registerFont(TTFont('Arial', font_path))
353
- font_name = 'Arial'
354
- except:
355
- pass
356
 
357
  y = height - 50
358
- c.setFont(font_name, 16)
 
359
 
360
- try:
361
- c.drawString(50, y, "ЧЕК-ЛИСТ ПРОВЕРКИ КД")
362
- except:
363
- c.setFont("Helvetica", 16)
364
- c.drawString(50, y, "CHECK-LIST KD")
365
-
366
  y -= 25
367
- c.setFont(font_name, 12)
368
- disp_cab = cabinet[:60] + "..." if len(cabinet) > 60 else cabinet
 
 
369
 
370
- try:
371
- c.drawString(50, y, f"Шкаф: {disp_cab}")
372
- except:
373
- c.setFont("Helvetica", 12)
374
- c.drawString(50, y, f"Cabinet: {disp_cab}")
375
-
376
- c.drawString(400, y, f"Дата: {datetime.now().strftime('%d.%m.%Y')}")
377
  y -= 20
378
  c.line(50, y, width - 50, y)
379
  y -= 30
380
 
381
  if not data:
382
- c.drawString(50, y, "Нет замечаний для загруженных файлов.")
383
  c.save()
384
  return path
385
 
@@ -387,134 +318,97 @@ class KDChecker:
387
  if y < 100:
388
  c.showPage()
389
  y = height - 50
390
- c.setFont(font_name, 12)
391
 
392
  c.setFillColor(colors.darkblue)
393
- c.setFont(font_name, 11)
394
- try:
395
- c.drawString(50, y, f"Файл: {filename}")
396
- except:
397
- c.setFont("Helvetica", 11)
398
- c.drawString(50, y, f"File: {filename}")
399
-
400
  c.setFillColor(colors.black)
401
- y -= 15
402
- c.setFont(font_name, 10)
403
 
404
  for task in tasks:
405
- if y < 80:
406
  c.showPage()
407
  y = height - 50
408
- c.setFont(font_name, 10)
409
 
410
  c.rect(50, y - 2, 8, 8, stroke=1, fill=0)
411
 
412
- # Разбивка текста на строки
413
- max_len = 90
414
- words = task.replace('\n', ' ').split(' ')
415
  lines = []
416
- cur_line = ""
417
  for w in words:
418
- if len(cur_line) + len(w) + 1 <= max_len:
419
- cur_line += w + " "
420
  else:
421
- lines.append(cur_line.strip())
422
- cur_line = w + " "
423
- if cur_line:
424
- lines.append(cur_line.strip())
425
 
426
- for i, line in enumerate(lines):
427
  if y < 40:
428
  c.showPage()
429
  y = height - 50
430
- c.setFont(font_name, 10)
431
- x_offset = 65 if i == 0 else 65
432
- try:
433
- c.drawString(x_offset, y, line)
434
- except:
435
- pass
436
  y -= 12
437
 
438
  y -= 5
439
 
440
- y -= 10
441
  c.setStrokeColor(colors.lightgrey)
442
  c.line(50, y, width - 50, y)
443
  c.setStrokeColor(colors.black)
444
- y -= 15
445
 
446
  c.save()
447
  return path
448
 
449
 
450
- def create_app():
451
- checker = KDChecker()
452
-
453
- with gr.Blocks(title="Генератор чек-листов КД", theme=gr.themes.Soft()) as app:
454
- gr.Markdown("# ✅ Генератор чек-листов КД")
455
- gr.Markdown("Автоматическая проверка конструкторской документации по базе знаний Excel.")
456
-
457
- with gr.Row():
458
- with gr.Column(scale=1):
459
- gr.Markdown("### 📁 1. База знаний")
460
- db_in = gr.File(
461
- label="Загрузить Excel (.xlsx)",
462
- file_types=[".xlsx", ".xls"],
463
- type="filepath"
464
- )
465
- manual_cab = gr.Dropdown(
466
- label="Или выберите шкаф вручную",
467
- choices=[],
468
- interactive=True,
469
- allow_custom_value=False
470
- )
471
- db_out = gr.Textbox(
472
- label="Статус загрузки",
473
- lines=8,
474
- max_lines=15
475
- )
476
-
477
- with gr.Column(scale=1):
478
- gr.Markdown("### 📄 2. Документация (PDF)")
479
- files_in = gr.File(
480
- label="Загрузить чертежи (PDF)",
481
- file_count="multiple",
482
- file_types=[".pdf"],
483
- type="filepath"
484
- )
485
- btn = gr.Button("🔍 Сформировать чек-лист", variant="primary", size="lg")
486
-
487
- with gr.Row():
488
- with gr.Column(scale=1):
489
- res_txt = gr.Textbox(label="Результат проверки", lines=8)
490
- with gr.Column(scale=1):
491
- res_pdf = gr.File(label="📥 Скачать PDF чек-лист")
492
-
493
- # Привязка событий
494
- db_in.change(
495
- fn=checker.load_excel_db,
496
- inputs=[db_in],
497
- outputs=[db_out, manual_cab]
498
- )
499
-
500
- btn.click(
501
- fn=checker.check_files,
502
- inputs=[files_in, manual_cab],
503
- outputs=[res_txt, res_pdf]
504
- )
505
-
506
- return app
507
 
 
 
 
 
 
 
508
 
509
- # ========== ЗАПУСК ПРИЛОЖЕНИЯ ==========
510
- app = create_app()
 
 
 
511
 
 
512
  if __name__ == "__main__":
513
- print(f"Starting with auth: {AUTH_USERNAME} / {'*' * len(AUTH_PASSWORD)}")
514
-
515
  app.launch(
516
  server_name="0.0.0.0",
517
  server_port=7860,
518
  auth=(AUTH_USERNAME, AUTH_PASSWORD),
519
- auth_message="Введите логин и пароль для доступа"
520
  )
 
18
 
19
  print("===== Application Startup =====")
20
  print(f"Time: {datetime.now()}")
21
+ print(f"Gradio version: {gr.__version__}")
22
+ print(f"Auth: {AUTH_USERNAME} / *****")
23
  print("================================")
24
 
25
 
 
28
  self.excel_db = pd.DataFrame()
29
  self.cabinet_list = []
30
  self.known_docs = ["Э3", "В4", "ПЭ3", "ВО", "ТЭ5", "СБ", "С5", "ОЛ", "Э1", "Э4", "Э7", "Д3", "Э6"]
31
+ print("[KDChecker] Initialized")
32
 
33
  def load_excel_db(self, excel_file):
34
  """Загрузка Excel базы знаний"""
35
+ log = [f"[{datetime.now().strftime('%H:%M:%S')}] Начало загрузки"]
36
 
 
37
  if excel_file is None:
38
+ return " Файл не выбран", gr.Dropdown.update(choices=[], value=None)
 
39
 
40
+ # Получаем путь к файлу
41
  if isinstance(excel_file, str):
42
  excel_path = excel_file
 
 
43
  else:
44
+ excel_path = excel_file.name if hasattr(excel_file, 'name') else str(excel_file)
45
 
46
+ log.append(f"Путь: {excel_path}")
47
+ log.append(f"Тип: {type(excel_file)}")
48
 
 
49
  if not os.path.exists(excel_path):
50
+ log.append(f" Файл не найден!")
51
+ return "\n".join(log), gr.Dropdown.update(choices=[], value=None)
52
 
53
  try:
54
+ file_size = os.path.getsize(excel_path)
55
+ log.append(f"Размер: {file_size} байт")
56
  except Exception as e:
57
+ log.append(f"Ошибка размера: {e}")
 
 
 
 
 
 
 
 
 
58
 
59
  all_data = []
60
+ sheets_info = []
61
 
62
  try:
63
+ log.append("Читаю Excel...")
64
  xls = pd.read_excel(excel_path, sheet_name=None, header=None, engine='openpyxl')
65
+ log.append(f"Листов найдено: {len(xls)}")
 
66
 
67
  for sheet_name, df_raw in xls.items():
68
+ header_row = -1
69
+ cab_col = -1
70
+ rem_col = -1
 
 
71
 
 
72
  for i in range(min(20, len(df_raw))):
73
+ row_vals = [str(x).lower().strip() for x in df_raw.iloc[i].values]
74
+ for idx, val in enumerate(row_vals):
 
 
75
  if "шкаф" in val or "cabinet" in val:
76
+ cab_col = idx
77
  if "примечание" in val or "remark" in val:
78
+ rem_col = idx
79
+ if cab_col != -1 and rem_col != -1:
80
+ header_row = i
 
 
 
81
  break
82
 
83
+ if header_row != -1:
84
+ df = pd.read_excel(excel_path, sheet_name=sheet_name, header=header_row, engine='openpyxl')
85
+ if cab_col < len(df.columns) and rem_col < len(df.columns):
86
+ df_sub = df.iloc[:, [cab_col, rem_col]]
87
+ df_sub.columns = ["Cabinet", "Remark"]
88
+ df_sub["Cabinet"] = df_sub["Cabinet"].ffill()
89
+ df_sub = df_sub.dropna(subset=["Remark"]).astype(str)
90
+ df_sub["Cabinet_Clean"] = df_sub["Cabinet"].apply(
 
 
91
  lambda x: x.strip().replace(" ", "").replace("\n", "").replace("\r", "")
92
  )
93
+ all_data.append(df_sub)
94
+ sheets_info.append(f"'{sheet_name}': {len(df_sub)}")
 
 
 
 
 
 
 
95
 
96
  if not all_data:
97
+ log.append(" Данные не найдены ни на одном листе")
98
+ return "\n".join(log), gr.Dropdown.update(choices=[], value=None)
99
 
100
  self.excel_db = pd.concat(all_data, ignore_index=True)
101
  self.cabinet_list = sorted(self.excel_db["Cabinet"].unique().tolist())
102
 
103
+ result = f" База загружена!\n\n"
104
+ result += f"📊 Записей: {len(self.excel_db)}\n"
105
+ result += f"🗄️ Шкафов: {len(self.cabinet_list)}\n"
106
+ result += f"📋 Листы: {', '.join(sheets_info)}"
 
 
 
107
 
108
+ return result, gr.Dropdown.update(choices=self.cabinet_list, value=None)
109
 
110
  except Exception as e:
111
+ log.append(f" ОШИБКА: {str(e)}")
112
  import traceback
113
  log.append(traceback.format_exc())
114
+ return "\n".join(log), gr.Dropdown.update(choices=[], value=None)
115
 
116
  def extract_text(self, pdf_path):
 
117
  try:
118
+ text = ""
119
  with pdfplumber.open(pdf_path) as pdf:
120
  for page in pdf.pages:
121
+ text += (page.extract_text() or "") + "\n"
122
+ return text
123
+ except:
 
124
  return ""
125
 
126
+ def find_decimal_numbers(self, text):
 
127
  pattern = r"(РЛТ|ЛДАР|ВНАР|ШТМ)[\s\.]*\d{1}[\s\.]*\d{3}[\s\.]*[А-ЯA-Z]{1,4}[\s\.]*\d{3}(-[\d]+)?"
128
  matches = []
129
+ for m in re.finditer(pattern, text):
130
+ clean = m.group(0).replace(" ", "").replace("\n", "")
131
+ if clean not in matches:
132
+ matches.append(clean)
133
  return matches
134
 
135
  def determine_doc_type(self, filename):
 
136
  fname = filename.upper()
137
  if "С2" in fname: return "С2"
138
  if "ПЭ3" in fname or "ПЕРЕЧЕНЬ" in fname: return "ПЭ3"
 
149
  if "Д3" in fname or "МОНТАЖ" in fname: return "Д3"
150
  return "UNKNOWN"
151
 
152
+ def get_remarks(self, cabinet_key, is_clean=True):
 
153
  if self.excel_db.empty:
154
  return {}
155
 
156
+ if is_clean:
157
  target = cabinet_key.replace(" ", "")
158
  mask = self.excel_db['Cabinet_Clean'].str.contains(re.escape(target), case=False, na=False)
159
  else:
 
164
  return {}
165
 
166
  parsed = {}
167
+ for remark in rows['Remark']:
168
+ text = str(remark)
169
+ text = re.sub(r'(\d+)\.([А-ЯA-Z])', r'\1. \2', text)
170
+ items = re.split(r'(?:^|\n)\s*(?=\d+[\.\)])', text)
171
 
172
  for item in items:
173
  if len(item) < 3:
174
  continue
175
  clean_item = item.strip()
176
+ no_num = re.sub(r'^\d+[\.\)]\s*', '', clean_item)
177
+ pattern = r'^(?:Документ\s+|В\s+)?([А-ЯA-Z0-9\s,\(\)\-]+?)(?:[\.\:\-]|\s+)(.*)'
178
+ match = re.match(pattern, no_num, re.IGNORECASE | re.DOTALL)
179
 
180
+ docs = []
181
  final_text = clean_item
182
 
183
  if match:
184
+ potential = match.group(1).upper()
185
+ cleaned = potential.replace("(", " ").replace(")", " ").replace(",", " ")
186
+ parts = cleaned.split()
187
+ valid = [p for p in parts if p in self.known_docs]
188
+ if valid:
189
+ docs = valid
190
  final_text = match.group(2).strip()
191
 
192
+ if not docs:
193
+ docs = ["ALL"]
194
 
195
+ for d in docs:
196
+ if d not in parsed:
197
+ parsed[d] = []
198
+ parsed[d].append(final_text)
199
 
200
  return parsed
201
 
202
  def check_files(self, files, manual_cabinet):
 
203
  if not files:
204
+ return "❌ Загрузите PDF файлы", None
205
  if self.excel_db.empty:
206
  return "❌ Сначала загрузите Excel базу!", None
207
 
208
+ # Получаем пути к файлам
 
 
 
 
 
209
  file_paths = []
210
  for f in files:
211
  if isinstance(f, str):
212
  file_paths.append(f)
 
 
213
  else:
214
+ file_paths.append(f.name if hasattr(f, 'name') else str(f))
215
+
216
+ checklist = {}
217
+ detected_cabinet = "Не определен"
218
+ method = ""
219
+ is_manual = False
220
 
221
  if manual_cabinet and manual_cabinet.strip():
222
  detected_cabinet = manual_cabinet
223
+ method = "manual"
224
  is_manual = True
225
  else:
226
+ all_text = ""
227
+ for fp in file_paths:
228
+ all_text += self.extract_text(fp) + "\n"
229
 
230
+ numbers = self.find_decimal_numbers(all_text)
231
+ db_keys = set(self.excel_db["Cabinet_Clean"].tolist())
232
 
233
+ for num in numbers:
234
+ if num in db_keys:
235
+ detected_cabinet = num
236
+ method = "number"
237
  break
238
 
239
  if detected_cabinet == "Не определен":
240
+ for cab in self.excel_db["Cabinet"].unique():
241
+ parts = [n.strip() for n in cab.split(',')]
242
+ for part in parts:
243
+ if len(part) >= 5 and part.lower() in all_text.lower():
244
+ detected_cabinet = cab
245
+ method = "name"
 
 
 
246
  break
247
+ if method == "name":
248
  break
249
 
250
  if detected_cabinet == "Не определен":
251
+ return "⚠️ Шкаф не определён. Выберите вручную из списка.", None
252
 
253
+ is_clean = (method == "number")
254
+ remarks = self.get_remarks(detected_cabinet, is_clean=is_clean)
255
 
256
  if not remarks:
257
+ return f"⚠️ Шкаф '{detected_cabinet}' найден, но замечаний нет.", None
258
 
259
+ for fp in file_paths:
260
+ fname = os.path.basename(fp)
 
261
  dtype = self.determine_doc_type(fname)
262
 
263
  tasks = []
 
268
 
269
  if tasks:
270
  checklist[fname] = list(dict.fromkeys(tasks))
 
271
 
272
+ title = detected_cabinet
273
  if is_manual:
274
+ title += " (ручной выбор)"
275
 
276
  try:
277
+ pdf_path = self.create_pdf(title, checklist)
278
  except Exception as e:
279
+ return f"❌ Ошибка PDF: {e}", None
280
 
281
  total = sum(len(v) for v in checklist.values())
282
+ method_str = "Ручной" if is_manual else ("По номеру" if is_clean else "По названию")
 
283
 
284
+ result = f"✅ Готово!\n\n"
285
  result += f"📂 Шкаф: {detected_cabinet}\n"
286
+ result += f"🔍 Метод: {method_str}\n"
287
+ result += f"📄 Файлов: {len(file_paths)}\n"
288
+ result += f"🚩 Замечаний: {total}"
289
 
290
  return result, pdf_path
291
 
292
  def create_pdf(self, cabinet, data):
293
+ path = os.path.join(tempfile.gettempdir(), "CheckList.pdf")
 
 
294
  c = canvas.Canvas(path, pagesize=A4)
295
  width, height = A4
296
+ font = 'Helvetica'
 
 
 
 
 
 
 
 
 
297
 
298
  y = height - 50
299
+ c.setFont(font, 16)
300
+ c.drawString(50, y, "CHECK-LIST KD")
301
 
 
 
 
 
 
 
302
  y -= 25
303
+ c.setFont(font, 12)
304
+ cab_display = cabinet[:50] + "..." if len(cabinet) > 50 else cabinet
305
+ c.drawString(50, y, f"Cabinet: {cab_display}")
306
+ c.drawString(400, y, f"Date: {datetime.now().strftime('%d.%m.%Y')}")
307
 
 
 
 
 
 
 
 
308
  y -= 20
309
  c.line(50, y, width - 50, y)
310
  y -= 30
311
 
312
  if not data:
313
+ c.drawString(50, y, "No remarks for loaded files.")
314
  c.save()
315
  return path
316
 
 
318
  if y < 100:
319
  c.showPage()
320
  y = height - 50
321
+ c.setFont(font, 12)
322
 
323
  c.setFillColor(colors.darkblue)
324
+ c.setFont(font, 11)
325
+ c.drawString(50, y, f"File: {filename}")
 
 
 
 
 
326
  c.setFillColor(colors.black)
327
+ y -= 18
328
+ c.setFont(font, 10)
329
 
330
  for task in tasks:
331
+ if y < 60:
332
  c.showPage()
333
  y = height - 50
334
+ c.setFont(font, 10)
335
 
336
  c.rect(50, y - 2, 8, 8, stroke=1, fill=0)
337
 
338
+ # Разбивка на строки
339
+ words = task.replace('\n', ' ').split()
 
340
  lines = []
341
+ line = ""
342
  for w in words:
343
+ if len(line) + len(w) < 85:
344
+ line += w + " "
345
  else:
346
+ lines.append(line.strip())
347
+ line = w + " "
348
+ if line:
349
+ lines.append(line.strip())
350
 
351
+ for i, ln in enumerate(lines):
352
  if y < 40:
353
  c.showPage()
354
  y = height - 50
355
+ c.setFont(font, 10)
356
+ c.drawString(65, y, ln)
 
 
 
 
357
  y -= 12
358
 
359
  y -= 5
360
 
361
+ y -= 15
362
  c.setStrokeColor(colors.lightgrey)
363
  c.line(50, y, width - 50, y)
364
  c.setStrokeColor(colors.black)
365
+ y -= 10
366
 
367
  c.save()
368
  return path
369
 
370
 
371
+ # ========== СОЗДАНИЕ ИНТЕРФЕЙСА ==========
372
+ checker = KDChecker()
373
+
374
+ with gr.Blocks(title="КД Checker") as app:
375
+ gr.Markdown("# ✅ Генератор чек-листов КД")
376
+ gr.Markdown("Загрузите Excel базу и PDF чертежи для формирования чек-листа.")
377
+
378
+ with gr.Row():
379
+ with gr.Column():
380
+ gr.Markdown("### 📁 1. База знаний (Excel)")
381
+ db_input = gr.File(label="Загрузить .xlsx файл", file_types=[".xlsx", ".xls"])
382
+ cabinet_dropdown = gr.Dropdown(label="Или выберите шкаф вручную", choices=[], interactive=True)
383
+ db_status = gr.Textbox(label="Статус", lines=6)
384
+
385
+ with gr.Column():
386
+ gr.Markdown("### 📄 2. Чертежи (PDF)")
387
+ pdf_input = gr.File(label="Загрузить PDF", file_count="multiple", file_types=[".pdf"])
388
+ run_btn = gr.Button("🔍 Сформировать чек-лист", variant="primary")
389
+
390
+ with gr.Row():
391
+ result_text = gr.Textbox(label="Результат", lines=6)
392
+ result_pdf = gr.File(label="📥 Скачать PDF")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
393
 
394
+ # События
395
+ db_input.change(
396
+ fn=checker.load_excel_db,
397
+ inputs=[db_input],
398
+ outputs=[db_status, cabinet_dropdown]
399
+ )
400
 
401
+ run_btn.click(
402
+ fn=checker.check_files,
403
+ inputs=[pdf_input, cabinet_dropdown],
404
+ outputs=[result_text, result_pdf]
405
+ )
406
 
407
+ # ========== ЗАПУСК ==========
408
  if __name__ == "__main__":
 
 
409
  app.launch(
410
  server_name="0.0.0.0",
411
  server_port=7860,
412
  auth=(AUTH_USERNAME, AUTH_PASSWORD),
413
+ auth_message="Введите логин и пароль"
414
  )