Update app.py

app.py

@@ -5,21 +5,32 @@ import google.generativeai as genai
 # APIキーの設定
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# モデルの初期化（Gemini 1.5 Flashを使用）
-model = genai.GenerativeModel(model='gemini-2.0-flash')
+# モデルの初期化（Gemini 2.0 Flash）
+model = genai.GenerativeModel(model_name='gemini-2.0-flash')
 
 # チャット履歴の初期化
 chat_history = []
 
 # 応答生成関数
-def generate_response(user_input):
+def generate_response(user_input, temperature, top_p, top_k, max_output_tokens):
     global chat_history
     # ユーザー入力を履歴に追加
     chat_history.append({"role": "user", "parts": [user_input]})
-    # モデルからの応答を取得
-    response = model.generate_content(chat_history)
+
+    # 設定に基づいて応答生成
+    response = model.generate_content(
+        chat_history,
+        generation_config={
+            "temperature": temperature,
+            "top_p": top_p,
+            "top_k": top_k,
+            "max_output_tokens": int(max_output_tokens)
+        }
+    )
+
     # 応答を履歴に追加
     chat_history.append({"role": "model", "parts": [response.text]})
+
     # 履歴を整形して表示
     conversation = ""
     for message in chat_history:
@@ -30,10 +41,16 @@ def generate_response(user_input):
 # Gradioインターフェースの設定
 iface = gr.Interface(
     fn=generate_response,
-    inputs="text",
+    inputs=[
+        gr.Textbox(label="ユーザー入力"),
+        gr.Slider(0.0, 1.0, value=0.7, step=0.05, label="Temperature"),
+        gr.Slider(0.0, 1.0, value=0.9, step=0.05, label="Top-p"),
+        gr.Slider(1, 100, value=40, step=1, label="Top-k"),
+        gr.Number(value=1024, label="Max Output Tokens"),
+    ],
     outputs="text",
     title="Gemini Chatbot",
-    description="Gemini 2.0 Flashモデルを使用したチャットボットです。会話の履歴が表示されます。",
+    description="Gemini 2.0 Flashモデルを使用したチャットボットです。temperatureやtop-pなどを自由に調整できます。",
 )
 
 # インターフェースの起動