new code for showing chunks

app.py

@@ -51,11 +51,13 @@ def initialize_system(repo_id, hf_token, download_dir, chunks_filename=None,
         
         all_documents = []
         chunks_df = None
+        chunk_info = []
         
         if use_json_instead_csv and json_files_dir:
             log_message("Используем JSON файлы вместо CSV")
-            json_documents = load_json_documents(repo_id, hf_token, json_files_dir, download_dir)
+            json_documents, json_chunk_info = load_json_documents(repo_id, hf_token, json_files_dir, download_dir)
             all_documents.extend(json_documents)
+            chunk_info.extend(json_chunk_info)
         else:
             if chunks_filename:
                 log_message("Загружаем данные из CSV")
@@ -78,11 +80,11 @@ def initialize_system(repo_id, hf_token, download_dir, chunks_filename=None,
         query_engine = create_query_engine(vector_index)
         
         log_message(f"Система успешно инициализирована")
-        return query_engine, chunks_df, reranker, vector_index
+        return query_engine, chunks_df, reranker, vector_index, chunk_info
         
     except Exception as e:
         log_message(f"Ошибка инициализации: {str(e)}")
-        return None, None, None, None
+        return None, None, None, None, []
 
 def switch_model(model_name, vector_index):
     from llama_index.core import Settings
@@ -148,6 +150,8 @@ def create_demo_interface(answer_question_func, switch_model_func, current_model
                             "О чем этот рисунок: ГОСТ Р 50.04.07-2022 Приложение Л. Л.1.5 Рисунок Л.2",
                             "Л.9 Формула в ГОСТ Р 50.04.07 - 2022 что и о чем там?", 
                             "Какой стандарт устанавливает порядок признания протоколов испытаний продукции в области использования атомной энергии?",
+                            "Кто несет ответственность за организацию и проведение признания протоколов испытаний продукции?",
+                            "В каких случаях могут быть признаны протоколы испытаний, проведенные лабораториями?",
                         ],
                         inputs=question_input
                     )

documents_prep.py

@@ -226,13 +226,13 @@ def load_json_documents(repo_id, hf_token, json_files_dir, download_dir):
         log_message(f"Всего создано {len(all_documents)} исходных документов")
         log_message(f"После chunking получено {len(chunked_documents)} чанков")
         
-        # Return only chunked_documents, not the tuple
-        return chunked_documents
+        return chunked_documents, chunk_info
         
     except Exception as e:
         log_message(f"Ошибка загрузки JSON документов: {str(e)}")
-        return []
-
+        return [], []
+    
+    
 def extract_section_title(section_text):
     if not section_text.strip():
         return ""