ginipick commited on
Commit
d6ceac1
ยท
verified ยท
1 Parent(s): 2b46bb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -14
app.py CHANGED
@@ -31,23 +31,47 @@ def create_border_decoration(qr_image, decoration_prompt="flowers"):
31
 
32
  # Parse decoration prompt and draw accordingly
33
  if "flower" in decoration_prompt.lower():
34
- # Draw simple flower patterns in corners
35
- corner_positions = [
36
- (padding//2, padding//2), # Top-left
37
- (new_width - padding//2, padding//2), # Top-right
38
- (padding//2, new_height - padding//2), # Bottom-left
39
- (new_width - padding//2, new_height - padding//2) # Bottom-right
40
- ]
41
 
42
- for x, y in corner_positions:
43
- # Draw flower (simple version)
44
- petal_size = 15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  for angle in range(0, 360, 45): # 8 petals
46
- x1 = x + petal_size * cos(radians(angle))
47
- y1 = y + petal_size * sin(radians(angle))
48
- draw.ellipse([x1-5, y1-5, x1+5, y1+5], fill='pink')
49
- draw.ellipse([x-5, y-5, x+5, y+5], fill='yellow') # Center of flower
 
 
 
50
 
 
 
 
 
 
51
  return decorated_image
52
 
53
  # QR ์ฝ”๋“œ ์ƒ์„ฑ ํ•จ์ˆ˜
 
31
 
32
  # Parse decoration prompt and draw accordingly
33
  if "flower" in decoration_prompt.lower():
34
+ # ๊ฝƒ ํฌ๊ธฐ ์„ค์ •
35
+ flower_size = 12
36
+ gap = flower_size * 2 # ๊ฝƒ ์‚ฌ์ด์˜ ๊ฐ„๊ฒฉ
 
 
 
 
37
 
38
+ # ํ…Œ๋‘๋ฆฌ๋ฅผ ๋”ฐ๋ผ ์ ๋“ค์˜ ์œ„์น˜ ๊ณ„์‚ฐ
39
+ border_points = []
40
+
41
+ # ์ƒ๋‹จ ํ…Œ๋‘๋ฆฌ
42
+ for x in range(padding//2, new_width - padding//2, gap):
43
+ border_points.append((x, padding//2))
44
+
45
+ # ์šฐ์ธก ํ…Œ๋‘๋ฆฌ
46
+ for y in range(padding//2, new_height - padding//2, gap):
47
+ border_points.append((new_width - padding//2, y))
48
+
49
+ # ํ•˜๋‹จ ํ…Œ๋‘๋ฆฌ
50
+ for x in range(new_width - padding//2, padding//2, -gap):
51
+ border_points.append((x, new_height - padding//2))
52
+
53
+ # ์ขŒ์ธก ํ…Œ๋‘๋ฆฌ
54
+ for y in range(new_height - padding//2, padding//2, -gap):
55
+ border_points.append((padding//2, y))
56
+
57
+ # ๊ฐ ์œ„์น˜์— ๊ฝƒ ๊ทธ๋ฆฌ๊ธฐ
58
+ for x, y in border_points:
59
+ # ๊ฝƒ์žŽ ๊ทธ๋ฆฌ๊ธฐ (8๊ฐœ์˜ ๊ฝƒ์žŽ)
60
+ petal_colors = ['pink', 'lightpink', 'peachpuff'] # ๋‹ค์–‘ํ•œ ์ƒ‰์ƒ ์‚ฌ์šฉ
61
  for angle in range(0, 360, 45): # 8 petals
62
+ x1 = x + flower_size * cos(radians(angle))
63
+ y1 = y + flower_size * sin(radians(angle))
64
+ petal_color = petal_colors[angle % len(petal_colors)]
65
+ draw.ellipse([x1-4, y1-4, x1+4, y1+4], fill=petal_color)
66
+
67
+ # ๊ฝƒ ์ค‘์‹ฌ ๊ทธ๋ฆฌ๊ธฐ
68
+ draw.ellipse([x-3, y-3, x+3, y+3], fill='yellow')
69
 
70
+ # ์žŽ ์ถ”๊ฐ€ (์„ ํƒ์ )
71
+ if angle % 180 == 0: # ์–‘์ชฝ์— ์žŽ ์ถ”๊ฐ€
72
+ draw.ellipse([x-8, y+8, x+8, y+16], fill='lightgreen')
73
+ draw.ellipse([x-8, y-16, x+8, y-8], fill='lightgreen')
74
+
75
  return decorated_image
76
 
77
  # QR ์ฝ”๋“œ ์ƒ์„ฑ ํ•จ์ˆ˜