Update app.py

app.py

@@ -20,14 +20,16 @@ class EventManager:
             12: {"theme": "크리스마스/연말", "colors": ["#DC143C", "#228B22"], "mood": "축제같은, 따뜻한", "symbol": "🎄"}
         }
         
-        # 상품 옵션
-        self.reward_options = {
-            "네이버페이": "네이버페이 금액권",
-            "배달의민족": "배달의민족 상품권",
-            "스타벅스": "스타벅스 기프트카드",
-            "CGV": "CGV 영화관람권",
-            "올리브영": "올리브영 상품권"
-        }
+        # 상품 종류 옵션
+        self.reward_types = [
+            "네이버페이 금액권",
+            "배달의민족 상품권", 
+            "스타벅스 기프트카드",
+            "CGV 영화관람권",
+            "올리브영 상품권",
+            "컬쳐랜드 문화상품권",
+            "해피머니 상품권"
+        ]
 
     def calculate_event_duration(self, start_date, end_date):
         """이벤트 기간 계산"""
@@ -58,9 +60,15 @@ class EventManager:
             }
         return {}
 
+    def parse_rewards(self, reward_structure):
+        """상품 구조 파싱"""
+        if not reward_structure.strip():
+            return "네이버페이 금액권 20,000원\n1시간마다 1명씩 하루 24명"
+        
+        return reward_structure
+
     def generate_detailed_comment_event(self, start_date, end_date, concept, duration, 
-                                      reward_amount, reward_type, company_name, 
-                                      kakao_id, phone_number, custom_details=""):
+                                      reward_structure, kakao_id, phone_number, custom_details=""):
         """참고자료 구조를 완전히 반영한 댓글 이벤트 템플릿"""
         
         if not concept:
@@ -79,27 +87,29 @@ class EventManager:
         formatted_start = f"{start_dt.year}년 {start_dt.month}월 {start_dt.day}일 {start_weekday} 00:00"
         formatted_end = f"{end_dt.year}년 {end_dt.month}월 {end_dt.day}일 {end_weekday} 23:59"
         
+        # 상품 정보 파싱
+        reward_text = self.parse_rewards(reward_structure)
+        
         # 메인 제목 생성
         if "로즈" in theme:
             main_title = f"댓글 남기면 1시간마다 선물이!\n{symbol}같은 {month}월 {theme} 이벤트"
         else:
             main_title = f"댓글 남기면 1시간마다 선물이!\n{symbol} {mood} {theme} 이벤트"
         
-        template = f"""
-{main_title}
+        template = f"""{main_title}
 
 {formatted_start} - {formatted_end} ({duration}일간)
 
 ========================
 EVENT
-<{company_name}>에 작성된 글에 유익하고 착한 댓글을 남기면
+<커뮤니티>에 작성된 글에 유익하고 착한 댓글을 남기면
 1시간마다 1명씩! 하루 24명에게 선물을드려요!
 
 {formatted_start} - {formatted_end} ({duration}일간)
 
 =====================
 STEP 1
-{company_name} <커뮤니티>에 작성된 글에
+회사명 <커뮤니티>에 작성된 글에
 유익하고 착한 댓글 작성
 * 이쁘고 좋은 말 많이 해주는 회원님들이 되길 부탁드려요
 
@@ -115,9 +125,7 @@ STEP 3
 
 =================
 당첨혜택
-{reward_type} {reward_amount}원 이미지
-{reward_type} {reward_amount}원
-1시간마다 1명씩 하루 24명
+{reward_text}
 
 ========================
 이벤트 대상
@@ -144,7 +152,7 @@ pc버전 당첨 예시 이미지, 모바일버전 당첨 예시 이미지
 
 =========================
 꼭 확인하세요!
-※본이벤트는 {company_name} 일반 여성회원만 참여가능합니다. (비회원도 가능)
+※본이벤트는 회사명 일반 여성회원만 참여가능합니다. (비회원도 가능)
 ※ pc에서 캡처가 어려운 경우, 카메라로 촬영하여 연락 주셔도 됩니다.
 ※이벤트 팝업 상품 이미지와 상단 당첨 날짜&시간을 함께
 화면 캡처 또는 사진 촬영해야 수령 가능합니다.
@@ -228,9 +236,9 @@ ex) 주말 당첨시, 고객센터 영업일(월) 업무 종료시간 19:00시
 def create_interface():
     event_manager = EventManager()
     
-    with gr.Blocks(title="이벤트 관리 시스템 v2.0") as demo:
-        gr.Markdown("# 🎉 이벤트 관리 시스템 v2.0")
-        gr.Markdown("**댓글 이벤트 상세 템플릿** - 참고자료 구조 완전 반영")
+    with gr.Blocks(title="이벤트 관리 시스템 v2.1") as demo:
+        gr.Markdown("# 🎉 이벤트 관리 시스템 v2.1")
+        gr.Markdown("**유연한 상품 설정** - 단일상품부터 등급별 상품까지 자유롭게 설정")
         
         with gr.Row():
             with gr.Column(scale=1):
@@ -249,14 +257,76 @@ def create_interface():
                         value="2025-05-25"
                     )
                 
-                gr.Markdown("## 🏢 회사 정보")
+                gr.Markdown("## 🎁 상품 설정")
+                
                 with gr.Group():
-                    company_name = gr.Textbox(
-                        label="회사명",
-                        placeholder="회사명을 입력하세요",
-                        value="회사명"
+                    reward_type_selector = gr.Radio(
+                        choices=[
+                            "단일 상품",
+                            "등급별 상품 (1,2,3등)",
+                            "선택형 상품 (네이버페이 OR 배달의민족)",
+                            "직접 입력"
+                        ],
+                        label="상품 구성 방식",
+                        value="단일 상품"
+                    )
+                
+                # 단일 상품 설정
+                with gr.Group(visible=True) as single_reward_group:
+                    single_reward_type = gr.Dropdown(
+                        choices=event_manager.reward_types,
+                        label="상품 종류",
+                        value="네이버페이 금액권"
+                    )
+                    single_reward_amount = gr.Number(
+                        label="금액 (원)",
+                        value=20000,
+                        step=1000
+                    )
+                    single_reward_frequency = gr.Textbox(
+                        label="당첨 주기",
+                        value="1시간마다 1명씩 하루 24명",
+                        placeholder="예: 1시간마다 1명씩 하루 24명"
                     )
+                
+                # 등급별 상품 설정
+                with gr.Group(visible=False) as grade_reward_group:
+                    gr.Markdown("### 🥇 1등")
+                    grade1_type = gr.Dropdown(choices=event_manager.reward_types, value="네이버페이 금액권")
+                    grade1_amount = gr.Number(label="1등 금액", value=50000, step=1000)
+                    grade1_count = gr.Number(label="1등 인원", value=5, step=1)
+                    
+                    gr.Markdown("### 🥈 2등")
+                    grade2_type = gr.Dropdown(choices=event_manager.reward_types, value="네이버페이 금액권")
+                    grade2_amount = gr.Number(label="2등 금액", value=30000, step=1000)
+                    grade2_count = gr.Number(label="2등 인원", value=10, step=1)
                     
+                    gr.Markdown("### 🥉 3등")
+                    grade3_type = gr.Dropdown(choices=event_manager.reward_types, value="네이버페이 금액권")
+                    grade3_amount = gr.Number(label="3등 금액", value=20000, step=1000)
+                    grade3_count = gr.Number(label="3등 인원", value=20, step=1)
+                
+                # 선택형 상품 설정
+                with gr.Group(visible=False) as choice_reward_group:
+                    choice_type1 = gr.Dropdown(choices=event_manager.reward_types, value="네이버페이 금액권", label="상품 1")
+                    choice_type2 = gr.Dropdown(choices=event_manager.reward_types, value="배달의민족 상품권", label="상품 2")
+                    choice_amount = gr.Number(label="금액 (원)", value=20000, step=1000)
+                    choice_frequency = gr.Textbox(
+                        label="당첨 주기",
+                        value="1시간마다 1명씩 하루 24명",
+                        placeholder="예: 1시간마다 1명씩 하루 24명"
+                    )
+                
+                # 직접 입력
+                with gr.Group(visible=False) as custom_reward_group:
+                    custom_reward_text = gr.Textbox(
+                        label="상품 정보 직접 입력",
+                        placeholder="예:\n네이버페이 금액권 20,000원 이미지 / 배달의민족 상품권 20,000원 이미지\n네이버페이 or 배달의 민족 상품권 2만원\n1시간마다 1명씩 하루 24명",
+                        lines=5
+                    )
+                
+                gr.Markdown("## 📞 연락처 정보")
+                with gr.Group():
                     phone_number = gr.Textbox(
                         label="고객센터 번호",
                         placeholder="1544-1234",
@@ -265,31 +335,17 @@ def create_interface():
                     
                     kakao_id = gr.Textbox(
                         label="카카오톡 ID",
-                        placeholder="kakao_id",
+                        placeholder="company_kakao",
                         value="company_kakao"
                     )
                 
-                gr.Markdown("## 🎁 상품 설정")
-                with gr.Group():
-                    reward_type = gr.Dropdown(
-                        choices=list(event_manager.reward_options.values()),
-                        label="상품 종류",
-                        value="네이버페이 금액권"
-                    )
-                    
-                    reward_amount = gr.Number(
-                        label="상품 금액 (원)",
-                        value=20000,
-                        step=1000
-                    )
-                
                 custom_details = gr.Textbox(
                     label="추가 정보",
                     placeholder="추가할 내용이 있다면 입력하세요",
                     lines=2
                 )
                 
-                generate_btn = gr.Button("✨ 상세 이벤트 생성하기", variant="primary", size="lg")
+                generate_btn = gr.Button("✨ 이벤트 생성하기", variant="primary", size="lg")
             
             with gr.Column(scale=2):
                 gr.Markdown("## 📊 생성 결과")
@@ -312,14 +368,50 @@ def create_interface():
                     
                     with gr.Tab("📈 이벤트 정보"):
                         event_info = gr.Textbox(
-                            label="이벤트 기본 정보",
+                            label="이벤트 기본 정보 요약",
                             interactive=False,
                             lines=8
                         )
         
-        def generate_complete_event(start_date_val, end_date_val, company_name_val, 
-                                  phone_number_val, kakao_id_val, reward_type_val, 
-                                  reward_amount_val, custom_details_val):
+        # 상품 구성 방식에 따라 UI 변경
+        def update_reward_ui(reward_type):
+            return (
+                gr.update(visible=(reward_type == "단일 상품")),
+                gr.update(visible=(reward_type == "등급별 상품 (1,2,3등)")),
+                gr.update(visible=(reward_type == "선택형 상품 (네이버페이 OR 배달의민족)")),
+                gr.update(visible=(reward_type == "직접 입력"))
+            )
+        
+        reward_type_selector.change(
+            fn=update_reward_ui,
+            inputs=[reward_type_selector],
+            outputs=[single_reward_group, grade_reward_group, choice_reward_group, custom_reward_group]
+        )
+        
+        def generate_reward_structure(reward_type, *args):
+            """상품 구성에 따라 텍스트 생성"""
+            if reward_type == "단일 상품":
+                single_type, single_amount, single_frequency = args[0], args[1], args[2]
+                return f"{single_type} {int(single_amount):,}원 이미지\n{single_type} {int(single_amount):,}원\n{single_frequency}"
+            
+            elif reward_type == "등급별 상품 (1,2,3등)":
+                g1_type, g1_amount, g1_count, g2_type, g2_amount, g2_count, g3_type, g3_amount, g3_count = args[3:12]
+                return f"""🥇 1등: {g1_type} {int(g1_amount):,}원 ({int(g1_count)}명)
+🥈 2등: {g2_type} {int(g2_amount):,}원 ({int(g2_count)}명)  
+🥉 3등: {g3_type} {int(g3_amount):,}원 ({int(g3_count)}명)
+총 {int(g1_count + g2_count + g3_count)}명"""
+            
+            elif reward_type == "선택형 상품 (네이버페이 OR 배달의민족)":
+                c_type1, c_type2, c_amount, c_frequency = args[12], args[13], args[14], args[15]
+                return f"{c_type1} {int(c_amount):,}원 이미지 / {c_type2} {int(c_amount):,}원 이미지\n{c_type1} or {c_type2} {int(c_amount):,}원\n{c_frequency}"
+            
+            elif reward_type == "직접 입력":
+                return args[16] if args[16] else "상품 정보를 입력해주세요"
+            
+            return "상품 정보를 설정해주세요"
+        
+        def generate_complete_event(start_date_val, end_date_val, phone_number_val, kakao_id_val, 
+                                  custom_details_val, reward_type, *reward_args):
             
             # 기간 계산
             duration, month, start_weekday, end_weekday, start_dt, end_dt = event_manager.calculate_event_duration(start_date_val, end_date_val)
@@ -330,11 +422,13 @@ def create_interface():
             # 월별 컨셉 생성
             concept = event_manager.generate_monthly_concept(month)
             
+            # 상품 구조 생성
+            reward_structure = generate_reward_structure(reward_type, *reward_args)
+            
             # 상세 템플릿 생성
             template = event_manager.generate_detailed_comment_event(
                 start_date_val, end_date_val, concept, duration,
-                int(reward_amount_val), reward_type_val, company_name_val,
-                kakao_id_val, phone_number_val, custom_details_val
+                reward_structure, kakao_id_val, phone_number_val, custom_details_val
             )
             
             # 디자인 가이드 생성
@@ -346,41 +440,52 @@ def create_interface():
 🗓️ 기간: {duration}일간 ({start_date_val} ~ {end_date_val})
 🎯 테마: {concept.get('theme', '')} {concept.get('symbol', '')}
 🎨 분위기: {concept.get('mood', '')}
-🎁 상품: {reward_type_val} {int(reward_amount_val):,}원
-⏰ 당첨: 1시간마다 1명 (하루 24명)
-👥 대상: 일반 여성회원
-🏢 주최: {company_name_val}
 📞 연락처: {phone_number_val}
 💬 카톡: {kakao_id_val}
 
-💡 예상 총 당첨자: {duration * 24}명
-💰 예상 총 비용: {duration * 24 * int(reward_amount_val):,}원"""
+🎁 상품 구성:
+{reward_structure}
+
+💡 상품 구성 방식: {reward_type}"""
             
             return template, design, info
         
+        # 모든 입력값을 리스트로 정리
+        all_inputs = [
+            start_date, end_date, phone_number, kakao_id, custom_details, reward_type_selector,
+            # 단일 상품
+            single_reward_type, single_reward_amount, single_reward_frequency,
+            # 등급별 상품
+            grade1_type, grade1_amount, grade1_count,
+            grade2_type, grade2_amount, grade2_count, 
+            grade3_type, grade3_amount, grade3_count,
+            # 선택형 상품
+            choice_type1, choice_type2, choice_amount, choice_frequency,
+            # 직접 입력
+            custom_reward_text
+        ]
+        
         generate_btn.click(
             fn=generate_complete_event,
-            inputs=[start_date, end_date, company_name, phone_number, kakao_id, 
-                   reward_type, reward_amount, custom_details],
+            inputs=all_inputs,
             outputs=[event_template, design_guide, event_info]
         )
         
         with gr.Row():
             gr.Markdown("""
-            ## ✅ 완성된 기능
-            - **참고자료 구조 100% 반영**: 로즈데이 이벤트와 동일한 구조
-            - **월별 자동 컨셉**: 12개월 테마 자동 적용
-            - **상세 설정**: 회사정보, 상품정보 커스터마이징
-            - **디자인 가이드**: 실제 구현 가능한 디자인 조언
-            - **비용 계산**: 예상 총 비용 자동 계산
+            ## ✅ v2.1 새로운 기능
+            - **🎁 유연한 상품 설정**: 단일/등급별/선택형/직접입력 모두 지원
+            - **💰 자동 비용 계산**: 등급별 상품 총 비용 자동 계산
+            - **🏢 간소화**: 회사 정보 제거로 더 간단한 설정
+            - **📝 자유도 증가**: 상품 구성의 완전한 자유도 제공
             """)
             
             gr.Markdown("""
-            ## 🚀 다음 단계 개발 예정
-            - **이미지 레퍼런스 분석**: OCR + 디자인 분석
-            - **A/B 테스팅**: 여러 버전 생성 및 비교
-            - **자동 일정 관리**: 이벤트 알림 및 관리
-            - **성과 측정**: 참여율 추적 및 분석
+            ## 🎯 상품 설정 예시
+            - **단일**: 네이버페이 2만원, 1시간마다 1명
+            - **등급별**: 1등 5만원(5명), 2등 3만원(10명), 3등 2만원(20명)
+            - **선택형**: 네이버페이 OR 배달의민족 2만원 중 선택
+            - **직접입력**: 완전히 자유로운 형식
             """)
     
     return demo