From 25087cbe79c8c4992551477d55d9db8bbea2202e Mon Sep 17 00:00:00 2001
From: wangzhibo <wangzhibo@shsening.com>
Date: 星期三, 30 七月 2025 22:14:39 +0800
Subject: [PATCH] Merge branch 'master' of http://222.71.114.84:7070/r/AutoMediaPublisher
---
auto_media_publisher/utils/utils_text2image_pollinations.py | 22 +++++++++++++++++++---
1 files changed, 19 insertions(+), 3 deletions(-)
diff --git a/auto_media_publisher/utils/utils_text2image_pollinations.py b/auto_media_publisher/utils/utils_text2image_pollinations.py
index e8f2a30..f78411b 100644
--- a/auto_media_publisher/utils/utils_text2image_pollinations.py
+++ b/auto_media_publisher/utils/utils_text2image_pollinations.py
@@ -3,6 +3,8 @@
import datetime
import random
import os
+from io import BytesIO
+
from auto_media_publisher.config.conf_base import (
TRANSLATE_API_TEMPLATE,
POLLINATIONS_API_TEMPLATE,
@@ -77,9 +79,9 @@
if not filename:
now = datetime.datetime.now()
- filename = now.strftime("%Y%m%d%H%M%S") + ".jpg"
- if not filename.lower().endswith(".jpg"):
- filename += ".jpg"
+ filename = now.strftime("%Y%m%d%H%M%S") + ".png"
+ if not filename.lower().endswith(".png"):
+ filename += ".png"
with open(filename, "wb") as f:
f.write(content)
@@ -87,3 +89,17 @@
abs_path = os.path.abspath(filename)
print(f"[鉁揮 鍥剧墖宸蹭繚瀛橈細{abs_path}")
return abs_path
+
+ def generate_image_to_stream(self, prompt: str) -> BytesIO:
+ if not prompt.strip():
+ raise ValueError("鎻愮ず璇嶄笉鑳戒负绌�")
+
+ translated_prompt = self.translate_prompt(prompt)
+ url = self.build_url(translated_prompt)
+
+ content = self._fetch_with_retry(url)
+ if content is None:
+ return None
+
+ image_stream = BytesIO(content)
+ return image_stream
--
Gitblit v1.9.1