Update app.py

app.py

@@ -1,4 +1,23 @@
-import gradio as gr
+def respond(message, history):
+    """
+    チャットボットの応答を生成する関数
+    historyはタプルのリスト形式: [(user_msg, bot_msg), ...]
+    """
+    try:
+        # システムメッセージとして使用するプロンプト
+        system_message = "あなたは親切で知識豊富な日本語アシスタントです。ユーザーの質問に丁寧に答えてください。"
+        
+        # 会話履歴を文字列形式に変換
+        conversation = ""
+        if system_message.strip():
+            conversation += f"システム: {system_message}\n"
+        
+        # 会話履歴を追加（タプル形式）
+        for user_msg, bot_msg in history:
+            if user_msg:
+                conversation += f"ユーザー: {user_msg}\n"
+            if bot_msg:
+                conversation += f"アシスタント: {bot_msg}\n"import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import warnings
@@ -25,7 +44,7 @@ print("モデルの読み込みが完了しました。")
 def respond(message, history):
     """
     チャットボットの応答を生成する関数
-    ※ type="messages"の場合、historyはメッセージ辞書のリスト形式になります
+    ChatInterfaceではtype="messages"でもhistoryの形式が異なる場合があります
     """
     try:
         # システムメッセージとして使用するプロンプト
@@ -36,12 +55,22 @@ def respond(message, history):
         if system_message.strip():
             conversation += f"システム: {system_message}\n"
         
-        # 会話履歴を追加（messages形式の場合）
-        for msg in history:
-            if msg["role"] == "user":
-                conversation += f"ユーザー: {msg['content']}\n"
-            elif msg["role"] == "assistant":
-                conversation += f"アシスタント: {msg['content']}\n"
+        # 会話履歴を追加（historyの形式を確認して処理）
+        if history:
+            for item in history:
+                if isinstance(item, dict):
+                    # messages形式の場合
+                    if item.get("role") == "user":
+                        conversation += f"ユーザー: {item.get('content', '')}\n"
+                    elif item.get("role") == "assistant":
+                        conversation += f"アシスタント: {item.get('content', '')}\n"
+                elif isinstance(item, (list, tuple)) and len(item) >= 2:
+                    # タプル形式の場合 (user_msg, bot_msg)
+                    user_msg, bot_msg = item[0], item[1]
+                    if user_msg:
+                        conversation += f"ユーザー: {user_msg}\n"
+                    if bot_msg:
+                        conversation += f"アシスタント: {bot_msg}\n"
         
         # 現在のメッセージを追加
         conversation += f"ユーザー: {message}\nアシスタント: "
@@ -92,7 +121,7 @@ type="messages"を設定してOpenAI形式のメッセージを使用
 """
 demo = gr.ChatInterface(
     respond,
-    type="messages",  # 重要：これによりhistoryがメッセージ辞書形式になります
+    # type="messages"を削除 - デフォルトのタプル形式を使用
     title="🤖 Sarashina Chatbot",
     description="Sarashina2.2-3b-instruct モデルを使用した日本語チャットボットです。",
     theme=gr.themes.Soft(),