new code for showing chunks

app.py

@@ -13,21 +13,21 @@ from config import (
 
 def create_chunks_display_html(chunk_info):
     if not chunk_info:
-        return "<div style='padding: 20px; text-align: center;'>Нет данных о чанках</div>"
+        return "<div style='padding: 20px; text-align: center; color: black;'>Нет данных о чанках</div>"
     
-    html = "<div style='max-height: 500px; overflow-y: auto; padding: 10px;'>"
-    html += f"<h4>Всего чанков: {len(chunk_info)}</h4>"
+    html = "<div style='max-height: 500px; overflow-y: auto; padding: 10px; color: black;'>"
+    html += f"<h4 style='color: black;'>Найдено релевантных чанков: {len(chunk_info)}</h4>"
     
     for i, chunk in enumerate(chunk_info):
         bg_color = "#f8f9fa" if i % 2 == 0 else "#e9ecef"
         html += f"""
-        <div style='background-color: {bg_color}; padding: 10px; margin: 5px 0; border-radius: 5px; border-left: 4px solid #007bff;'>
+        <div style='background-color: {bg_color}; padding: 10px; margin: 5px 0; border-radius: 5px; border-left: 4px solid #007bff; color: black;'>
             <strong>Документ:</strong> {chunk['document_id']}<br>
             <strong>Раздел:</strong> {chunk['section_id']}<br>
-            <strong>Чанк:</strong> {chunk['chunk_id']} | <strong>Размер:</strong> {chunk['chunk_size']} символов<br>
+            <strong>Ранг:</strong> {i+1} | <strong>Размер:</strong> {chunk['chunk_size']} символов<br>
             <strong>Содержание:</strong><br>
-            <div style='background-color: white; padding: 8px; margin-top: 5px; border-radius: 3px; font-family: monospace; font-size: 12px;'>
-                {chunk['chunk_preview']}
+            <div style='background-color: white; padding: 8px; margin-top: 5px; border-radius: 3px; font-family: monospace; font-size: 12px; color: black; max-height: 200px; overflow-y: auto;'>
+                {chunk['chunk_text']}
             </div>
         </div>
         """
@@ -108,6 +108,46 @@ def switch_model(model_name, vector_index):
         log_message(error_msg)
         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>")
+    
+    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
+        
+    except Exception as e:
+        log_message(f"Ошибка при ответе на вопрос: {str(e)}")
+        return (f"<div style='color: red;'>Ошибка: {str(e)}</div>", 
+                "<div style='color: black;'>Источники недоступны из-за ошибки</div>")
+
+
+
 def create_demo_interface(answer_question_func, switch_model_func, current_model, chunk_info=None):
     with gr.Blocks(title="AIEXP - AI Expert для нормативной документации", theme=gr.themes.Soft()) as demo:
         
@@ -166,7 +206,7 @@ def create_demo_interface(answer_question_func, switch_model_func, current_model
                 with gr.Column(scale=1):
                     sources_output = gr.HTML(
                         label="",
-                        value="<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Здесь появятся источники...</div>",
+                        value="<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Здесь появятся релевантные чанки...</div>",
                     )
             
             switch_btn.click(
@@ -186,17 +226,9 @@ def create_demo_interface(answer_question_func, switch_model_func, current_model
                 inputs=[question_input],
                 outputs=[answer_output, sources_output]
             )
-        
-        with gr.Tab("Просмотр чанков"):
-            gr.Markdown("### Содержание обработанных чанков документов")
-            
-            chunks_display = gr.HTML(
-                value=create_chunks_display_html(chunk_info),
-                label="Информация о чанках"
-            )
-    
     return demo
 
+
 query_engine = None
 chunks_df = None
 reranker = None