Upload app.py

app.py

@@ -116,14 +116,10 @@ def chat(user_msg, mode, threshold=None):
 with gr.Blocks() as demo:
     # チャットボットUI処理
     chatbot = gr.Chatbot()
-    input = gr.Textbox(show_label=False, placeholder="チェックしたい文章を入力してください")
-    mode = gr.Radio(["理由を出力", "確率を出力"], label="モードを選択")
 
-    if mode == "理由を出力":
-        input.submit(fn=chat, inputs=[input, mode], outputs=chatbot) # メッセージ送信されたら、AIと会話してチャット欄に全会話内容を表示
-    elif mode == "確率を出力":
-        threshold = gr.Slider(minimum=0, maximum=1, value=0.3, step=0.05, label="NG度の閾値")
-        input.submit(fn=chat, inputs=[input, mode, threshold], outputs=chatbot) # メッセージ送信されたら、AIと会話してチャット欄に全会話内容を表示
+    input = gr.Textbox(show_label=False, placeholder="チェックしたい文章を入力してください")
+    
+    input.submit(fn=chat, inputs=[input, gr.Radio(["理由を出力", "確率を出力"], label="モードを選択"), threshold = gr.Slider(minimum=0, maximum=1, value=0.3, step=0.05, label="NG度の閾値")], outputs=chatbot) # メッセージ送信されたら、AIと会話してチャット欄に全会話内容を表示
     
     input.submit(fn=lambda: "", inputs=None, outputs=input) # （上記に加えて）入力欄をクリア