Update app.py

app.py

@@ -1,4 +1,3 @@
-#app.py
 import time
 import gradio as gr
 import google.generativeai as genai
@@ -31,6 +30,12 @@ def transform_history(history):
 # 回應生成函數
 def response(message, history):
     global chat
+
+    # 增加問題篩選邏輯：僅回應與面試流程或建議相關的問題
+    if not ("面試" in message or "建議" in message):
+        yield "螃蟹小助理只能回答面試流程或面試建議的問題喔！"
+        return
+
     # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
     chat.history = transform_history(history)
     
@@ -43,7 +48,9 @@ def response(message, history):
         time.sleep(0.05)  # 每個字符間隔 0.05 秒
         yield response.text[: i+1]
 
-# 建立 Gradio 聊天界面
-gr.ChatInterface(response,
-                 title='Gemini Chat',
-                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)
+# 建立 Gradio 聊天界面，更新標題與設定
+gr.ChatInterface(
+    response,
+    title='瑞昱面試小助手🦀',
+    textbox=gr.Textbox(placeholder="請問螃蟹小助理有關面試流程或建議的問題")
+).launch(share=True)