SC999 commited on
Commit
55a462c
1 Parent(s): d5cde43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -11
app.py CHANGED
@@ -33,24 +33,23 @@ def response(message, history):
33
  global chat
34
  # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
35
  chat.history = transform_history(history)
36
-
37
- prompt = "你是一個半導體智慧製造專家,熟悉半導體製程及生產過程,請用專業的角度回應接下來的問題!"
38
- if len(history) == 0:
39
- prompt_message = f"{prompt} \n {message}"
40
- else:
41
- prompt_message = message
42
-
43
  # 發送訊息到 Gemini API
44
- response = chat.send_message(prompt_message)
45
  response.resolve()
46
 
47
  # 逐字回傳生成的文字,實現打字機效果
48
  for i in range(len(response.text)):
49
  time.sleep(0.05) # 每個字符間隔 0.05 秒
50
  yield response.text[: i+1]
51
-
52
 
53
  # 建立 Gradio 聊天界面
54
  gr.ChatInterface(response,
55
- title='智慧製造小幫手',
56
- textbox=gr.Textbox(placeholder="Question to Gemini")).launch(share=True)
 
33
  global chat
34
  # 將 Gradio 的歷史紀錄轉換為 Gemini 的格式
35
  chat.history = transform_history(history)
36
+
37
+ # 自訂 Prompt:限縮為與半導體智慧製造相關的回答
38
+ prompt = "你是一位半導體智慧製造。你負責協助回答先進封裝半導體智慧製造相關,請只回答與半導體相關的問題,回答語氣需要溫柔且有耐心,且所有回答一律使用繁體中文。"
39
+
40
+ # 合併Prompt與使用者訊息
41
+ final_message = prompt + "\n" + message
42
+
43
  # 發送訊息到 Gemini API
44
+ response = chat.send_message(final_message)
45
  response.resolve()
46
 
47
  # 逐字回傳生成的文字,實現打字機效果
48
  for i in range(len(response.text)):
49
  time.sleep(0.05) # 每個字符間隔 0.05 秒
50
  yield response.text[: i+1]
 
51
 
52
  # 建立 Gradio 聊天界面
53
  gr.ChatInterface(response,
54
+ title='智慧小助手',
55
+ textbox=gr.Textbox(placeholder="請輸入與半導體相關的問題")).launch(share=True)