some fixes
This commit is contained in:
parent
f66aa41a1e
commit
b763933132
1 changed files with 39 additions and 38 deletions
77
main.py
77
main.py
|
@ -4,16 +4,16 @@ from PIL import Image, ImageDraw, ImageFont
|
|||
|
||||
zodiac_signs = [
|
||||
"Belier",
|
||||
"Taureau",
|
||||
"Gemeaux",
|
||||
"Cancer",
|
||||
"Lion",
|
||||
"Vierge",
|
||||
"Balance",
|
||||
"Scorpion",
|
||||
"Sagittaire",
|
||||
"Capricorne",
|
||||
"Taureau",
|
||||
"Lion",
|
||||
"Scorpion",
|
||||
"Verseau",
|
||||
"Gemeaux",
|
||||
"Vierge",
|
||||
"Sagittaire",
|
||||
"Poissons",
|
||||
]
|
||||
|
||||
|
@ -21,52 +21,53 @@ zodiac_signs = [
|
|||
def generate_horoscope():
|
||||
horoscope = {}
|
||||
for sign in zodiac_signs:
|
||||
heart = choice(["+", "++", "+++", "-", "--"])
|
||||
work = choice(["+", "++", "+++", "-", "--"])
|
||||
|
||||
if heart.count("+") > 3:
|
||||
heart = "+++"
|
||||
elif heart.count("-") > 2:
|
||||
heart = "--"
|
||||
|
||||
if work.count("+") > 3:
|
||||
work = "+++"
|
||||
elif work.count("-") > 2:
|
||||
work = "--"
|
||||
|
||||
heart = choice(["+++", "++", "+", "-", "- -"])
|
||||
work = choice(["+++", "++", "+", "-", "- -"])
|
||||
horoscope[sign] = {"heart": heart, "work": work}
|
||||
|
||||
return horoscope
|
||||
|
||||
|
||||
def create_horoscope_image(horoscope):
|
||||
img_width, img_height = 800, 600
|
||||
image = Image.new("RGBA", (img_width, img_height), color="white")
|
||||
draw = ImageDraw.Draw(image, "RGBA")
|
||||
img_width, img_height = 1050, 450
|
||||
image = Image.new("RGBA", (img_width, img_height), color=(255, 255, 255, 0))
|
||||
draw = ImageDraw.Draw(image)
|
||||
|
||||
font = ImageFont.load_default()
|
||||
font = ImageFont.load_default(24)
|
||||
|
||||
x, y = 20, 20
|
||||
x, y = 10, 10
|
||||
for sign, prediction in horoscope.items():
|
||||
sign_image = Image.open(join("images", f"{sign.lower()}.png"))
|
||||
image.paste(sign_image, (x, y))
|
||||
sign_image = Image.open(join("images", f"{sign.lower()}.png")).convert("RGBA")
|
||||
sign_image.thumbnail((100, 100))
|
||||
image.paste(sign_image, (x, y), sign_image)
|
||||
|
||||
draw.text((x, y), sign, fill="black", font=font)
|
||||
draw.text((x, y + 30), f"♥: {prediction['heart']}", fill="red", font=font)
|
||||
draw.text((x, y + 60), f"💼: {prediction['work']}", fill="blue", font=font)
|
||||
draw.text((x + 110, y), sign, fill=(0, 0, 0, 255), font=font)
|
||||
draw.text(
|
||||
(x + 110, y + 40),
|
||||
f"♥: {prediction['heart']}",
|
||||
fill=(255, 0, 0, 255),
|
||||
font=font,
|
||||
)
|
||||
draw.text(
|
||||
(x + 110, y + 80),
|
||||
f"💼: {prediction['work']}",
|
||||
fill=(0, 0, 255, 255),
|
||||
font=font,
|
||||
)
|
||||
|
||||
x += 200
|
||||
if x > 600:
|
||||
x = 20
|
||||
x += 250
|
||||
if x > 900:
|
||||
x = 10
|
||||
y += 150
|
||||
|
||||
return image
|
||||
|
||||
|
||||
new_horoscope = generate_horoscope()
|
||||
if __name__ == "__main__":
|
||||
# Generate new horoscope
|
||||
new_horoscope = generate_horoscope()
|
||||
|
||||
horoscope_image = create_horoscope_image(new_horoscope)
|
||||
# Create and save the image
|
||||
horoscope_image = create_horoscope_image(new_horoscope)
|
||||
horoscope_image.save("nouvel_horoscope.png")
|
||||
|
||||
horoscope_image.save("nouvel_horoscope.png")
|
||||
|
||||
print("Nouvel horoscope généré et sauvegardé sous 'nouvel_horoscope.png'")
|
||||
print("Nouvel horoscope généré et sauvegardé sous 'nouvel_horoscope.png'")
|
||||
|
|
Loading…
Reference in a new issue