fixed chunking showring problem

app.py

@@ -109,42 +109,22 @@ def switch_model(model_name, vector_index):
         return None, f"❌ {error_msg}"
 
 def main_answer_question(question):
-    from index_retriever import rerank_nodes
     global query_engine, reranker, current_model, chunks_df
     if not question.strip():
-        return ("<div style='color: black;'>Пожалуйста, введите вопрос</div>", 
-                "<div style='color: black;'>Источники появятся после обработки запроса</div>")
-    
+        return ("<div style='color: black;'>Пожалуйста, введите вопрос</div>",
+                "<div style='color: black;'>Источники появятся после обработки запроса</div>",
+                "<div style='color: black;'>Чанки появятся после обработки запроса</div>")
+
     try:
-        # Get response and retrieve nodes for reranking display
-        response = query_engine.query(question)
-        
-        # Get source nodes from response
-        source_nodes = response.source_nodes if hasattr(response, 'source_nodes') else []
-        
-        # Rerank the nodes
-        reranked_nodes = rerank_nodes(question, source_nodes, reranker, top_k=10)
-        
-        # Create chunk info for display
-        chunk_info = []
-        for node in reranked_nodes:
-            chunk_info.append({
-                'document_id': node.metadata.get('document_id', 'unknown'),
-                'section_id': node.metadata.get('section_id', 'unknown'), 
-                'chunk_size': len(node.text),
-                'chunk_text': node.text
-            })
-        
-        # Format response
-        answer_html = f"<div style='color: black; padding: 15px; background-color: #f8f9fa; border-radius: 8px;'>{str(response)}</div>"
-        sources_html = create_chunks_display_html(chunk_info)
-        
-        return answer_html, sources_html
-        
+        # Call the answer_question function which returns 3 values
+        answer_html, sources_html, chunks_html = answer_question(question, query_engine, reranker, current_model, chunks_df)
+        return answer_html, sources_html, chunks_html
+
     except Exception as e:
         log_message(f"Ошибка при ответе на вопрос: {str(e)}")
-        return (f"<div style='color: red;'>Ошибка: {str(e)}</div>", 
-                "<div style='color: black;'>Источники недоступны из-за ошибки</div>")
+        return (f"<div style='color: red;'>Ошибка: {str(e)}</div>",
+                "<div style='color: black;'>Источники недоступны из-за ошибки</div>",
+                "<div style='color: black;'>Чанки недоступны из-за ошибки</div>")
 
 
 
@@ -208,6 +188,12 @@ def create_demo_interface(answer_question_func, switch_model_func, current_model
                         label="",
                         value="<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Здесь появятся релевантные чанки...</div>",
                     )
+
+                with gr.Column(scale=1):
+                    chunks_output = gr.HTML(
+                        label="Релевантные чанки",
+                        value="<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Здесь появятся релевантные чанки...</div>",
+                    )
             
             switch_btn.click(
                 fn=switch_model_func,
@@ -218,13 +204,13 @@ def create_demo_interface(answer_question_func, switch_model_func, current_model
             ask_btn.click(
                 fn=answer_question_func,
                 inputs=[question_input],
-                outputs=[answer_output, sources_output]
+                outputs=[answer_output, sources_output, chunks_output]
             )
             
             question_input.submit(
                 fn=answer_question_func,
                 inputs=[question_input],
-                outputs=[answer_output, sources_output]
+                outputs=[answer_output, sources_output, chunks_output]
             )
     return demo
 
@@ -237,7 +223,10 @@ current_model = DEFAULT_MODEL
 
 def main_answer_question(question):
     global query_engine, reranker, current_model, chunks_df
-    return answer_question(question, query_engine, reranker, current_model, chunks_df)
+    answer_html, sources_html, chunks_html = answer_question(
+        question, query_engine, reranker, current_model, chunks_df
+    )
+    return answer_html, sources_html, chunks_html
 
 def main_switch_model(model_name):
     global query_engine, vector_index, current_model

utils.py

@@ -186,7 +186,20 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         </div>
         </div>"""
         
-        return answer_with_time, sources_html
+        # Релевантные чанки (text snippets)
+        chunk_info = []
+        for node in reranked_nodes:
+            chunk_info.append({
+                'document_id': node.metadata.get('document_id', 'unknown'),
+                'section_id': node.metadata.get('section_id', 'unknown'), 
+                'chunk_size': len(node.text),
+                'chunk_text': node.text
+            })
+        from app import create_chunks_display_html
+        chunks_html = create_chunks_display_html(chunk_info)
+
+
+        return answer_with_time, sources_html, chunks_html
         
     except Exception as e:
         log_message(f"Ошибка обработки вопроса: {str(e)}")