Update app.py

app.py

@@ -1,4 +1,3 @@
-#app.py
 import time
 import gradio as gr
 import google.generativeai as genai
@@ -20,6 +19,14 @@ try:
 except Exception as e:
     raise ValueError(f"無法載入模型：{e}")
 
+# 定義與酒類、調酒、酒吧相關的關鍵詞
+keywords = ["alcohol", "cocktail", "beer", "wine", "whiskey", "vodka", "rum", "bartender", "bar", "drink", "liquor", "spirits"]
+
+# 檢查訊息是否與酒類、調酒、酒吧相關
+def is_relevant_question(message):
+    message_lower = message.lower()  # 將訊息轉為小寫以便檢查
+    return any(keyword in message_lower for keyword in keywords)
+
 # 將 Gradio 的歷史紀錄轉換為 Gemini 格式
 def transform_history(history):
     new_history = []
@@ -31,6 +38,12 @@ def transform_history(history):
 # 回應生成函數
 def response(message, history):
     global chat
+    
+    # 檢查訊息是否與酒類、調酒、酒吧相關
+    if not is_relevant_question(message):
+        yield "I'm sorry, I can only answer questions related to alcohol, cocktails, and bars."
+        return
+    
     # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
     chat.history = transform_history(history)
     
@@ -46,4 +59,4 @@ def response(message, history):
 # 建立 Gradio 聊天界面
 gr.ChatInterface(response,
                  title='Gemini Chat',
-                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)
+                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)