Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -18,7 +18,8 @@ AUTH_PASSWORD = "12345"
|
|
| 18 |
|
| 19 |
print("===== Application Startup =====")
|
| 20 |
print(f"Time: {datetime.now()}")
|
| 21 |
-
print(f"
|
|
|
|
| 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')}]
|
| 35 |
|
| 36 |
-
# Проверка входных данных
|
| 37 |
if excel_file is None:
|
| 38 |
-
|
| 39 |
-
return "\n".join(log), gr.update(choices=[], value=None)
|
| 40 |
|
| 41 |
-
#
|
| 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"
|
| 50 |
-
log.append(f"
|
| 51 |
|
| 52 |
-
# Проверка существования файла
|
| 53 |
if not os.path.exists(excel_path):
|
| 54 |
-
log.append(f"
|
| 55 |
-
return "\n".join(log), gr.update(choices=[], value=None)
|
| 56 |
|
| 57 |
try:
|
| 58 |
-
|
| 59 |
-
log.append(f"
|
| 60 |
except Exception as e:
|
| 61 |
-
log.append(f"
|
| 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 |
-
|
| 74 |
|
| 75 |
try:
|
| 76 |
-
log.append("
|
| 77 |
xls = pd.read_excel(excel_path, sheet_name=None, header=None, engine='openpyxl')
|
| 78 |
-
log.append(f"
|
| 79 |
-
log.append(f"Имена листов: {list(xls.keys())}")
|
| 80 |
|
| 81 |
for sheet_name, df_raw in xls.items():
|
| 82 |
-
|
| 83 |
-
|
| 84 |
-
|
| 85 |
-
cab_col_idx = -1
|
| 86 |
-
rem_col_idx = -1
|
| 87 |
|
| 88 |
-
# Поиск заголовков в первых 20 строках
|
| 89 |
for i in range(min(20, len(df_raw))):
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
r_idx = -1
|
| 93 |
-
for idx, val in enumerate(row_values):
|
| 94 |
if "шкаф" in val or "cabinet" in val:
|
| 95 |
-
|
| 96 |
if "примечание" in val or "remark" in val:
|
| 97 |
-
|
| 98 |
-
if
|
| 99 |
-
|
| 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
|
| 106 |
-
df = pd.read_excel(excel_path, sheet_name=sheet_name, header=
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
| 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(
|
| 118 |
-
|
| 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 |
-
|
| 135 |
-
|
| 136 |
-
|
| 137 |
-
|
| 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
|
| 143 |
|
| 144 |
except Exception as e:
|
| 145 |
-
log.append(f"
|
| 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 |
-
|
| 154 |
with pdfplumber.open(pdf_path) as pdf:
|
| 155 |
for page in pdf.pages:
|
| 156 |
-
|
| 157 |
-
return
|
| 158 |
-
except
|
| 159 |
-
print(f"[extract_text] Error: {e}")
|
| 160 |
return ""
|
| 161 |
|
| 162 |
-
def
|
| 163 |
-
"""Поиск децимальных номеров в тексте"""
|
| 164 |
pattern = r"(РЛТ|ЛДАР|ВНАР|ШТМ)[\s\.]*\d{1}[\s\.]*\d{3}[\s\.]*[А-ЯA-Z]{1,4}[\s\.]*\d{3}(-[\d]+)?"
|
| 165 |
matches = []
|
| 166 |
-
for
|
| 167 |
-
|
| 168 |
-
if
|
| 169 |
-
matches.append(
|
| 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,
|
| 191 |
-
"""Получение замечаний для шкафа"""
|
| 192 |
if self.excel_db.empty:
|
| 193 |
return {}
|
| 194 |
|
| 195 |
-
if
|
| 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
|
| 207 |
-
|
| 208 |
-
|
| 209 |
-
items = re.split(r'(?:^|\n)\s*(?=\d+[\.\)])',
|
| 210 |
|
| 211 |
for item in items:
|
| 212 |
if len(item) < 3:
|
| 213 |
continue
|
| 214 |
clean_item = item.strip()
|
| 215 |
-
|
| 216 |
-
|
| 217 |
-
match = re.match(
|
| 218 |
|
| 219 |
-
|
| 220 |
final_text = clean_item
|
| 221 |
|
| 222 |
if match:
|
| 223 |
-
|
| 224 |
-
|
| 225 |
-
parts =
|
| 226 |
-
|
| 227 |
-
if
|
| 228 |
-
|
| 229 |
final_text = match.group(2).strip()
|
| 230 |
|
| 231 |
-
if not
|
| 232 |
-
|
| 233 |
|
| 234 |
-
for
|
| 235 |
-
if
|
| 236 |
-
parsed[
|
| 237 |
-
parsed[
|
| 238 |
|
| 239 |
return parsed
|
| 240 |
|
| 241 |
def check_files(self, files, manual_cabinet):
|
| 242 |
-
"""Проверка PDF файлов"""
|
| 243 |
if not files:
|
| 244 |
-
return "❌
|
| 245 |
if self.excel_db.empty:
|
| 246 |
return "❌ Сначала загрузите Excel базу!", None
|
| 247 |
|
| 248 |
-
|
| 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 |
-
|
| 266 |
is_manual = True
|
| 267 |
else:
|
| 268 |
-
|
| 269 |
-
for
|
| 270 |
-
|
| 271 |
|
| 272 |
-
|
| 273 |
-
|
| 274 |
|
| 275 |
-
for
|
| 276 |
-
if
|
| 277 |
-
detected_cabinet =
|
| 278 |
-
|
| 279 |
break
|
| 280 |
|
| 281 |
if detected_cabinet == "Не определен":
|
| 282 |
-
|
| 283 |
-
|
| 284 |
-
|
| 285 |
-
|
| 286 |
-
|
| 287 |
-
|
| 288 |
-
if sub_name.lower() in all_pdf_text.lower():
|
| 289 |
-
detected_cabinet = cab_name
|
| 290 |
-
found_by_method = "name"
|
| 291 |
break
|
| 292 |
-
if
|
| 293 |
break
|
| 294 |
|
| 295 |
if detected_cabinet == "Не определен":
|
| 296 |
-
return "⚠️ Шкаф не
|
| 297 |
|
| 298 |
-
|
| 299 |
-
remarks = self.get_remarks(detected_cabinet,
|
| 300 |
|
| 301 |
if not remarks:
|
| 302 |
-
return f"⚠️ Шкаф '{detected_cabinet}' найден, но замечаний
|
| 303 |
|
| 304 |
-
|
| 305 |
-
|
| 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 |
-
|
| 320 |
if is_manual:
|
| 321 |
-
|
| 322 |
|
| 323 |
try:
|
| 324 |
-
pdf_path = self.create_pdf(
|
| 325 |
except Exception as e:
|
| 326 |
-
return f"❌ Ошибка
|
| 327 |
|
| 328 |
total = sum(len(v) for v in checklist.values())
|
| 329 |
-
method_str = "Ручной
|
| 330 |
-
"По децимальному номеру" if is_clean_search else "По наименованию")
|
| 331 |
|
| 332 |
-
result = f"✅
|
| 333 |
result += f"📂 Шкаф: {detected_cabinet}\n"
|
| 334 |
-
result += f"🔍
|
| 335 |
-
result += f"📄
|
| 336 |
-
result += f"🚩
|
| 337 |
|
| 338 |
return result, pdf_path
|
| 339 |
|
| 340 |
def create_pdf(self, cabinet, data):
|
| 341 |
-
|
| 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(
|
|
|
|
| 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(
|
| 368 |
-
|
|
|
|
|
|
|
| 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(
|
| 391 |
|
| 392 |
c.setFillColor(colors.darkblue)
|
| 393 |
-
c.setFont(
|
| 394 |
-
|
| 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 -=
|
| 402 |
-
c.setFont(
|
| 403 |
|
| 404 |
for task in tasks:
|
| 405 |
-
if y <
|
| 406 |
c.showPage()
|
| 407 |
y = height - 50
|
| 408 |
-
c.setFont(
|
| 409 |
|
| 410 |
c.rect(50, y - 2, 8, 8, stroke=1, fill=0)
|
| 411 |
|
| 412 |
-
# Разбивка
|
| 413 |
-
|
| 414 |
-
words = task.replace('\n', ' ').split(' ')
|
| 415 |
lines = []
|
| 416 |
-
|
| 417 |
for w in words:
|
| 418 |
-
if len(
|
| 419 |
-
|
| 420 |
else:
|
| 421 |
-
lines.append(
|
| 422 |
-
|
| 423 |
-
if
|
| 424 |
-
lines.append(
|
| 425 |
|
| 426 |
-
for i,
|
| 427 |
if y < 40:
|
| 428 |
c.showPage()
|
| 429 |
y = height - 50
|
| 430 |
-
c.setFont(
|
| 431 |
-
|
| 432 |
-
try:
|
| 433 |
-
c.drawString(x_offset, y, line)
|
| 434 |
-
except:
|
| 435 |
-
pass
|
| 436 |
y -= 12
|
| 437 |
|
| 438 |
y -= 5
|
| 439 |
|
| 440 |
-
y -=
|
| 441 |
c.setStrokeColor(colors.lightgrey)
|
| 442 |
c.line(50, y, width - 50, y)
|
| 443 |
c.setStrokeColor(colors.black)
|
| 444 |
-
y -=
|
| 445 |
|
| 446 |
c.save()
|
| 447 |
return path
|
| 448 |
|
| 449 |
|
| 450 |
-
|
| 451 |
-
|
| 452 |
-
|
| 453 |
-
|
| 454 |
-
|
| 455 |
-
|
| 456 |
-
|
| 457 |
-
|
| 458 |
-
|
| 459 |
-
|
| 460 |
-
|
| 461 |
-
|
| 462 |
-
|
| 463 |
-
|
| 464 |
-
|
| 465 |
-
|
| 466 |
-
|
| 467 |
-
|
| 468 |
-
|
| 469 |
-
|
| 470 |
-
|
| 471 |
-
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
)
|