fix for gradio

app.py

@@ -59,19 +59,26 @@ class Perform:
 
         return self.scores, None, top_k_indices, f"Найдено {len(filtered_sorted)} результатов"
 
-    def format_retrieval_results(self, top_k_indices):
-        if len(top_k_indices) == 0:
+    def format_retrieval_results(self, indices):
+        if len(indices) == 0:
             return "Нет результатов"
 
-        texts = self.retrieval.paragraphs_df["texts"].iloc[top_k_indices]
-        return "\n\n".join(texts)
-
-    def format_selected_chunks(self, indices):
         texts = self.retrieval.paragraphs_df["texts"].iloc[indices]
         return "\n\n".join(texts)
 
 perform = Perform()
 
+def ui_search(query, top_k, year_from, year_to):
+    return perform.perform_search(query, top_k, year_from, year_to)
+
+
+def ui_format_results(indices, top_k):
+    if indices is None:
+        return "Нет результатов"
+
+    indices = indices[:top_k]
+    return perform.format_retrieval_results(indices)
+
 
 def ask_llm(query, filtered_indices_state):
     """Этап 2: Отправка отфильтрованных чанков в LLM с потоковой выдачей"""
@@ -84,7 +91,7 @@ def ask_llm(query, filtered_indices_state):
         return
     
     # Форматируем контекст используя ту же функцию, что и в интерфейсе
-    context = perform.format_selected_chunks(filtered_indices_state)
+    context = perform.format_retrieval_results(filtered_indices_state)
     
     if not context or context == "Нет валидных чанков":
         yield "Нет валидных чанков для отправки"
@@ -190,22 +197,21 @@ with gr.Blocks(title="RAG Application", theme=gr.themes.Soft()) as iface:
     
     # Обработчик поиска
     search_btn.click(
-        fn=perform.perform_search,
+        fn=ui_search,
         inputs=[search_query_input, top_k_slider, year_from_input, year_to_input],
         outputs=[all_scores_state, all_chunk_ids_state, top_k_indices_state, search_status]
     ).then(
-        fn=perform.format_retrieval_results,
+        fn=ui_format_results,
         inputs=[top_k_indices_state, top_k_slider],
         outputs=[retrieval_results]
     )
 
     # Обработчик изменения слайдера top_k
     top_k_slider.change(
-        fn=perform.format_retrieval_results,
+        fn=ui_format_results,
         inputs=[top_k_indices_state, top_k_slider],
         outputs=[retrieval_results]
     )
-
     # Отправка в LLM с потоковой выдачей
     llm_btn.click(
         fn=ask_llm,