Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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"
|
| 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}
|
| 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}]}
|