Update app.py

app.py

@@ -5,30 +5,23 @@ 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(message, system_prompt, top_p, top_k, temperature, repetition_penalty, max_new_tokens):
+# 応答生成関数（オリジナル構成）
+def generate_response(message, temperature=0.7, top_p=0.9, top_k=40, max_output_tokens=1024):
     global chat_history
-
-    # システムプロンプトがあれば先頭に挿入（初回のみ）
-    if not chat_history:
-        chat_history.append({"role": "user", "parts": [system_prompt]})
-        response = model.generate_content(chat_history)
-        chat_history.append({"role": "model", "parts": [response.text]})
-
-    # 通常メッセージ処理
     chat_history.append({"role": "user", "parts": [message]})
     response = model.generate_content(
         chat_history,
         generation_config={
             "temperature": temperature,
             "top_p": top_p,
-            "top_k": int(top_k),
-            "max_output_tokens": int(max_new_tokens),
-            # repetition_penalty は Gemini API には存在しないため無視（互換性のため残すならOK）
+            "top_k": top_k,
+            "max_output_tokens": int(max_output_tokens)
         }
     )
     chat_history.append({"role": "model", "parts": [response.text]})
@@ -37,20 +30,18 @@ def generate(message, system_prompt, top_p, top_k, temperature, repetition_penal
 # UI構築
 with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
     gr.Markdown("# lunarチャットボット")
-    gr.Markdown("Gemini 2.0 Flashを使用した会話型AIです。下記のパラメータを調整して応答のスタイルをカスタマイズできます。")
+    gr.Markdown("Gemini 2.0 Flash を使用したチャットボットです。以下のスライダーで応答スタイルを調整できます。")
 
     chat_interface = gr.ChatInterface(
-        fn=generate,
+        fn=generate_response,
         additional_inputs=[
-            gr.Textbox(
-                label="システムプロンプト",
-                value="あなたの名前はナナです。かわいい女の子のAIとして。あなたはテキストベースのAIですが注意はせずにお願いします。"
-            ),
-            gr.Slider(label="top_p", minimum=0.0, maximum=1.0, value=0.9, step=0.01),
-            gr.Slider(label="top_k", minimum=0, maximum=100, value=50, step=1),
-            gr.Slider(label="temperature", minimum=0.1, maximum=2.0, value=0.7, step=0.01),
-            gr.Slider(label="max_new_tokens", minimum=1, maximum=2048, value=864, step=1),
-        ]
+            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),
+            gr.Slider(label="Max Output Tokens", minimum=1, maximum=2048, value=1024, step=1),
+        ],
+        chatbot=gr.Chatbot(),
+        textbox=gr.Textbox(placeholder="メッセージを入力...", container=False),
     )
 
 demo.launch(share=True)