hsun0602 commited on
Commit
4aa5143
·
verified ·
1 Parent(s): 7e208f7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -116,7 +116,7 @@ def generate_and_upload_image(prompt: str) -> str:
116
  # --- 5. 回傳圖片連結 ---
117
  # 取得目前的伺服器基礎位址 (例如 Hugging Face Space 或本地端)
118
  base_url = os.getenv("HF_SPACE", "http://localhost:7860").rstrip("/")
119
- return f"請分析這張圖片 {image_path},問題是:{user_text}"
120
  else:
121
  return "從 URL 下載圖片失敗。"
122
  else:
@@ -151,7 +151,7 @@ system_prompt = """
151
  """
152
 
153
  agent_executor = create_agent(
154
- model="gemini-2.5-flash",
155
  tools=tools,
156
  system_prompt=system_prompt,
157
  )
@@ -188,7 +188,7 @@ def handle_message(event):
188
  previous_message = get_previous_message(user_id)
189
  if previous_message["type"] == "image":
190
  image_path = previous_message["content"]
191
- user_text = f"{image_path} {user_text}"
192
  user_message_history[user_id].pop()
193
 
194
  agent_input = {"messages": [{"role": "user", "content": user_text}]}
 
116
  # --- 5. 回傳圖片連結 ---
117
  # 取得目前的伺服器基礎位址 (例如 Hugging Face Space 或本地端)
118
  base_url = os.getenv("HF_SPACE", "http://localhost:7860").rstrip("/")
119
+ return f"{base_url}/{file_name}"
120
  else:
121
  return "從 URL 下載圖片失敗。"
122
  else:
 
151
  """
152
 
153
  agent_executor = create_agent(
154
+ model="google_genai:gemini-2.5-flash",
155
  tools=tools,
156
  system_prompt=system_prompt,
157
  )
 
188
  previous_message = get_previous_message(user_id)
189
  if previous_message["type"] == "image":
190
  image_path = previous_message["content"]
191
+ user_text = f"請分析這張圖片 {image_path},問題是:{user_text}"
192
  user_message_history[user_id].pop()
193
 
194
  agent_input = {"messages": [{"role": "user", "content": user_text}]}