Kims12 commited on
Commit
f41a559
ยท
verified ยท
1 Parent(s): efacf1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -109,7 +109,9 @@ def generate_with_images(prompt, images, max_retries=2):
109
  logger.info(f"Gemini API ์š”์ฒญ ์‹œ์ž‘ - ํ”„๋กฌํ”„ํŠธ: {prompt} (์‹œ๋„: {retries+1}/{max_retries+1})")
110
 
111
  # ์ด๋ฏธ์ง€ ์ถ”๊ฐ€ (์ด๋ฏธ์ง€๊ฐ€ ์žˆ๋Š” ๊ฒฝ์šฐ๋งŒ)
112
- image_parts = []
 
 
113
  for idx, img in enumerate(images, 1):
114
  if img is not None:
115
  # ์ด๋ฏธ์ง€๋ฅผ ๋ฐ”์ดํŠธ๋กœ ๋ณ€ํ™˜
@@ -118,17 +120,18 @@ def generate_with_images(prompt, images, max_retries=2):
118
  image_bytes = buffered.getvalue()
119
 
120
  # ์ด๋ฏธ์ง€ ํŒŒํŠธ ์ƒ์„ฑ
121
- image_part = types.Content({
122
  'inline_data': {
123
  'mime_type': 'image/png',
124
  'data': image_bytes
125
  }
126
  })
127
- image_parts.append(image_part)
128
  logger.info(f"์ด๋ฏธ์ง€ #{idx} ์ถ”๊ฐ€๋จ")
129
 
130
- # ์ด๋ฏธ์ง€์™€ ํ…์ŠคํŠธ ์ปจํ…์ธ  ๊ฒฐํ•ฉ
131
- contents = image_parts + [types.Content({'text': prompt})]
 
 
132
 
133
  # ์ƒ์„ฑ ์„ค์ • - ๊ณต์‹ ๋ฌธ์„œ์— ๋”ฐ๋ผ responseModalities ์„ค์ •
134
  response = client.models.generate_content(
 
109
  logger.info(f"Gemini API ์š”์ฒญ ์‹œ์ž‘ - ํ”„๋กฌํ”„ํŠธ: {prompt} (์‹œ๋„: {retries+1}/{max_retries+1})")
110
 
111
  # ์ด๋ฏธ์ง€ ์ถ”๊ฐ€ (์ด๋ฏธ์ง€๊ฐ€ ์žˆ๋Š” ๊ฒฝ์šฐ๋งŒ)
112
+ contents = []
113
+
114
+ # ์ด๋ฏธ์ง€ ํŒŒํŠธ ์ถ”๊ฐ€
115
  for idx, img in enumerate(images, 1):
116
  if img is not None:
117
  # ์ด๋ฏธ์ง€๋ฅผ ๋ฐ”์ดํŠธ๋กœ ๋ณ€ํ™˜
 
120
  image_bytes = buffered.getvalue()
121
 
122
  # ์ด๋ฏธ์ง€ ํŒŒํŠธ ์ƒ์„ฑ
123
+ contents.append({
124
  'inline_data': {
125
  'mime_type': 'image/png',
126
  'data': image_bytes
127
  }
128
  })
 
129
  logger.info(f"์ด๋ฏธ์ง€ #{idx} ์ถ”๊ฐ€๋จ")
130
 
131
+ # ํ…์ŠคํŠธ ํ”„๋กฌํ”„ํŠธ ์ถ”๊ฐ€
132
+ contents.append({
133
+ 'text': prompt
134
+ })
135
 
136
  # ์ƒ์„ฑ ์„ค์ • - ๊ณต์‹ ๋ฌธ์„œ์— ๋”ฐ๋ผ responseModalities ์„ค์ •
137
  response = client.models.generate_content(