Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import google.generativeai as genai
 
-# APIキーの設定（環境変数 GOOGLE_API_KEY に事前設定されている必要あり）
+# APIキーの設定
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # モデルの初期化（Gemini 2.0 Flash）
@@ -11,54 +11,47 @@ model = genai.GenerativeModel(model_name='gemini-2.0-flash')
 # チャット履歴の初期化
 chat_history = []
 
-# 応答生成関数（*args を追加して、引数を可変に対応）
-def generate_response(message, system_prompt, temperature, top_p, top_k, max_output_tokens, *args):
+# 応答生成関数
+def generate_response(user_input, temperature, top_p, top_k, max_output_tokens):
     global chat_history
-    
-    # システムプロンプトが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]})
-    
-    # モデルからの応答を取得
+    # ユーザー入力を履歴に追加
+    chat_history.append({"role": "user", "parts": [user_input]})
+
+    # 設定に基づいて応答生成
     response = model.generate_content(
         chat_history,
         generation_config={
-            "temperature": float(temperature),
-            "top_p": float(top_p),
-            "top_k": int(top_k),
+            "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]})
-    
-    return response.text
 
-# UIの構築（Blocks + ChatInterface + テーマ付き）
-with gr.Blocks(theme='Nymbo/Alyx_theme') as demo:
-    gr.Markdown("# lunarチャットボット")
-    gr.Markdown("Gemini 2.0 Flash を使用したチャットボットです。下のスライダーで応答のスタイルを調整できます。")
+    # 履歴を整形して表示
+    conversation = ""
+    for message in chat_history:
+        role = "ユーザー" if message["role"] == "user" else "Gemini"
+        conversation += f"{role}: {message['parts'][0]}\n"
+    return conversation
 
-    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),
-            gr.Slider(label="Max Output Tokens", minimum=1, maximum=2048, value=1024, step=1),
-        ],
-        chatbot=gr.Chatbot(),
-        textbox=gr.Textbox(placeholder="メッセージを入力...", container=False),
-    )
+# Gradioインターフェースの設定
+iface = gr.Interface(
+    fn=generate_response,
+    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モデルを使用したチャットボットです。temperatureやtop-pなどを自由に調整できます。",
+)
 
-# アプリの起動
-demo.launch(share=True)
+# インターフェースの起動
+iface.launch()