diff --git a/libs/ktem/ktem/index/file/ui.py b/libs/ktem/ktem/index/file/ui.py
index e973c21..834259e 100644
--- a/libs/ktem/ktem/index/file/ui.py
+++ b/libs/ktem/ktem/index/file/ui.py
@@ -187,6 +187,20 @@ class FileIndexPage(BasePage):
)
with gr.Row():
self.is_zipped_state = gr.State(value=False)
+
+ self.download_single_button = gr.DownloadButton(
+ "Download file",
+ visible=False,
+ )
+
+ with gr.Row() as self.selection_info:
+ self.selected_file_id = gr.State(value=None)
+ with gr.Column(scale=2):
+ self.selected_panel = gr.Markdown(self.selected_panel_false)
+
+ self.chunks = gr.HTML(visible=False)
+
+ with gr.Accordion("Advance Options", open=False):
with gr.Row():
self.download_all_button = gr.DownloadButton(
"Download all files",
@@ -204,18 +218,6 @@ class FileIndexPage(BasePage):
"Cancel", visible=False
)
- self.download_single_button = gr.DownloadButton(
- "Download file",
- visible=False,
- )
-
- with gr.Row() as self.selection_info:
- self.selected_file_id = gr.State(value=None)
- with gr.Column(scale=2):
- self.selected_panel = gr.Markdown(self.selected_panel_false)
-
- self.chunks = gr.HTML(visible=False)
-
def on_subscribe_public_events(self):
"""Subscribe to the declared public event of the app"""
if self._app.f_user_management:
diff --git a/libs/ktem/ktem/rerankings/ui.py b/libs/ktem/ktem/rerankings/ui.py
index 311a794..6ac8fee 100644
--- a/libs/ktem/ktem/rerankings/ui.py
+++ b/libs/ktem/ktem/rerankings/ui.py
@@ -7,6 +7,8 @@ from ktem.app import BasePage
from ktem.utils.file import YAMLNoDateSafeLoader
from theflow.utils.modules import deserialize
+from kotaemon.base import Document
+
from .manager import reranking_models_manager
@@ -348,7 +350,7 @@ class RerankingManagement(BasePage):
log_content += "- Sending a message ([`Hello`], `Hi`)
"
yield log_content
- _ = rerank(["Hello"], "Hi")
+ _ = rerank([Document(content="Hello")], "Hi")
log_content += (
"- Connection success. "