Upload app.py

app.py

@@ -84,7 +84,7 @@ def estimate_ng_probability(input_text, thresold=0.3):
 # 会話履歴用リスト型変数
 message_history = []
 
-def chat(user_msg, mode, threshold):
+def chat(user_msg, mode, threshold=None):
     """
     AIとの会話を実行後、全会話履歴を返す
     user_msg: 入力されたユーザのメッセージ
@@ -118,11 +118,13 @@ with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     input = gr.Textbox(show_label=False, placeholder="チェックしたい文章を入力してください")
     mode = gr.Radio(["理由を出力", "確率を出力"], label="モードを選択")
-    if mode == "確率を出力":
+
+    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度の閾値")
-    else:
-        threshold = None
-    input.submit(fn=chat, inputs=[input, mode, threshold], outputs=chatbot) # メッセージ送信されたら、AIと会話してチャット欄に全会話内容を表示
+        input.submit(fn=chat, inputs=[input, mode, threshold], outputs=chatbot) # メッセージ送信されたら、AIと会話してチャット欄に全会話内容を表示
+    
     input.submit(fn=lambda: "", inputs=None, outputs=input) # （上記に加えて）入力欄をクリア
 
 demo.launch(share=True)