MrSimple07 commited on
Commit
292f192
·
1 Parent(s): ab99142

added to UI restart button

Browse files
Files changed (1) hide show
  1. app.py +40 -14
app.py CHANGED
@@ -465,13 +465,22 @@ Rerank Top K: {retrieval_params['rerank_top_k']}"""
465
  type="filepath"
466
  )
467
 
468
- upload_btn = gr.Button("Загрузить и обработать", variant="primary", size="lg")
 
 
469
 
470
  upload_status = gr.Textbox(
471
  label="Статус загрузки",
472
  value="Ожидание загрузки файла...",
473
  interactive=False,
474
- lines=3
 
 
 
 
 
 
 
475
  )
476
 
477
  with gr.Column(scale=1):
@@ -481,10 +490,10 @@ Rerank Top K: {retrieval_params['rerank_top_k']}"""
481
  **Таблицы (Excel → JSON):**
482
  - Формат: .xlsx или .xls
483
  - Обязательные колонки:
484
- - Номер таблицы
485
- - Обозначение документа
486
- - Раздел документа
487
- - Название таблицы
488
 
489
  **Изображения (Excel → CSV):**
490
  - Формат: .xlsx, .xls или .csv
@@ -494,15 +503,32 @@ Rerank Top K: {retrieval_params['rerank_top_k']}"""
494
  - Формат: .json
495
  - Структурированные данные
496
 
497
- ⚠️ **Важно:** После загрузки необходимо перезапустить систему для применения изменений!
 
 
 
 
 
 
498
  """)
499
-
500
- upload_btn.click(
501
- fn=process_uploaded_file,
502
- inputs=[file_upload, file_type_radio],
503
- outputs=[upload_status]
504
- )
505
-
 
 
 
 
 
 
 
 
 
 
 
506
  switch_btn.click(
507
  fn=switch_model_func,
508
  inputs=[model_dropdown],
 
465
  type="filepath"
466
  )
467
 
468
+ with gr.Row():
469
+ upload_btn = gr.Button("📤 Загрузить и обработать", variant="primary", size="lg")
470
+ restart_btn = gr.Button("🔄 Перезапустить систему", variant="secondary", size="lg")
471
 
472
  upload_status = gr.Textbox(
473
  label="Статус загрузки",
474
  value="Ожидание загрузки файла...",
475
  interactive=False,
476
+ lines=5
477
+ )
478
+
479
+ restart_status = gr.Textbox(
480
+ label="Статус перезапуска",
481
+ value="Система готова к работе",
482
+ interactive=False,
483
+ lines=2
484
  )
485
 
486
  with gr.Column(scale=1):
 
490
  **Таблицы (Excel → JSON):**
491
  - Формат: .xlsx или .xls
492
  - Обязательные колонки:
493
+ - Номер таблицы
494
+ - Обозначение документа
495
+ - Раздел документа
496
+ - Название таблицы
497
 
498
  **Изображения (Excel → CSV):**
499
  - Формат: .xlsx, .xls или .csv
 
503
  - Формат: .json
504
  - Структурированные данные
505
 
506
+ ### Процесс загрузки:
507
+ 1. Выберите тип документа
508
+ 2. Загрузите файл
509
+ 3. Дождитесь обработки
510
+ 4. Нажмите "Перезапустить систему"
511
+
512
+ ⚠️ **Важно:** Перезапуск может занять несколько минут!
513
  """)
514
+
515
+ # Display current chunks info
516
+ current_chunks_display = gr.HTML(
517
+ label="Текущие документы",
518
+ value=create_chunks_display_html(chunk_info) if chunk_info else "<div>Нет данных</div>"
519
+ )
520
+
521
+ upload_btn.click(
522
+ fn=process_uploaded_file,
523
+ inputs=[file_upload, file_type_radio],
524
+ outputs=[upload_status]
525
+ )
526
+
527
+ restart_btn.click(
528
+ fn=restart_system,
529
+ inputs=[],
530
+ outputs=[restart_status, current_chunks_display]
531
+ )
532
  switch_btn.click(
533
  fn=switch_model_func,
534
  inputs=[model_dropdown],