Update app.py

app.py

@@ -2,16 +2,22 @@ import os
 import gradio as gr
 import google.generativeai as genai
 
+# APIキー設定
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 model = genai.GenerativeModel(model_name='gemini-2.0-flash')
 
-chat_history = []
+# 応答生成
+def generate_response(message, history, temperature, top_p, top_k, max_output_tokens):
+    # Gemini 用チャット履歴の変換
+    gemini_history = []
+    for user, bot in history:
+        gemini_history.append({"role": "user", "parts": [user]})
+        gemini_history.append({"role": "model", "parts": [bot]})
+    gemini_history.append({"role": "user", "parts": [message]})
 
-def generate_response(message, temperature, top_p, top_k, max_output_tokens):
-    global chat_history
-    chat_history.append({"role": "user", "parts": [message]})
+    # 応答生成
     response = model.generate_content(
-        chat_history,
+        gemini_history,
         generation_config={
             "temperature": temperature,
             "top_p": top_p,
@@ -19,14 +25,17 @@ def generate_response(message, temperature, top_p, top_k, max_output_tokens):
             "max_output_tokens": int(max_output_tokens),
         }
     )
-    chat_history.append({"role": "model", "parts": [response.text]})
-    return response.text
 
+    history.append((message, response.text))
+    return "", history
+
+# UI構築
 with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
     gr.Markdown("## Gemini Chatbot - Gemini 2.0 Flash + 調整可能パラメータ")
 
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="メッセージを入力...")
+    state = gr.State([])  # チャット履歴を保持
 
     with gr.Row():
         temperature = gr.Slider(0.0, 1.0, value=0.7, step=0.05, label="Temperature")
@@ -36,16 +45,14 @@ with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
 
     clear = gr.Button("会話をリセット")
 
-    def respond(user_message, temperature, top_p, top_k, max_output_tokens):
-        response = generate_response(user_message, temperature, top_p, top_k, max_output_tokens)
-        return "", chatbot.append((user_message, response))
-
+    # 入力処理
     msg.submit(
-        respond,
-        inputs=[msg, temperature, top_p, top_k, max_output_tokens],
-        outputs=[msg, chatbot]
+        generate_response,
+        inputs=[msg, state, temperature, top_p, top_k, max_output_tokens],
+        outputs=[msg, chatbot, state]
     )
 
-    clear.click(lambda: ([], []), None, outputs=[chatbot, msg])
+    # リセット処理
+    clear.click(lambda: ([], []), None, outputs=[chatbot, state])
 
 demo.launch()