Update app.py

app.py

@@ -22,31 +22,28 @@ class KDChecker:
         log = [f"[{datetime.now().strftime('%H:%M:%S')}] Старт функции загрузки"]
         
         if excel_path is None:
-            return "Файл не передан (excel_path is None)", gr.update(choices=[], value=None)
+            return "Файл не передан (excel_path is None)", gr.Dropdown(choices=[], value=None)
 
         log.append(f"Путь к файлу: {excel_path}")
         
-        # Проверка размера файла (чтобы убедиться, что он загрузился)
         try:
             size = os.path.getsize(excel_path)
             log.append(f"Размер файла: {size} байт")
         except Exception as e:
             log.append(f"Ошибка доступа к файлу: {e}")
-            return "\n".join(log), gr.update(choices=[], value=None)
+            return "\n".join(log), gr.Dropdown(choices=[], value=None)
 
-        # Проверка наличия движка
         try:
             import openpyxl
             log.append("Библиотека openpyxl найдена.")
         except ImportError:
-            return "ОШИБКА: Библиотека openpyxl не установлена на сервере!", gr.update(choices=[], value=None)
+            return "ОШИБКА: Библиотека openpyxl не установлена на сервере!", gr.Dropdown(choices=[], value=None)
 
         all_data = []
         sheets_log = []
 
         try:
             log.append("Попытка чтения Excel через pandas...")
-            # Явно указываем движок
             xls = pd.read_excel(excel_path, sheet_name=None, header=None, engine='openpyxl')
             log.append(f"Файл прочитан. Найдены листы: {list(xls.keys())}")
 
@@ -71,36 +68,43 @@ class KDChecker:
 
                 if header_row_index != -1:
                     df = pd.read_excel(excel_path, sheet_name=sheet_name, header=header_row_index, engine='openpyxl')
-                    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)} стр.")
+                    # Проверяем, что индексы колонок существуют в этом df
+                    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:
+                        sheets_log.append(f"Лист '{sheet_name}': ошибка индексов колонок")
                 else:
                     sheets_log.append(f"Лист '{sheet_name}': заголовки не найдены")
 
             if not all_data:
                 log.append("Не найдены данные ни на одном листе.")
-                return "\n".join(log), gr.update(choices=[], value=None)
+                return "\n".join(log), gr.Dropdown(choices=[], value=None)
 
             self.excel_db = pd.concat(all_data, ignore_index=True)
             self.cabinet_list = sorted(self.excel_db["Cabinet"].unique().tolist())
             
             log.append("Успешно объединили данные.")
             msg = f"✅ База знаний загружена успешно!\nВсего записей: {len(self.excel_db)}\nОбработаны листы: {', '.join(sheets_log)}"
-            return msg, gr.update(choices=self.cabinet_list, value=None, interactive=True)
+            
+            # Возвращаем новый Dropdown с обновленным списком
+            return msg, gr.Dropdown(choices=self.cabinet_list, value=None, interactive=True)
 
         except Exception as e:
-            # Возвращаем полный лог ошибки
             log.append(f"КРИТИЧЕСКАЯ ОШИБКА: {str(e)}")
-            return "\n".join(log), gr.update(choices=[], value=None)
+            import traceback
+            log.append(traceback.format_exc())
+            return "\n".join(log), gr.Dropdown(choices=[], value=None)
 
     def extract_text(self, pdf_path):
         try:
@@ -239,7 +243,6 @@ class KDChecker:
         pdf_title = detected_cabinet
         if is_manual: pdf_title += " (Выбор вручную)"
 
-        # Создание PDF с защитой от отсутствия шрифта
         try:
             pdf = self.create_pdf(pdf_title, checklist)
         except Exception as e:
@@ -258,21 +261,19 @@ class KDChecker:
         form = c.acroForm
         width, height = A4
 
-        # Ищем шрифт на сервере
         font_name = 'Helvetica'
-        font_path = "arial.ttf" # Ожидаем в корне
+        font_path = "arial.ttf"
 
         if os.path.exists(font_path):
             try:
                 pdfmetrics.registerFont(TTFont('Arial', font_path))
                 font_name = 'Arial'
             except:
-                pass # Если шрифт битый, останется Helvetica
+                pass
 
         y = height - 50
         c.setFont(font_name, 16)
         
-        # Безопасная печать заголовка
         try:
             c.drawString(50, y, f"ЧЕК-ЛИСТ ПРОВЕРКИ КД")
         except:
@@ -352,7 +353,7 @@ class KDChecker:
                         try:
                             c.drawString(65, text_start_y, l.strip())
                         except:
-                            pass # Игнорируем непечатаемые символы
+                            pass
                         text_start_y -= 12
 
                 y = text_start_y - 8
@@ -376,16 +377,18 @@ def create_app():
         with gr.Row():
             with gr.Column():
                 gr.Markdown("### 1. База знаний")
-                # type="filepath" ОБЯЗАТЕЛЬНО ДЛЯ СЕРВЕРА
+                
                 db_in = gr.File(label="Загрузить Excel (.xlsx)", type="filepath")
-                manual_cab = gr.Dropdown(label="Или выберите шкаф-аналог вручную", choices=[], interactive=True)
+                
+                # Инициализируем с choices=None, чтобы Gradio не ругался на пустой список при старте
+                manual_cab = gr.Dropdown(label="Или выберите шкаф-аналог вручную", choices=None, 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" ОБЯЗАТЕЛЬНО ДЛЯ СЕРВЕРА
                 files_in = gr.File(label="Загрузить чертежи", file_count="multiple", type="filepath")
                 btn = gr.Button("Сформировать чек-лист", variant="primary")