Update app.py

app.py

@@ -9,8 +9,9 @@ import base64
 import mimetypes
 import logging
 
-from google import genai
-from google.genai import types
+# 기존의 "from google import genai" 대신 다음과 같이 수정
+import google.generativeai as genai
+from google.generativeai import types
 
 # .env 파일에 저장된 환경변수 로드 (python-dotenv 설치 필요: pip install python-dotenv)
 from dotenv import load_dotenv
@@ -21,17 +22,14 @@ logging.basicConfig(level=logging.DEBUG,
                     format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
-
 def save_binary_file(file_name, data):
     logger.debug(f"파일에 이진 데이터 저장 중: {file_name}")
     with open(file_name, "wb") as f:
         f.write(data)
     logger.debug(f"파일 저장 완료: {file_name}")
 
-
 def generate(text, file_name, background_file=None, style_file=None, model="gemini-2.0-flash-exp-image-generation"):
     logger.debug(f"generate 함수 시작 - 텍스트: '{text}', 파일명: '{file_name}', 모델: '{model}'")
-
     try:
         # API 키는 환경변수에서 불러옴
         effective_api_key = os.environ.get("GEMINI_API_KEY")
@@ -77,7 +75,6 @@ def generate(text, file_name, background_file=None, style_file=None, model="gemi
             )
         # 스타일 이미지 파트 (존재 시)
         if style_file is not None:
-            # 배경이 없는 경우 uploaded_files[1]가 스타일 이미지가 됨에 주의
             style_index = 2 if background_file is not None else 1
             parts.append(
                 types.Part.from_uri(
@@ -141,8 +138,7 @@ def generate(text, file_name, background_file=None, style_file=None, model="gemi
 
     except Exception as e:
         logger.exception("이미지 생성 중 오류 발생:")
-        return None  # 오류 발생 시 None 반환
-
+        return None
 
 def process_image_and_prompt(original_pil, prompt, background_pil=None, style_pil=None):
     logger.debug(f"process_image_and_prompt 함수 시작 - 프롬프트: '{prompt}'")
@@ -186,12 +182,10 @@ def process_image_and_prompt(original_pil, prompt, background_pil=None, style_pi
             return [result_img]
         else:
             logger.error("generate 함수에서 None 반환됨.")
-            return []  # 오류 시 빈 리스트 반환
-
+            return []
     except Exception as e:
         logger.exception("process_image_and_prompt 함수에서 오류 발생:")
-        return []  # 오류 시 빈 리스트 반환
-
+        return []
 
 # --- Gradio 인터페이스 구성 ---
 with gr.Blocks() as demo:
@@ -231,7 +225,6 @@ with gr.Blocks() as demo:
     )
 
 # --- 테스트 코드 ---
-# 테스트용 더미 이미지 (실제 이미지로 대체 가능)
 dummy_original = Image.new("RGBA", (100, 100), color="red")
 dummy_background = Image.new("RGBA", (100, 100), color="green")
 dummy_style = Image.new("RGBA", (100, 100), color="blue")