From 7ac8f0329a1a70b680ea6e31c04ccc7770eaa397 Mon Sep 17 00:00:00 2001
From: "Tuan Anh Nguyen Dang (Tadashi_Cin)" <tadashi@cinnamon.is>
Date: Tue, 1 Oct 2024 12:03:19 +0700
Subject: [PATCH] fix: improve Download UI bump:minor (#352)

* fix: rerank test ui

* fix: improve download all UI
---
 libs/ktem/ktem/index/file/ui.py | 26 ++++++++++++++------------
 libs/ktem/ktem/rerankings/ui.py |  4 +++-
 2 files changed, 17 insertions(+), 13 deletions(-)

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`)<br>"
             yield log_content
-            _ = rerank(["Hello"], "Hi")
+            _ = rerank([Document(content="Hello")], "Hi")
 
             log_content += (
                 "<mark style='background: green; color: white'>- Connection success. "