Update app.py
Browse files
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 |
-
|
|
|
|
|
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 |
-
|
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
|
|
|
|
|
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(
|