[mod] add modes

.gitignore

@@ -0,0 +1,2 @@
+app_v1.py
+app_v2.py

app.py

@@ -30,7 +30,8 @@ async def astream_generate(prompt: str):
     params = SamplingParams(
         temperature=0.0,
         max_tokens=4096,
-        stop=STOP_STRINGS,     # ★ここがポイント
+        repetition_penalty=1.2, # 回答重複の防止
+        stop=STOP_STRINGS,
     )
 
     async for out in model.generate(prompt, params, request_id=str(uuid.uuid4())):
@@ -53,18 +54,27 @@ async def astream_generate(prompt: str):
             yield chunk
 
 # Gradio 用の応答関数（同期ジェネレータ）
-def respond(user_input, history):
+def respond(user_input, history, mode):
     history = history or []
     # システムプロンプトと過去履歴を組み立て
-    base_prompt = "以下は親切で何でも答えてくれるAIアシスタントとの会話です。\n"
+    base_prompt = "あなたは製薬に関する専門家です。製薬に関するユーザーの質問に親切に回答してください。参照した文献は常に提示してください。\n"
+
     for u, b in history:
         base_prompt += f"ユーザー: {u}\nアシスタント: {b}\n"
+
+    if "製薬の専門家" in mode:
+        base_prompt += "あなたは製薬に関する専門家です。製薬に関するユーザーの質問に親切に回答してください。参照した文献は常に提示してください。\n"
+    if "グローバル対応" in mode:
+        base_prompt += "グローバル基準に照らした回答をしてください。\n"
+
     base_prompt += f"ユーザー: {user_input}\nアシスタント:"
 
     # ユーザー発話と空応答を履歴に追加
     history.append((user_input, ""))
 
     # 単一のイベントループを使ってストリーミング
+    print("log: base_prompt")
+    print(base_prompt)
     agen = astream_generate(base_prompt)
     try:
         while True:
@@ -74,19 +84,24 @@ def respond(user_input, history):
             yield history, history
     except StopAsyncIteration:
         return
-
-# Gradio UI 定義
+    
 with gr.Blocks() as demo:
     gr.Markdown("# 💊 製薬に関する質問をしてみてください。")
     gr.Markdown("※ ストリーミングデモ用コードです。")
 
+    mode = gr.CheckboxGroup(
+        label="モード",
+        choices=["製薬の専門家", "グローバル対応", "スイッチ3"],
+        value=[],
+    )
+
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="あなたのメッセージ　（Enterで送信）")
     send = gr.Button("チャットを送信")
     clear = gr.Button("チャット履歴をクリア")
     state = gr.State([])
 
-    # ▼ プリセット（クリックでテキストボックスに反映）
+     # ▼ プリセット（クリックでテキストボックスに反映）
     gr.Markdown("### 🔧 すぐ使えるプリセット")
     preset_list = [
         "グレープフルーツと薬を一緒に飲んじゃだめなんですか？",
@@ -99,9 +114,9 @@ with gr.Blocks() as demo:
         label="プリセット（クリックで入力欄に反映）"
     )
 
-    # 既存の送信/クリア動作
-    msg.submit(respond, [msg, state], [chatbot, state]).then(lambda: "", None, msg)
-    send.click(respond, [msg, state], [chatbot, state]).then(lambda: "", None, msg)
+    # 送信ボタンやEnter送信でチェックボックスの状態も一緒に渡す
+    msg.submit(respond, [msg, state, mode], [chatbot, state]).then(lambda: "", None, msg)
+    send.click(respond, [msg, state, mode], [chatbot, state]).then(lambda: "", None, msg)
     clear.click(lambda: ([], []), None, [chatbot, state])
 
 # エントリポイント