モード追加

run.py

@@ -148,52 +148,64 @@ with gr.Blocks(theme=theme_, css=load_css()) as demo:
 
         dataset_display = gr.Markdown(visible=False)
 
-        question_text = gr.Textbox(label="質問: ", interactive=False)
-
-        with gr.Row(equal_height=True):
-            good_btn = gr.Button("良い")
-            bad_btn = gr.Button("悪い")
-            
-        gr.Markdown("#### 下記回答は、余裕があればで大丈夫です。")
-            
-        score_slider = gr.Slider(1, 5, label="スコア: 1-5　(1:大変悪い、2:悪い、3:普通、4:良い、5:大変良い)", step=1, value=score.value, interactive=True)
-
-        answer_text = gr.Textbox(label="回答:　改行はShift+Enterです。　文章を修正して頂けると、さらに高品質になります。", elem_id="answer", interactive=True)
-
-        data_load_btn.click(
-                display_dataset,
-                inputs=[gr_current_dataset],
-                outputs=[dataset_display, question_text, answer_text],
-            )
+        question_text = gr.Textbox(label="質問: ", interactive=False)        
         
-        # def good_click(profile, oauth_token):
-        #     print("good")
-        #     # インデックスを更新
-        #     current_idx.value = (current_idx.value + 1)
-        #     # current_dataset を更新
-        #     current_dataset.value = target_dataset.value['train'][current_idx.value]
+        with gr.Tab("シンプルモード"):
             
-        #     print(current_dataset.value)
+            with gr.Column():
+
+                with gr.Row(equal_height=True):
+                    good_btn = gr.Button("良い")
+                    bad_btn = gr.Button("悪い")
+                
+                answer_text = gr.Textbox(label="回答:",lines=20, interactive=False)
+                
+                data_load_btn.click(
+                    display_dataset,
+                    inputs=[gr_current_dataset],
+                    outputs=[dataset_display, question_text, answer_text],
+                )    
+                
+                # def good_click(profile, oauth_token):
+                #     print("good")
+                #     # インデックスを更新
+                #     current_idx.value = (current_idx.value + 1)
+                #     # current_dataset を更新
+                #     current_dataset.value = target_dataset.value['train'][current_idx.value]
+                    
+                #     print(current_dataset.value)
+                    
+                #     # データを表示
+                #     question = current_dataset.value.get("question", "質問が見つかりません")
+                #     answer = current_dataset.value.get("answer", "解答が見つかりません")
+                #     return gr.update(visible=False), gr.update(value=question, interactive=False), gr.update(value=answer)
+
+                # good_btn.click(
+                #     good_click,
+                #     inputs=[None, None],  # Noneはprofileとoauth_tokenに相当
+                #     outputs=[dataset_display, question_text, answer_text]
+                # )
+                
+                def bad_click():
+                    print("bad")
+                    
+                bad_btn.click(
+                    bad_click,
+                    inputs=[],
+                    outputs=None
+                )
             
-        #     # データを表示
-        #     question = current_dataset.value.get("question", "質問が見つかりません")
-        #     answer = current_dataset.value.get("answer", "解答が見つかりません")
-        #     return gr.update(visible=False), gr.update(value=question, interactive=False), gr.update(value=answer)
-
-        # good_btn.click(
-        #     good_click,
-        #     inputs=[None, None],  # Noneはprofileとoauth_tokenに相当
-        #     outputs=[dataset_display, question_text, answer_text]
-        # )
-        
-        def bad_click():
-            print("bad")
+        with gr.Tab("丁寧モード"):
             
-        bad_btn.click(
-            bad_click,
-            inputs=[],
-            outputs=None
-        )        
+            score_slider = gr.Slider(1, 5, label="スコア: 1-5　(1:大変悪い、2:悪い、3:普通、4:良い、5:大変良い)", step=1, value=score.value, interactive=True)
+
+            answer_text = gr.Textbox(label="回答:　改行はShift+Enterです。　文章を修正して頂けると、さらに高品質になります。", lines=20, elem_id="answer", interactive=True)
+
+            data_load_btn.click(
+                    display_dataset,
+                    inputs=[gr_current_dataset],
+                    outputs=[dataset_display, question_text, answer_text],
+                )               
 
     with gr.Tab("アノテ済みデータセット(管理画面)"):