Update main.py

main.py

@@ -100,7 +100,7 @@ def generate_and_upload_image(prompt: str) -> str:
         prompt: 用於生成圖片的文字提示。
         
     Returns:
-        回傳生成圖片的網址。
+        回傳生成圖片的 URL。
     """
     try:
         response = genai_client.models.generate_content(
@@ -172,7 +172,9 @@ llm = ChatGoogleGenerativeAI(google_api_key=google_api, model="gemini-2.5-flash"
 
 # 建立提示模板
 prompt_template = ChatPromptTemplate([
-    ("system", "你是一個強大的助理，可以根據用戶的請求使用提供的工具。"),
+    ("system", "你是一個強大的助理，可以根據用戶的請求使用提供的工具。當用戶要求你生成圖片時，你必須使用 generate_and_upload_image 工具。
+    在工具執行成功後，你將會獲得一個圖片的 URL。你的任務是將這個 URL 作為最終回應的一部分，直接、清楚地回傳給用戶。
+    切勿在回傳 URL 時添加任何額外的解釋性文字、符號或句子，以確保圖片能夠被正確解析和顯示。如果工具執行失敗，你則需要向用戶回報錯誤。"),
     ("user", "{input}"),
     ("placeholder", "{agent_scratchpad}"),
 ])