Update app.py

app.py

@@ -1,3 +1,4 @@
+
 from openai import OpenAI
 import gradio as gr
 import os
@@ -7,10 +8,13 @@ api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
 def chat_with_openai(user_input, history):
-    if history is None or len(history) == 0:
-        # チャット履歴がない場合、システムメッセージを追加
-        history = [{"role": "system", "content": "You are a helpful assistant."}]
-    
+    if history is None:
+        history = []
+
+    # 最初のメッセージの場合、システムメッセージの代わりにユーザーメッセージとして追加
+    if len(history) == 0:
+        history.append({"role": "user", "content": "You are a helpful assistant."})
+
     # ユーザーの入力を履歴に追加
     history.append({"role": "user", "content": user_input})
 
@@ -21,17 +25,22 @@ def chat_with_openai(user_input, history):
     )
 
     # アシスタントの応答を取得
-    assistant_message = completion.choices[0].message.content  # 修正箇所
+    assistant_message = completion.choices[0].message.content
 
     # 履歴にアシスタントの応答を追加
     history.append({"role": "assistant", "content": assistant_message})
 
-    # チャットボットにメッセージを表示するため、履歴をそのまま返す
-    return history, history
+    # チャットボットに表示するメッセージを整形
+    messages_to_display = []
+    for i in range(1, len(history)-1, 2):
+        user_msg = history[i]["content"]
+        assistant_msg = history[i+1]["content"]
+        messages_to_display.append((user_msg, assistant_msg))
+
+    return messages_to_display, history
 
 with gr.Blocks() as demo:
-    # Chatbotコンポーネントにtype='messages'を指定
-    chatbot = gr.Chatbot(type='messages')
+    chatbot = gr.Chatbot()
     state = gr.State([])
 
     with gr.Row():