jimi
00166d7ebf
Some checks failed
Pre-commit / run (ubuntu-latest) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_en (ubuntu-latest, 3.10) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_zh (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.12) (push) Has been cancelled
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import tempfile
|
|
import uuid
|
|
from typing import Any
|
|
|
|
import requests
|
|
|
|
from .config import AUTO_DRAW_TIMEOUT_SECONDS
|
|
|
|
|
|
async def auto_draw_preview(
|
|
image_url: str,
|
|
customer_id: str,
|
|
requirement: str = "",
|
|
) -> dict[str, Any]:
|
|
"""
|
|
统一自动作图入口（直调本地链路）：
|
|
1) 下载客户图
|
|
2) 调 Gemini 生成
|
|
3) 上传图绘，返回可外发 URL
|
|
"""
|
|
try:
|
|
from services.service_gemini import GeminiExtractV2Service # type: ignore
|
|
from services.service_tuhui_upload import upload_to_tuhui # type: ignore
|
|
except Exception as e:
|
|
return {"ok": False, "error": f"import_failed:{e}"}
|
|
|
|
prompt = requirement.strip() or "按原图做高清修复，保留主体细节，输出清晰可用版本"
|
|
input_path = os.path.join(tempfile.gettempdir(), f"qjcs_in_{uuid.uuid4().hex}.jpg")
|
|
output_path = os.path.join(tempfile.gettempdir(), f"qjcs_out_{uuid.uuid4().hex}.jpg")
|
|
|
|
try:
|
|
headers = {
|
|
"User-Agent": (
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
|
"Chrome/122.0.0.0 Safari/537.36"
|
|
),
|
|
"Referer": "https://www.taobao.com/",
|
|
"Accept": "image/avif,image/webp,image/apng,image/*,*/*;q=0.8",
|
|
}
|
|
resp = requests.get(image_url, headers=headers, timeout=AUTO_DRAW_TIMEOUT_SECONDS)
|
|
if resp.status_code != 200:
|
|
return {"ok": False, "error": f"download_http_{resp.status_code}"}
|
|
with open(input_path, "wb") as f:
|
|
f.write(resp.content)
|
|
|
|
service = GeminiExtractV2Service()
|
|
ok_extract, msg_extract, _ = await service.extract_pattern(
|
|
input_path=input_path,
|
|
output_path=output_path,
|
|
custom_prompt=prompt,
|
|
aspect_ratio="1:1",
|
|
)
|
|
if not ok_extract:
|
|
return {"ok": False, "error": f"extract_failed:{msg_extract}"}
|
|
if not os.path.exists(output_path):
|
|
return {"ok": False, "error": "extract_no_output_file"}
|
|
|
|
ok_upload, link, _ = await upload_to_tuhui(
|
|
output_path,
|
|
title=f"客户{customer_id[-4:]}-预览图" if customer_id else "预览图",
|
|
description="AI自动作图预览",
|
|
price=1,
|
|
)
|
|
if not ok_upload:
|
|
return {"ok": False, "error": f"upload_failed:{link}"}
|
|
return {"ok": True, "url": str(link)}
|
|
except Exception as e:
|
|
return {"ok": False, "error": str(e)}
|
|
finally:
|
|
try:
|
|
if os.path.exists(input_path):
|
|
os.remove(input_path)
|
|
except Exception:
|
|
pass
|
|
try:
|
|
if os.path.exists(output_path):
|
|
os.remove(output_path)
|
|
except Exception:
|
|
pass
|