backup
This commit is contained in:
@@ -0,0 +1,55 @@
|
||||
"""Regenerate commercial-overview.jpg with a clear commercial carpet scene."""
|
||||
import os, sys
|
||||
try:
|
||||
from google import genai
|
||||
from google.genai import types
|
||||
except ImportError:
|
||||
os.system(f"{sys.executable} -m pip install google-genai --quiet")
|
||||
from google import genai
|
||||
from google.genai import types
|
||||
|
||||
API_KEY = os.environ.get("GEMINI_API_KEY", "AIzaSyB_1p8KvaT_rdNJGPs8HKk8bKsvUlcL6Kg")
|
||||
OUT_DIR = os.path.join(os.path.dirname(os.path.dirname(__file__)), "assets", "images", "services")
|
||||
client = genai.Client(api_key=API_KEY)
|
||||
|
||||
PROMPTS = [
|
||||
(
|
||||
"Wide shot of a modern commercial office building lobby with clean grey carpet throughout. "
|
||||
"Professional corporate interior, glass doors, white walls, overhead lighting. "
|
||||
"The carpet is spotless and freshly cleaned — uniform, well-maintained. "
|
||||
"No people. No machines. No equipment. Professional architectural photography, ultra-realistic."
|
||||
),
|
||||
(
|
||||
"Wide interior shot of a bright commercial building corridor with clean, dark grey commercial carpet. "
|
||||
"Modern office environment, glass partitions, professional lighting. "
|
||||
"The carpet looks freshly cleaned and spotless. "
|
||||
"No people, no equipment. Professional photography, ultra-realistic."
|
||||
),
|
||||
]
|
||||
|
||||
out_path = os.path.join(OUT_DIR, "commercial-overview.jpg")
|
||||
|
||||
for i, prompt in enumerate(PROMPTS):
|
||||
print(f"Attempt {i+1}...")
|
||||
try:
|
||||
resp = client.models.generate_images(
|
||||
model="imagen-4.0-generate-001",
|
||||
prompt=prompt,
|
||||
config=types.GenerateImagesConfig(
|
||||
number_of_images=1, aspect_ratio="4:3",
|
||||
output_mime_type="image/jpeg",
|
||||
safety_filter_level="block_low_and_above",
|
||||
),
|
||||
)
|
||||
if resp.generated_images:
|
||||
b = resp.generated_images[0].image.image_bytes
|
||||
with open(out_path, "wb") as f:
|
||||
f.write(b)
|
||||
print(f"Saved ({len(b)//1024}KB)")
|
||||
break
|
||||
else:
|
||||
print("No image returned")
|
||||
except Exception as e:
|
||||
print(f"Error: {e}")
|
||||
|
||||
print("Done.")
|
||||
Reference in New Issue
Block a user