Update app.py

app.py

@@ -24,9 +24,19 @@ logger = logging.getLogger(__name__)
 def get_api_endpoint():
     """환경변수에서 API 엔드포인트를 가져오는 함수"""
     endpoint = os.getenv('API_ENDPOINT')
+    
+    # 환경변수가 없으면 None 반환
     if not endpoint:
-        logger.error("API_ENDPOINT 환경변수가 설정되지 않았습니다.")
         return None
+    
+    # 허깅페이스 스페이스 형식 처리
+    if "/" in endpoint and not endpoint.startswith("http"):
+        # "username/spacename" 형식을 전체 URL로 변환
+        endpoint = f"https://{endpoint.replace('/', '-')}.hf.space"
+    elif not endpoint.startswith("http"):
+        # 단순 도메인인 경우 https 추가
+        endpoint = f"https://{endpoint}"
+    
     return endpoint
 
 def encode_image_to_base64(image):
@@ -59,6 +69,36 @@ def call_api_endpoint(endpoint_url, data):
         return {"error": f"API 호출 실패: {str(e)}"}
 
 # ========== 이미지 생성기 관련 함수 ==========
+def get_api_endpoint():
+    """환경변수에서 API 엔드포인트를 가져오는 함수"""
+    endpoint = os.getenv('API_ENDPOINT')
+    
+    # 환경변수가 없으면 기본값 사용 (테스트용)
+    if not endpoint:
+        endpoint = "https://happydoggg-03hd74jd.hf.space"
+        logger.info("환경변수가 설정되지 않아 기본 엔드포인트를 사용합니다.")
+    
+    # 허깅페이스 스페이스 형식 처리
+    if "/" in endpoint and not endpoint.startswith("http"):
+        # "username/spacename" 형식을 전체 URL로 변환
+        endpoint = f"https://{endpoint.replace('/', '-')}.hf.space"
+    elif not endpoint.startswith("http"):
+        # 단순 도메인인 경우 https 추가
+        endpoint = f"https://{endpoint}"
+    
+    return endpoint
+
+def call_api_endpoint(endpoint_url, data):
+    """API 엔드포인트 호출 함수"""
+    try:
+        response = requests.post(endpoint_url, json=data, timeout=300)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        # 엔드포인트 정보를 숨기고 일반적인 오류 메시지만 표시
+        logger.error("API 호출 중 오류 발생")
+        return {"error": "API 호출 실패"}
+
 def translate_prompt_to_english(prompt):
     """프롬프트 번역 - API 엔드포인트로 전달"""
     endpoint_url = get_api_endpoint()
@@ -73,14 +113,49 @@ def translate_prompt_to_english(prompt):
     try:
         result = call_api_endpoint(f"{endpoint_url}/translate", data)
         if "error" in result:
-            logger.warning(f"번역 API 오류: {result['error']}")
+            logger.warning("번역 API 오류 발생")
             return prompt
         return result.get("translated_prompt", prompt)
     except Exception as e:
-        logger.warning(f"번역 중 오류 발생: {str(e)}")
+        logger.warning("번역 중 오류 발생")
         return prompt
 
-def preprocess_prompt(prompt, image1, image2, image3):
+def generate_with_images(prompt, images, variation_index=0):
+    """API 엔드포인트로 이미지 생성 요청"""
+    endpoint_url = get_api_endpoint()
+    if not endpoint_url:
+        return None, "API 엔드포인트가 설정되지 않았습니다."
+    
+    # 이미지들을 base64로 인코딩
+    encoded_images = []
+    for img in images:
+        if img is not None:
+            encoded_img = encode_image_to_base64(img)
+            encoded_images.append(encoded_img)
+        else:
+            encoded_images.append(None)
+    
+    data = {
+        "function": "generate_with_images",
+        "prompt": prompt,
+        "images": encoded_images,
+        "variation_index": variation_index
+    }
+    
+    try:
+        result = call_api_endpoint(f"{endpoint_url}/generate", data)
+        if "error" in result:
+            return None, "이미지 생성 중 오류가 발생했습니다."
+        
+        # base64 이미지를 PIL 이미지로 변환
+        if "image" in result:
+            generated_image = decode_base64_to_image(result["image"])
+            return generated_image, result.get("status", "이미지가 성공적으로 생성되었습니다.")
+        else:
+            return None, "API에서 이미지를 생성하지 못했습니다."
+    except Exception as e:
+        logger.error("이미지 생성 API 호출 중 오류")
+        return None, "이미지 생성 중 오류가 발생했습니다."
     """프롬프트 전처리 - 기존 로직 유지"""
     has_img1 = image1 is not None
     has_img2 = image2 is not None
@@ -210,12 +285,12 @@ def process_images_with_prompt(image1, image2, image3, prompt, variation_index=0
             else:
                 last_error = status
                 retry_count += 1
-                logger.warning(f"이미지 생성 실패, 재시도 {retry_count}/{max_retries}: {status}")
+                logger.warning(f"이미지 생성 실패, 재시도 {retry_count}/{max_retries}")
                 time.sleep(1)
         except Exception as e:
-            last_error = str(e)
+            last_error = "처리 중 오류가 발생했습니다."
             retry_count += 1
-            logger.exception(f"이미지 처리 중 오류 발생, 재시도 {retry_count}/{max_retries}:")
+            logger.exception(f"이미지 처리 중 오류 발생, 재시도 {retry_count}/{max_retries}")
             time.sleep(1)
     
     return None, f"최대 재시도 횟수({max_retries}회) 초과 후 실패: {last_error}", prompt
@@ -873,12 +948,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Default(
 
 # 애플리케이션 실행
 if __name__ == "__main__":
-    # API 엔드포인트 확인
+    # API 엔드포인트 확인 (로그에는 민감정보 출력하지 않음)
     endpoint = get_api_endpoint()
     if endpoint:
         logger.info("컨트롤 타워가 시작되었습니다.")
     else:
-        logger.warning("API_ENDPOINT 환경변수를 설정해주세요.")
+        logger.warning("API 설정을 확인해주세요.")
     
     demo.queue()
     demo.launch(share=False, inbrowser=True, width="100%")