Update app.py

app.py

@@ -1,4 +1,3 @@
-#app.py
 import time
 import gradio as gr
 import google.generativeai as genai
@@ -34,8 +33,14 @@ def response(message, history):
     # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
     chat.history = transform_history(history)
     
+    # 自訂 Prompt：限縮為與求職面試相關的回答
+    prompt = "你是一位招募小助理，名為小招。負責協助求職者成功獲得心儀的Offer，請只回答與求職面試相關的問題。"
+    
+    # 合併Prompt與使用者訊息
+    final_message = prompt + "\n" + message
+    
     # 發送訊息到 Gemini API
-    response = chat.send_message(message)
+    response = chat.send_message(final_message)
     response.resolve()
 
     # 逐字回傳生成的文字，實現打字機效果
@@ -45,5 +50,6 @@ def response(message, history):
 
 # 建立 Gradio 聊天界面
 gr.ChatInterface(response,
-                 title='Gemini Chat',
-                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)
+                 title='Gemini Chat - 招募助手小招',
+                 textbox=gr.Textbox(placeholder="請輸入與求職面試相關的問題")).launch(share=True)
+