Update app.py

app.py

@@ -65,32 +65,28 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
     genai.configure(api_key=api_key)
 
     # 이미지 파일 업로드
-    uploaded_file = genai.upload_file(path=file_name)
-
+    with open(file_name, "rb") as f:
+        image_data = f.read()
+    
+    # 최신 Gemini API 구조에 맞춰 요청 구성
+    model = genai.GenerativeModel(model)
+    
     # API에 전달할 content 구성
-    contents = [
-        genai_types.Content(
-            role="user",
-            parts=[
-                # 먼저 업로드된 파일 URI를 포함
-                genai_types.Part.from_uri(
-                    file_uri=uploaded_file.uri,
-                    mime_type=uploaded_file.mime_type,
-                ),
-                # 이어서 text 지시사항을 포함
-                genai_types.Part.from_text(text=text),
-            ],
-        ),
+    image_parts = [
+        {
+            "mime_type": "image/png",
+            "data": image_data
+        },
+        text
     ]
-
+    
     # 생성(편집) 설정
-    generation_config = genai_types.GenerationConfig(
-        temperature=1,
-        top_p=0.95,
-        top_k=40,
-        max_output_tokens=8192,      # 출력 토큰 제한
-        response_mime_type="text/plain",
-    )
+    generation_config = {
+        "temperature": 1,
+        "top_p": 0.95,
+        "top_k": 40,
+        "max_output_tokens": 8192,
+    }
 
     text_response = ""   # API가 반환한 텍스트 누적
     image_path = None    # API가 반환한 이미지 파일의 로컬 경로
@@ -99,34 +95,28 @@ def generate_by_google_genai(text, file_name, model="gemini-2.0-flash-exp"):
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
         temp_path = tmp.name
 
-        # 스트리밍 형태로 응답을 받음
-        response = genai.generate_content(
-            model=model,
-            contents=contents,
-            generation_config=generation_config,
-            stream=True
-        )
-
-        # 스트리밍된 chunk들에서 이미지와 텍스트를 추출
-        for chunk in response:
-            for candidate in chunk.candidates:
-                for part in candidate.content.parts:
-                    # 이미지인 경우
-                    if hasattr(part, 'inline_data') and part.inline_data:
-                        save_binary_file(temp_path, part.inline_data.data)
-                        image_path = temp_path
-                        break
-                    # 텍스트인 경우
-                    elif hasattr(part, 'text'):
-                        text_response += part.text + "\n"
-
-                if image_path:
-                    break
-            if image_path:
-                break
+        try:
+            # 스트리밍 형태로 응답을 받음
+            response = model.generate_content(
+                contents=image_parts,
+                generation_config=generation_config,
+                stream=True
+            )
 
-    # 업로드된 임시 파일 삭제
-    genai.delete_file(uploaded_file.name)
+            # 스트리밍된 chunk들에서 이미지와 텍스트를 추출
+            for chunk in response:
+                # 이미지 데이터가 있는지 확인
+                if hasattr(chunk, 'parts'):
+                    for part in chunk.parts:
+                        if hasattr(part, 'inline_data') and part.inline_data:
+                            save_binary_file(temp_path, part.inline_data.data)
+                            image_path = temp_path
+                        elif hasattr(part, 'text'):
+                            text_response += part.text
+        
+        except Exception as e:
+            logging.error(f"Gemini API error: {e}")
+            return None, str(e)
 
     return image_path, text_response