Update app.py

app.py

@@ -36,7 +36,7 @@ class KDChecker:
             import openpyxl
             log.append("Библиотека openpyxl найдена.")
         except ImportError:
-            return "ОШИБКА: Библиотека openpyxl не установлена на сервере!", gr.update(choices=[], value=None)
+            return "ОШИБКА: Библиотека openpyxl не установлена!", gr.update(choices=[], value=None)
 
         all_data = []
         sheets_log = []
@@ -56,8 +56,10 @@ class KDChecker:
                     c_idx = -1
                     r_idx = -1
                     for idx, val in enumerate(row_values):
-                        if "шкаф" in val or "cabinet" in val: c_idx = idx
-                        if "примечание" in val or "remark" in val: r_idx = idx
+                        if "шкаф" in val or "cabinet" in val: 
+                            c_idx = idx
+                        if "примечание" in val or "remark" in val: 
+                            r_idx = idx
                     if c_idx != -1 and r_idx != -1:
                         header_row_index = i
                         cab_col_idx = c_idx
@@ -69,14 +71,11 @@ class KDChecker:
                     if cab_col_idx < len(df.columns) and rem_col_idx < len(df.columns):
                         df_subset = df.iloc[:, [cab_col_idx, rem_col_idx]]
                         df_subset.columns = ["Cabinet", "Remark"]
-
                         df_subset["Cabinet"] = df_subset["Cabinet"].ffill()
                         df_subset = df_subset.dropna(subset=["Remark"]).astype(str)
-
                         df_subset["Cabinet_Clean"] = df_subset["Cabinet"].apply(
                             lambda x: x.strip().replace(" ", "").replace("\n", "").replace("\r", "")
                         )
-
                         all_data.append(df_subset)
                         sheets_log.append(f"Лист '{sheet_name}': {len(df_subset)} стр.")
                     else:
@@ -94,7 +93,6 @@ class KDChecker:
             log.append("Успешно объединили данные.")
             msg = f"✅ База знаний загружена успешно!\nВсего записей: {len(self.excel_db)}\nОбработаны листы: {', '.join(sheets_log)}"
             
-            # ИЗМЕНЕНО: gr.update() вместо gr.Dropdown()
             return msg, gr.update(choices=self.cabinet_list, value=None)
 
         except Exception as e:
@@ -140,7 +138,8 @@ class KDChecker:
         return "UNKNOWN"
 
     def get_remarks(self, cabinet_key, is_clean_key=True):
-        if self.excel_db.empty: return {}
+        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)
@@ -148,14 +147,17 @@ class KDChecker:
             mask = self.excel_db['Cabinet'] == cabinet_key
 
         rows = self.excel_db[mask]
-        if rows.empty: return {}
+        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
+                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+)(.*)'
@@ -170,15 +172,19 @@ class KDChecker:
                     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] = []
+                    if doc not in parsed:
+                        parsed[doc] = []
                     parsed[doc].append(final_text)
         return parsed
 
     def check_files(self, files, manual_cabinet):
-        if not files: return "Файлы не загружены", None
-        if self.excel_db.empty: return "Сначала загрузите Excel базу!", None
+        if not files:
+            return "Файлы не загружены", None
+        if self.excel_db.empty:
+            return "Сначала загрузите Excel базу!", None
 
         checklist = {}
         detected_cabinet = "Не определен"
@@ -208,15 +214,17 @@ class KDChecker:
                 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 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
+                    if found_by_method == "name":
+                        break
 
         if detected_cabinet == "Не определен":
-            return f"⚠️ Шкаф не опознан автоматически.\nСовет: Выберите похожий шкаф из выпадающего списка вручную.", None
+            return "⚠️ Шкаф не опознан автоматически.\nСовет: Выберите шкаф из выпадающего списка вручную.", None
 
         is_clean_search = (found_by_method == "number")
         remarks = self.get_remarks(detected_cabinet, is_clean_key=is_clean_search)
@@ -230,15 +238,18 @@ class KDChecker:
             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 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
 
         pdf_title = detected_cabinet
-        if is_manual: pdf_title += " (Выбор вручную)"
+        if is_manual:
+            pdf_title += " (Выбор вручную)"
 
         try:
             pdf = self.create_pdf(pdf_title, checklist)
@@ -252,7 +263,7 @@ class KDChecker:
         return f"✅ Готово!\n\n📂 Шкаф: {detected_cabinet}\n🔍 Метод: {method_str}\n📄 Обработано файлов: {processed_count}\n🚩 Всего замечаний: {total}", pdf
 
     def create_pdf(self, cabinet, data):
-        fname = f"CheckList_Result.pdf"
+        fname = "CheckList_Result.pdf"
         path = os.path.join(tempfile.gettempdir(), fname)
         c = canvas.Canvas(path, pagesize=A4)
         form = c.acroForm
@@ -272,11 +283,11 @@ class KDChecker:
         c.setFont(font_name, 16)
         
         try:
-            c.drawString(50, y, f"ЧЕК-ЛИСТ ПРОВЕРКИ КД")
+            c.drawString(50, y, "ЧЕК-ЛИСТ ПРОВЕРКИ КД")
         except:
-             c.setFont("Helvetica", 16)
-             c.drawString(50, y, "CHECK-LIST")
-             c.setFont(font_name, 16)
+            c.setFont("Helvetica", 16)
+            c.drawString(50, y, "CHECK-LIST")
+            c.setFont(font_name, 16)
 
         y -= 25
         c.setFont(font_name, 12)
@@ -285,9 +296,9 @@ class KDChecker:
         try:
             c.drawString(50, y, f"Шкаф: {disp_cab}")
         except:
-             c.setFont("Helvetica", 12)
-             c.drawString(50, y, "Cabinet name error (font)")
-             c.setFont(font_name, 12)
+            c.setFont("Helvetica", 12)
+            c.drawString(50, y, "Cabinet name error (font)")
+            c.setFont(font_name, 12)
 
         c.drawString(400, y, f"Дата: {datetime.now().strftime('%d.%m.%Y')}")
         y -= 20
@@ -301,7 +312,10 @@ class KDChecker:
 
         cb_id = 0
         for filename, tasks in data.items():
-            if y < 100: c.showPage(); y = height - 50; c.setFont(font_name, 12)
+            if y < 100:
+                c.showPage()
+                y = height - 50
+                c.setFont(font_name, 12)
 
             c.setFillColor(colors.darkblue)
             c.setFont(font_name, 11)
@@ -368,6 +382,7 @@ class KDChecker:
 
 def create_app():
     checker = KDChecker()
+    
     with gr.Blocks(title="Генератор чек-листов КД") as app:
         gr.Markdown("# ✅ Генератор чек-листов КД")
         gr.Markdown("Автоматическая проверка конструкторской документации по базе знаний Excel.")
@@ -375,18 +390,13 @@ def create_app():
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### 1. База знаний")
-                
                 db_in = gr.File(label="Загрузить Excel (.xlsx)", type="filepath")
-                
-                # ИЗМЕНЕНО: choices=[] вместо choices=None
                 manual_cab = gr.Dropdown(
                     label="Или выберите шкаф-аналог вручную", 
                     choices=[], 
                     interactive=True
                 )
-                
-                db_out = gr.Textbox(label="Статус загрузки", lines=8)
-
+                db_out = gr.Textbox(label="Статус загрузки", lines=6)
                 db_in.upload(checker.load_excel_db, inputs=[db_in], outputs=[db_out, manual_cab])
 
             with gr.Column():
@@ -395,14 +405,21 @@ def create_app():
                 btn = gr.Button("Сформировать чек-лист", variant="primary")
 
         with gr.Row():
-            res_txt = gr.Textbox(label="Результат проверки", lines=8)
+            res_txt = gr.Textbox(label="Результат проверки", lines=6)
             res_pdf = gr.File(label="Скачать PDF чек-лист")
 
         btn.click(checker.check_files, inputs=[files_in, manual_cab], outputs=[res_txt, res_pdf])
+    
     return app
 
 
+# Создаём приложение
 app = create_app()
 
+# Запуск для Hugging Face Spaces
 if __name__ == "__main__":
-    app.launch(auth=("admin", "12345"))
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )