Update app.py

app.py

@@ -1,4 +1,3 @@
-#app.py
 import time
 import gradio as gr
 import google.generativeai as genai
@@ -28,12 +27,22 @@ def transform_history(history):
         new_history.append({"parts": [{"text": chat[1]}], "role": "model"})
     return new_history
 
+# 檢查問題是否與設備維護相關的函數
+def is_maintenance_related(question):
+    keywords = ["maintenance", "repair", "inspection", "equipment", "servicing", "troubleshooting", "machine", "calibration", "preventive", "predictive", "breakdown"]
+    return any(keyword in question.lower() for keyword in keywords)
+
 # 回應生成函數
 def response(message, history):
     global chat
     # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
     chat.history = transform_history(history)
     
+    # 檢查問題是否與設備維護相關
+    if not is_maintenance_related(message):
+        yield "抱歉，我只能回答與設備維護相關的問題。請輸入有關設備維護的問題。"
+        return
+    
     # 發送訊息到 Gemini API
     response = chat.send_message(message)
     response.resolve()
@@ -45,5 +54,5 @@ def response(message, history):
 
 # 建立 Gradio 聊天界面
 gr.ChatInterface(response,
-                 title='Gemini Chat',
-                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)
+                 title='設備製造商助理 - 專注於設備維護問題',
+                 textbox=gr.Textbox(placeholder="請輸入有關設備維護的問題")).launch(share=True)