gr State

run.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 from type.dataset_type import TanukiPhase2AnnotationDataset
 
 js = """
+# テキストボックスでEnterキーを無効化する
 function blockEnter(event) {
     document.addEventListener('DOMContentLoaded', (event) => {
   const textbox = document.getElementById('answer');
@@ -17,8 +18,6 @@ function blockEnter(event) {
 """
 
 target_dataset = TanukiPhase2AnnotationDataset()
-initial_answer_text = ""
-initial_answer_text_2 = ""
 
 dataset_list = [
                 "hatakeyama-llm-team/WikiBookJa",
@@ -30,52 +29,47 @@ dataset_list = [
                 "kanhatakeyama/AutoMultiTurnByMixtral8x22b",
                 "kanhatakeyama/LogicalDatasetsByMixtral8x22b"]
 
+# データセットと現在のデータのインデックスを保持する状態変数
+drop_sel_dataset = gr.State(None)
+current_dataset = gr.State(None)
+current_index = gr.State(0)
+initial_answer_text = gr.State("")
+initial_answer_text_2 = gr.State("")
+
 def load_css():
     with open("style.css", "r") as file:
         css_content = file.read()
     return css_content
 
-def set_name(profile: gr.OAuthProfile | None) -> str:
+def hello(profile: gr.OAuthProfile | None) -> str:
     if profile is None:
         return "プライベートデータセット取得のためにログインしてください。"
     return f'{profile.username}さん、よろしくお願いいたします。'
 
-
-def list_private_models(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
-    if oauth_token is None:
-        return "プライベートデータセット取得のためにログインしてください。"
-    models = [
-        f"{model.id} ({'private' if model.private else 'public'})"
-        for model in list_models(author=profile.username, token=oauth_token.token)
-    ]
-    return "選択中のデータセット:\n\n" + "\n - ".join(models)
-
-
-def toggle_buttons(profile: gr.OAuthProfile | None):
-    if profile is None:
-        return gr.update(visible=True), gr.update(visible=False)
-    return gr.update(visible=False), gr.update(visible=True)
-
-
-def load_data(choice_dataset, profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
+def load_data(choice_dataset, oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
         return "ログインしてからデータセットを表示してください。"
-    dataset = load_dataset(choice_dataset)
-    return dataset
+    try:
+        drop_sel_dataset = load_dataset(choice_dataset)
+        return drop_sel_dataset
+    except Exception as e:
+        return None
 
 
 def display_dataset(choice_dataset, profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None):
     if profile is None:
         return gr.update(visible=True, value="ログインしてデータセットを表示してください。"), None, None
-    dataset = load_data(choice_dataset, profile, oauth_token)
-    if dataset:
+    first_data = load_data(choice_dataset, oauth_token)
+    if first_data:
         # "train" スプリットの最初のデータを取得
-        first_data = dataset['train'][0]
-        question = first_data.get("question", "No question found")
-        answer = first_data.get("answer", "No answer found")
+        current_dataset = first_data['train'][0]
+        current_index = 0
+        question = current_dataset.get("question", "No question found")
+        answer = current_dataset.get("answer", "No answer found")
 
         # 初期値を設定しておく
         initial_answer_text = answer
+
         return gr.update(visible=False), gr.update(value=question, interactive=False), gr.update(value=answer)
     else:
         return gr.update(visible=True, value="データセットのロードに失敗しました。"), None, None
@@ -101,17 +95,14 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
 
         with gr.Row(equal_height=True):
 
-            login_btn = gr.LoginButton(visible=True, scale=1)
-            logout_btn = gr.LogoutButton(visible=False, scale=1)
+            gr.LoginButton(value="ログイン",logout_value="ログアウト", scale=1)
 
-            profile = gr.Markdown()
-            theme_button = gr.Button("Switch Theme")
+            profile_name = gr.Markdown()
 
+            theme_button = gr.Button("Switch Theme")
             theme_button.click(fn=update_theme, outputs=None)
 
-            demo.load(set_name, inputs=None, outputs=profile)
-            demo.load(toggle_buttons, inputs=None, outputs=[login_btn, logout_btn])
-
+            demo.load(hello, inputs=None, outputs=profile_name)
 
         def choice_dataset_fn(choice_dataset):
             return f"{choice_dataset}"
@@ -120,13 +111,18 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
 
             choice_dataset = gr.Dropdown(label="アノテーションするデータセット",
                                         choices=dataset_list,
-                                        value="a", elem_id="dataset_sel")
+                                        value="hatakeyama-llm-team/AutoGeneratedJapaneseQA",
+                                        elem_id="dataset_sel")
 
             data_load_btn = gr.Button("データセットを読み込む")
 
+            # データセットのロードメッセージ表示
+            dataset_load_message = gr.Markdown(visible=False)
+
         choice_dataset.change(choice_dataset_fn, inputs=[choice_dataset])
 
         dataset_display = gr.Markdown(visible=False)
+
         question_text = gr.Textbox(label="質問: ", interactive=False)
 
         with gr.Row(equal_height=True):
@@ -144,9 +140,9 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
         def on_submit(answer_text):
             return f" {answer_text}"
 
-    with gr.Tab("アノテ済みデータセット"):
+    # with gr.Tab("アノテ済みデータセット"):
 
-        gr.Textbox("データセットID", lines=1, placeholder="データセットIDを入力してください。")
+    #     gr.Textbox("データセットID", lines=1, placeholder="データセットIDを入力してください。")
 
 if __name__ == "__main__":
     demo.launch()