feat: 初始版本

tools/generate_images.py

@@ -0,0 +1,40 @@
+# -*- coding: utf-8 -*-
+import urllib2
+import urllib
+import os
+
+# create save directory
+save_dir = r"e:\Temp\desktop-pets\images"
+if not os.path.exists(save_dir):
+    os.makedirs(save_dir)
+
+# image API base URL
+base_url = "https://trae-api-cn.mchost.guru/api/ide/v1/text_to_image"
+
+# cockroach action list
+cockroach_actions = [
+    ("cockroach_idle.png", "pixel art cockroach standing still, cute pixel style, transparent background, 64x64 pixels, dark brown color, detailed antennae, cartoon style"),
+    ("cockroach_walk1.png", "pixel art cockroach walking pose 1, cute pixel style, transparent background, 64x64 pixels, dark brown color, legs in motion, cartoon style"),
+    ("cockroach_walk2.png", "pixel art cockroach walking pose 2, cute pixel style, transparent background, 64x64 pixels, dark brown color, legs in different position, cartoon style"),
+    ("cockroach_crawl.png", "pixel art cockroach crawling low, cute pixel style, transparent background, 64x64 pixels, dark brown color, cartoon style"),
+    ("cockroach_scared.png", "pixel art cockroach scared pose, antennae raised, cute pixel style, transparent background, 64x64 pixels, dark brown color, cartoon style")
+]
+
+# download each image
+for filename, prompt in cockroach_actions:
+    print("Generating: " + filename)
+    url = base_url + "?prompt=" + urllib.quote(prompt) + "&image_size=square_hd"
+    
+    try:
+        response = urllib2.urlopen(url, timeout=60)
+        image_data = response.read()
+        
+        # save image
+        save_path = os.path.join(save_dir, filename)
+        with open(save_path, 'wb') as f:
+            f.write(image_data)
+        print("Saved: " + save_path)
+    except Exception as e:
+        print("Failed to generate " + filename + ": " + str(e))
+
+print("All images generated!")