diff --git a/smarketing-ai/app.py b/smarketing-ai/app.py index 05754aa..6a4b9d4 100644 --- a/smarketing-ai/app.py +++ b/smarketing-ai/app.py @@ -12,7 +12,7 @@ from config.config import Config from services.poster_service import PosterService from services.sns_content_service import SnsContentService from models.request_models import ContentRequest, PosterRequest, SnsContentGetRequest, PosterContentGetRequest -from services.poster_service_v2 import PosterServiceV2 +from services.poster_service_v3 import PosterServiceV3 def create_app(): @@ -30,7 +30,7 @@ def create_app(): # 서비스 인스턴스 생성 poster_service = PosterService() - poster_service_v2 = PosterServiceV2() + poster_service_v3 = PosterServiceV3() sns_content_service = SnsContentService() @app.route('/health', methods=['GET']) @@ -97,8 +97,8 @@ def create_app(): @app.route('/api/ai/poster', methods=['GET']) def generate_poster_content(): """ - 홍보 포스터 생성 API (개선된 버전) - 원본 이미지 보존 + 한글 텍스트 오버레이 + 홍보 포스터 생성 API + 실제 제품 이미지를 포함한 분위기 배경 포스터 생성 """ try: # JSON 요청 데이터 검증 @@ -115,6 +115,23 @@ def create_app(): if field not in data: return jsonify({'error': f'필수 필드가 누락되었습니다: {field}'}), 400 + # 날짜 변환 처리 + start_date = None + end_date = None + if data.get('startDate'): + try: + from datetime import datetime + start_date = datetime.strptime(data['startDate'], '%Y-%m-%d').date() + except ValueError: + return jsonify({'error': 'startDate 형식이 올바르지 않습니다. YYYY-MM-DD 형식을 사용하세요.'}), 400 + + if data.get('endDate'): + try: + from datetime import datetime + end_date = datetime.strptime(data['endDate'], '%Y-%m-%d').date() + except ValueError: + return jsonify({'error': 'endDate 형식이 올바르지 않습니다. YYYY-MM-DD 형식을 사용하세요.'}), 400 + # 요청 모델 생성 poster_request = PosterContentGetRequest( title=data.get('title'), @@ -127,16 +144,18 @@ def create_app(): emotionIntensity=data.get('emotionIntensity'), menuName=data.get('menuName'), eventName=data.get('eventName'), - startDate=data.get('startDate'), - endDate=data.get('endDate') + startDate=start_date, + endDate=end_date ) - # 포스터 생성 - # result = poster_service.generate_poster(poster_request) - result = poster_service_v2.generate_poster(poster_request) + # 포스터 생성 (V3 사용) + result = poster_service_v3.generate_poster(poster_request) if result['success']: - return jsonify({'content': result['content']}) + return jsonify({ + 'content': result['content'], + 'analysis': result.get('analysis', {}) + }) else: return jsonify({'error': result['error']}), 500 diff --git a/smarketing-ai/config/config.py b/smarketing-ai/config/config.py index de6f276..6b63540 100644 --- a/smarketing-ai/config/config.py +++ b/smarketing-ai/config/config.py @@ -4,7 +4,10 @@ Flask 애플리케이션 설정 """ import os from dotenv import load_dotenv + load_dotenv() + + class Config: """애플리케이션 설정 클래스""" # Flask 기본 설정 @@ -19,8 +22,9 @@ class Config: ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'webp'} # 템플릿 설정 POSTER_TEMPLATE_PATH = 'templates/poster_templates' + @staticmethod def allowed_file(filename): """업로드 파일 확장자 검증""" return '.' in filename and \ - filename.rsplit('.', 1)[1].lower() in Config.ALLOWED_EXTENSIONS \ No newline at end of file + filename.rsplit('.', 1)[1].lower() in Config.ALLOWED_EXTENSIONS diff --git a/smarketing-ai/models/request_models.py b/smarketing-ai/models/request_models.py index b47b257..653dfb8 100644 --- a/smarketing-ai/models/request_models.py +++ b/smarketing-ai/models/request_models.py @@ -4,6 +4,7 @@ API 요청 데이터 구조를 정의 """ from dataclasses import dataclass from typing import List, Optional +from datetime import date @dataclass @@ -19,8 +20,8 @@ class SnsContentGetRequest: emotionIntensity: Optional[str] = None menuName: Optional[str] = None eventName: Optional[str] = None - startDate: Optional[str] = None - endDate: Optional[str] = None + startDate: Optional[date] = None # LocalDate -> date + endDate: Optional[date] = None # LocalDate -> date @dataclass @@ -36,8 +37,8 @@ class PosterContentGetRequest: emotionIntensity: Optional[str] = None menuName: Optional[str] = None eventName: Optional[str] = None - startDate: Optional[str] = None - endDate: Optional[str] = None + startDate: Optional[date] = None # LocalDate -> date + endDate: Optional[date] = None # LocalDate -> date # 기존 모델들은 유지 diff --git a/smarketing-ai/services/poster_service_v3.py b/smarketing-ai/services/poster_service_v3.py new file mode 100644 index 0000000..d55662c --- /dev/null +++ b/smarketing-ai/services/poster_service_v3.py @@ -0,0 +1,204 @@ +""" +포스터 생성 서비스 V3 +OpenAI DALL-E를 사용한 이미지 생성 (메인 메뉴 이미지 1개 + 프롬프트 내 예시 링크 10개) +""" +import os +from typing import Dict, Any, List +from datetime import datetime +from utils.ai_client import AIClient +from utils.image_processor import ImageProcessor +from models.request_models import PosterContentGetRequest + + +class PosterServiceV3: + """포스터 생성 서비스 V3 클래스""" + + def __init__(self): + """서비스 초기화""" + self.ai_client = AIClient() + self.image_processor = ImageProcessor() + + # Azure Blob Storage 예시 이미지 링크 10개 (카페 음료 관련) + self.example_images = [ + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example1.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example2.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example3.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example4.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example5.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example6.png", + "https://stdigitalgarage02.blob.core.windows.net/ai-content/example7.png" + ] + + # 포토 스타일별 프롬프트 + self.photo_styles = { + '미니멀': '미니멀하고 깔끔한 디자인, 단순함, 여백 활용', + '모던': '현대적이고 세련된 디자인, 깔끔한 레이아웃', + '빈티지': '빈티지 느낌, 레트로 스타일, 클래식한 색감', + '컬러풀': '다채로운 색상, 밝고 생동감 있는 컬러', + '우아한': '우아하고 고급스러운 느낌, 세련된 분위기', + '캐주얼': '친근하고 편안한 느낌, 접근하기 쉬운 디자인' + } + + # 카테고리별 이미지 스타일 + self.category_styles = { + '음식': '음식 사진, 먹음직스러운, 맛있어 보이는', + '매장': '레스토랑 인테리어, 아늑한 분위기', + '이벤트': '홍보용 디자인, 눈길을 끄는', + '메뉴': '메뉴 디자인, 정리된 레이아웃', + '할인': '세일 포스터, 할인 디자인', + '음료': '시원하고 상쾌한, 맛있어 보이는 음료' + } + + # 톤앤매너별 디자인 스타일 + self.tone_styles = { + '친근한': '따뜻하고 친근한 색감, 부드러운 느낌', + '정중한': '격식 있고 신뢰감 있는 디자인', + '재미있는': '밝고 유쾌한 분위기, 활기찬 색상', + '전문적인': '전문적이고 신뢰할 수 있는 디자인' + } + + # 감정 강도별 디자인 + self.emotion_designs = { + '약함': '은은하고 차분한 색감, 절제된 표현', + '보통': '적당히 활기찬 색상, 균형잡힌 디자인', + '강함': '강렬하고 임팩트 있는 색상, 역동적인 디자인' + } + + def generate_poster(self, request: PosterContentGetRequest) -> Dict[str, Any]: + """ + 포스터 생성 (메인 이미지 1개 분석 + 예시 링크 10개 프롬프트 제공) + """ + try: + # 메인 이미지 확인 + if not request.images: + return {'success': False, 'error': '메인 메뉴 이미지가 제공되지 않았습니다.'} + + main_image_url = request.images[0] # 첫 번째 이미지가 메인 메뉴 + + # 메인 이미지 분석 + main_image_analysis = self._analyze_main_image(main_image_url) + + # 포스터 생성 프롬프트 생성 (예시 링크 10개 포함) + prompt = self._create_poster_prompt_v3(request, main_image_analysis) + + # OpenAI로 이미지 생성 + image_url = self.ai_client.generate_image_with_openai(prompt, "1024x1024") + + return { + 'success': True, + 'content': image_url, + 'analysis': { + 'main_image': main_image_analysis, + 'example_images_used': len(self.example_images) + } + } + + except Exception as e: + return { + 'success': False, + 'error': str(e) + } + + def _analyze_main_image(self, image_url: str) -> Dict[str, Any]: + """ + 메인 메뉴 이미지 분석 + """ + temp_files = [] + try: + # 이미지 다운로드 + temp_path = self.ai_client.download_image_from_url(image_url) + if temp_path: + temp_files.append(temp_path) + + # 이미지 분석 + image_info = self.image_processor.get_image_info(temp_path) + image_description = self.ai_client.analyze_image(temp_path) + colors = self.image_processor.analyze_colors(temp_path, 5) + + return { + 'url': image_url, + 'info': image_info, + 'description': image_description, + 'dominant_colors': colors, + 'is_food': self.image_processor.is_food_image(temp_path) + } + else: + return { + 'url': image_url, + 'error': '이미지 다운로드 실패' + } + + except Exception as e: + return { + 'url': image_url, + 'error': str(e) + } + finally: + # 임시 파일 정리 + for temp_file in temp_files: + try: + os.remove(temp_file) + except: + pass + + def _create_poster_prompt_v3(self, request: PosterContentGetRequest, + main_analysis: Dict[str, Any]) -> str: + """ + V3 포스터 생성을 위한 AI 프롬프트 생성 (한글, 글자 완전 제외, 메인 이미지 기반 + 예시 링크 10개 포함) + """ + # 기본 스타일 설정 + photo_style = self.photo_styles.get(request.photoStyle, '현대적이고 깔끔한 디자인') + category_style = self.category_styles.get(request.category, '홍보용 디자인') + tone_style = self.tone_styles.get(request.toneAndManner, '친근하고 따뜻한 느낌') + emotion_design = self.emotion_designs.get(request.emotionIntensity, '적당히 활기찬 디자인') + + # 메인 이미지 정보 활용 + main_description = main_analysis.get('description', '맛있는 음식') + main_colors = main_analysis.get('dominant_colors', []) + main_image_url = main_analysis.get('url', '') + image_info = main_analysis.get('info', {}) + is_food = main_analysis.get('is_food', False) + + # 이미지 크기 및 비율 정보 + aspect_ratio = image_info.get('aspect_ratio', 1.0) if image_info else 1.0 + image_orientation = "가로형" if aspect_ratio > 1.2 else "세로형" if aspect_ratio < 0.8 else "정사각형" + + # 색상 정보를 텍스트로 변환 + color_description = "" + if main_colors: + color_rgb = main_colors[:3] # 상위 3개 색상 + color_description = f"주요 색상 RGB 값: {color_rgb}를 기반으로 한 조화로운 색감" + + # 예시 이미지 링크들을 문자열로 변환 + example_links = "\n".join([f"- {link}" for link in self.example_images]) + + prompt = f""" + 메인 이미지 URL을 참조하여, "글이 없는" 심플한 카페 포스터를 디자인해주세요. + + **핵심 기준 이미지:** + 메인 이미지 URL: {main_image_url} + 이 이미지 URL에 들어가 이미지를 다운로드 후, 이 이미지를 그대로 반영한 채 홍보 포스터를 디자인해주세요. + 심플한 배경이 중요합니다. + AI가 생성하지 않은 것처럼 현실적인 요소를 반영해주세요. + + **절대 필수 조건:** + - 어떤 형태의 텍스트, 글자, 문자, 숫자도 절대 포함하지 말 것!!!! - 가장 중요 + - 위의 메인 이미지를 임의 변경 없이, 포스터의 중심 요소로 포함할 것 + - 하나의 포스터만 생성해주세요 + - 메인 이미지의 색감과 분위기를 살려서 심플한 포스터 디자인 + - 메인 이미지가 돋보이도록 배경과 레이아웃 구성 + - 확실하지도 않은 문자 절대 생성 x + + **특별 요구사항:** + {request.requirement} + + + + **반드시 제외할 요소:** + - 모든 형태의 텍스트 (한글, 영어, 숫자, 기호) + - 메뉴판, 가격표, 간판 + - 글자가 적힌 모든 요소 + - 브랜드 로고나 문자 + + """ + return prompt diff --git a/smarketing-ai/uploads/temp/temp_1e7658c3-43ba-4d61-9bfc-4c1d9c2a5098.jpg b/smarketing-ai/uploads/temp/temp_1e7658c3-43ba-4d61-9bfc-4c1d9c2a5098.jpg deleted file mode 100644 index 3d8f70d..0000000 Binary files a/smarketing-ai/uploads/temp/temp_1e7658c3-43ba-4d61-9bfc-4c1d9c2a5098.jpg and /dev/null differ diff --git a/smarketing-ai/uploads/temp/temp_44b7841c-56e8-4c94-b769-4580f00f7723.jpg b/smarketing-ai/uploads/temp/temp_44b7841c-56e8-4c94-b769-4580f00f7723.jpg deleted file mode 100644 index 4216da1..0000000 Binary files a/smarketing-ai/uploads/temp/temp_44b7841c-56e8-4c94-b769-4580f00f7723.jpg and /dev/null differ