Update app.py

app.py

@@ -1,3 +1,4 @@
+#app.py
 import time
 import gradio as gr
 import google.generativeai as genai
@@ -27,22 +28,12 @@ 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()
@@ -54,5 +45,5 @@ def response(message, history):
 
 # 建立 Gradio 聊天界面
 gr.ChatInterface(response,
-                 title='設備製造商助理 - 專注於設備維護問題',
-                 textbox=gr.Textbox(placeholder="請輸入有關設備維護的問題")).launch(share=True)
+                 title='Gemini Chat',
+                 textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)