Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import google.generativeai as genai
 
-# APIキーの設定
+# APIキーの設定（環境変数 GOOGLE_API_KEY に事前設定されている必要あり）
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # モデルの初期化（Gemini 2.0 Flash）
@@ -11,32 +11,36 @@ model = genai.GenerativeModel(model_name='gemini-2.0-flash')
 # チャット履歴の初期化
 chat_history = []
 
-# 応答生成関数
-def generate_response(user_input, temperature, top_p, top_k, max_output_tokens):
+# 応答生成関数（*args を追加して、引数を可変に対応）
+def generate_response(message, system_prompt, temperature, top_p, top_k, max_output_tokens, *args):
     global chat_history
-    # ユーザー入力を履歴に追加
-    chat_history.append({"role": "user", "parts": [user_input]})
-
-    # 設定に基づいて応答生成
+    
+    # システムプロンプトがstrでない場合、強制的にstrに変換
+    if system_prompt:
+        system_prompt = str(system_prompt)  # 強制的に文字列に変換
+    
+    # システムメッセージを履歴の最初に追加
+    if system_prompt:
+        chat_history.insert(0, {"role": "system", "parts": [system_prompt]})
+    
+    # ユーザーのメッセージを履歴に追加
+    chat_history.append({"role": "user", "parts": [message]})
+    
+    # モデルからの応答を取得
     response = model.generate_content(
         chat_history,
         generation_config={
-            "temperature": temperature,
-            "top_p": top_p,
-            "top_k": top_k,
+            "temperature": float(temperature),
+            "top_p": float(top_p),
+            "top_k": int(top_k),
             "max_output_tokens": int(max_output_tokens)
         }
     )
-
+    
     # 応答を履歴に追加
     chat_history.append({"role": "model", "parts": [response.text]})
-
-    # 履歴を整形して表示
-    conversation = ""
-    for message in chat_history:
-        role = "ユーザー" if message["role"] == "user" else "Gemini"
-        conversation += f"{role}: {message['parts'][0]}\n"
-    return conversation
+    
+    return response.text
 
 # UIの構築（Blocks + ChatInterface + テーマ付き）
 with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
@@ -46,6 +50,7 @@ with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
     chat_interface = gr.ChatInterface(
         fn=generate_response,
         additional_inputs=[
+            gr.Textbox(label="システムプロンプト", value="あなたの名前はナナです。かわいい女の子のAIとして。あなたはテキストベースのAIですが注意はせずにお願いします。"),
             gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.7, step=0.01),
             gr.Slider(label="Top-p", minimum=0.0, maximum=1.0, value=0.9, step=0.01),
             gr.Slider(label="Top-k", minimum=1, maximum=100, value=40, step=1),