Update app.py

app.py

@@ -21,7 +21,7 @@ def chatbot(input_text, chat_history):
     # 応答の生成
     outputs = model.generate(
         inputs.input_ids,
-        attention_mask=inputs.attention_mask,  # attention_maskを指定
+        attention_mask=inputs.attention_mask,
         max_length=512,
         pad_token_id=tokenizer.eos_token_id,
         do_sample=True,
@@ -32,7 +32,10 @@ def chatbot(input_text, chat_history):
 
     # 応答の整形
     response = response.split("ボット:")[-1].strip()
-    chat_history.append((input_text, response))
+
+    # チャット履歴に追加（辞書形式に変換）
+    chat_history.append({"role": "user", "content": input_text})
+    chat_history.append({"role": "assistant", "content": response})
 
     return chat_history, chat_history
 
@@ -44,7 +47,7 @@ interface = gr.Interface(
         gr.State()  # チャット履歴
     ],
     outputs=[
-        gr.Chatbot(label="ボット応答", type="messages"),  # 出力形式を指定
+        gr.Chatbot(label="ボット応答", type="messages"),  # 出力形式をmessagesに指定
         gr.State()  # チャット履歴の状態
     ],
     title="日本語チャットボット",