refactor: classes

app.py

@@ -37,6 +37,21 @@ import ssl
 # Fix SSL error for model download (macOS specific)
 ssl._create_default_https_context = ssl._create_unverified_context
 
+# ========== 여기만 수정하면 탭이 동적으로 반영됩니다 ==========
+# 예시: ["Response반", "Challenge반"]  → 연습 탭 + 2개 반 = 총 3개 탭
+# 예시: ["206호", "207호", "305호"]    → 연습 탭 + 3개 반 = 총 4개 탭
+CLASSES = ["Respect반", "Challenge반", "Originality반", "BCE반"]
+# ============================================================
+
+# 탭 아이콘 자동 배정
+_TAB_ICONS = ["🎯", "⚡", "🎨", "🔥", "💎", "🌟", "🚀", "💡", "🏅", "✨"]
+
+def _class_key(name):
+    """반 이름을 json 파일명/딕셔너리 키로 변환 (예: 'Respect반' → 'respect')"""
+    # 한글 '반' 제거, 공백→언더스코어, 소문자
+    return name.replace("반", "").replace("호", "").strip().lower().replace(" ", "_")
+
+
 class ImageSimilarityLeaderboard:
     def __init__(self, reference_image_path="label2.jpg", data_file="leaderboard.json"):
         self.reference_image_path = reference_image_path
@@ -431,93 +446,43 @@ class ImageSimilarityLeaderboard:
         self.leaderboard_data = []
         self.save_leaderboard()
         return "✅ 리더보드가 초기화되었습니다.", pd.DataFrame(columns=['순위', '사용자명', '점수', '날짜'])
-    
 
-# 전역 리더보드 인스턴스들 (5개 탭: 4개 반 + 연습)
-leaderboards = {
-    'respect': ImageSimilarityLeaderboard("label2.jpg", "respect.json"),
-    'challenge': ImageSimilarityLeaderboard("label2.jpg", "challenge.json"),
-    'originality': ImageSimilarityLeaderboard("label2.jpg", "originality.json"),
-    'bce': ImageSimilarityLeaderboard("label2.jpg", "bce.json"),
-    'practice': ImageSimilarityLeaderboard("label2.jpg", "practice.json")
-}
-
-def process_user_image_respect(image, username):
-    """Respect반 사용자 이미지 처리 함수"""
-    return leaderboards['respect'].process_image(image, username)
-
-def process_user_image_challenge(image, username):
-    """Challenge반 사용자 이미지 처리 함수"""
-    return leaderboards['challenge'].process_image(image, username)
-
-def process_user_image_originality(image, username):
-    """Originality반 사용자 이미지 처리 함수"""
-    return leaderboards['originality'].process_image(image, username)
-
-def process_user_image_bce(image, username):
-    """BCE반 사용자 이미지 처리 함수"""
-    return leaderboards['bce'].process_image(image, username)
-
-def get_current_leaderboard_respect():
-    """Respect반 현재 리더보드 반환"""
-    return leaderboards['respect'].get_leaderboard_df()
-
-def get_current_leaderboard_challenge():
-    """Challenge반 현재 리더보드 반환"""
-    return leaderboards['challenge'].get_leaderboard_df()
-
-def get_current_leaderboard_originality():
-    """Originality반 현재 리더보드 반환"""
-    return leaderboards['originality'].get_leaderboard_df()
-
-def get_current_leaderboard_bce():
-    """BCE반 현재 리더보드 반환"""
-    return leaderboards['bce'].get_leaderboard_df()
-
-def check_and_update_leaderboard_respect():
-    """Respect반 리더보드 업데이트 확인 및 반환"""
-    leaderboards['respect'].check_for_updates()
-    return leaderboards['respect'].get_leaderboard_df()
-
-def check_and_update_leaderboard_challenge():
-    """Challenge반 리더보드 업데이트 확인 및 반환"""
-    leaderboards['challenge'].check_for_updates()
-    return leaderboards['challenge'].get_leaderboard_df()
-
-def check_and_update_leaderboard_originality():
-    """Originality반 리더보드 업데이트 확인 및 반환"""
-    leaderboards['originality'].check_for_updates()
-    return leaderboards['originality'].get_leaderboard_df()
-
-def check_and_update_leaderboard_bce():
-    """BCE반 리더보드 업데이트 확인 및 반환"""
-    leaderboards['bce'].check_for_updates()
-    return leaderboards['bce'].get_leaderboard_df()
-
-def process_user_image_practice(image, username):
-    """연습 탭 사용자 이미지 처리 함수"""
-    return leaderboards['practice'].process_image(image, username)
-
-def get_current_leaderboard_practice():
-    """연습 탭 현재 리더보드 반환"""
-    return leaderboards['practice'].get_leaderboard_df()
-
-def check_and_update_leaderboard_practice():
-    """연습 탭 리더보드 업데이트 확인 및 반환"""
-    leaderboards['practice'].check_for_updates()
-    return leaderboards['practice'].get_leaderboard_df()
+
+# ── 전역 리더보드 인스턴스 (CLASSES 기반 동적 생성 + 연습) ──
+leaderboards = {}
+for _cls in CLASSES:
+    leaderboards[_class_key(_cls)] = ImageSimilarityLeaderboard("label2.jpg", f"{_class_key(_cls)}.json")
+leaderboards['practice'] = ImageSimilarityLeaderboard("label2.jpg", "practice.json")
+
+
+# ── 범용 처리 함수 ──
+def process_user_image(class_key, image, username):
+    """이미지 처리 (모든 반 공용)"""
+    return leaderboards[class_key].process_image(image, username)
+
+def get_current_leaderboard(class_key):
+    """현재 리더보드 반환 (모든 반 공용)"""
+    return leaderboards[class_key].get_leaderboard_df()
+
+def check_and_update_leaderboard(class_key):
+    """리더보드 업데이트 확인 및 반환 (모든 반 공용)"""
+    leaderboards[class_key].check_for_updates()
+    return leaderboards[class_key].get_leaderboard_df()
+
 
 def create_interface():
-    """Gradio 인터페이스 생성"""
+    """Gradio 인터페이스 생성 (CLASSES 기반 동적 탭)"""
     with gr.Blocks(title="비슷한 이미지를 만들어주세요!", theme=gr.themes.Soft()) as demo:
         gr.Markdown("""
         # 🏆 롯데 비전 스튜디오 리더보드
         """)
         
+        # 모든 리더보드 출력 위젯을 모아둘 딕셔너리
+        leaderboard_outputs = {}
         
         # 탭 생성
         with gr.Tabs():
-            # 연습 탭 (메인)
+            # ── 연습 탭 (항상 첫 번째) ──
             with gr.Tab("🎓 연습"):
                 gr.Markdown("### 💡 연습용 탭 (메인)")
                 gr.Markdown("업로드 및 이미지 유사도를 테스트해볼 수 있는 연습 공간입니다.")
@@ -553,204 +518,75 @@ def create_interface():
                             interactive=False
                         )
 
+                leaderboard_outputs['practice'] = leaderboard_output_practice
+
                 # 연습 탭 이벤트 핸들러
                 submit_btn_practice.click(
-                    fn=process_user_image_practice,
+                    fn=lambda img, name: process_user_image('practice', img, name),
                     inputs=[image_input_practice, username_input_practice],
                     outputs=[result_output_practice, leaderboard_output_practice]
                 )
 
-            # Respect반 탭
-            with gr.Tab("🎯 Respect반"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📤 이미지 업로드 (Respect반)")
-                        image_input_respect = gr.Image(
-                            label="비교할 이미지를 업로드하세요",
-                            type="pil",
-                            height=300
-                        )
-                        username_input_respect = gr.Textbox(
-                            label="사용자 이름",
-                            placeholder="이름을 입력하세요",
-                            max_lines=1
-                        )
-                        submit_btn_respect = gr.Button("🚀 점수 계산 및 등록", variant="primary", size="lg")
-
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📊 결과 (Respect반)")
-                        result_output_respect = gr.Textbox(
-                            label="계산 결과",
-                            lines=10,
-                            interactive=False
-                        )
-
-                        gr.Markdown("### 🏅 Respect반 리더보드")
-                        leaderboard_output_respect = gr.Dataframe(
-                            headers=["순위", "사용자명", "점수", "날짜"],
-                            datatype=["number", "str", "number", "str"],
-                            interactive=False
-                        )
-
-                # Respect반 이벤트 핸들러
-                submit_btn_respect.click(
-                    fn=process_user_image_respect,
-                    inputs=[image_input_respect, username_input_respect],
-                    outputs=[result_output_respect, leaderboard_output_respect]
-                )
-
-            # Challenge반 탭
-            with gr.Tab("⚡ Challenge반"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📤 이미지 업로드 (Challenge반)")
-                        image_input_challenge = gr.Image(
-                            label="비교할 이미지를 업로드하세요",
-                            type="pil",
-                            height=300
-                        )
-                        username_input_challenge = gr.Textbox(
-                            label="사용자 이름",
-                            placeholder="이름을 입력하세요",
-                            max_lines=1
-                        )
-                        submit_btn_challenge = gr.Button("🚀 점수 계산 및 등록", variant="primary", size="lg")
-
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📊 결과 (Challenge반)")
-                        result_output_challenge = gr.Textbox(
-                            label="계산 결과",
-                            lines=10,
-                            interactive=False
-                        )
+            # ── 반별 탭 (CLASSES 기반 동적 생성) ──
+            for idx, class_name in enumerate(CLASSES):
+                icon = _TAB_ICONS[idx % len(_TAB_ICONS)]
+                key = _class_key(class_name)
+
+                with gr.Tab(f"{icon} {class_name}"):
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            gr.Markdown(f"### 📤 이미지 업로드 ({class_name})")
+                            image_input = gr.Image(
+                                label="비교할 이미지를 업로드하세요",
+                                type="pil",
+                                height=300
+                            )
+                            username_input = gr.Textbox(
+                                label="사용자 이름",
+                                placeholder="이름을 입력하세요",
+                                max_lines=1
+                            )
+                            submit_btn = gr.Button("🚀 점수 계산 및 등록", variant="primary", size="lg")
+
+                        with gr.Column(scale=1):
+                            gr.Markdown(f"### 📊 결과 ({class_name})")
+                            result_output = gr.Textbox(
+                                label="계산 결과",
+                                lines=10,
+                                interactive=False
+                            )
+
+                            gr.Markdown(f"### 🏅 {class_name} 리더보드")
+                            leaderboard_output = gr.Dataframe(
+                                headers=["순위", "사용자명", "점수", "날짜"],
+                                datatype=["number", "str", "number", "str"],
+                                interactive=False
+                            )
+
+                    leaderboard_outputs[key] = leaderboard_output
+
+                    # 클로저에서 key 값을 캡처하기 위해 default argument 사용
+                    submit_btn.click(
+                        fn=lambda img, name, k=key: process_user_image(k, img, name),
+                        inputs=[image_input, username_input],
+                        outputs=[result_output, leaderboard_output]
+                    )
+
+        # ── 모든 키 순서 (practice 먼저, 이후 CLASSES 순서) ──
+        all_keys = ['practice'] + [_class_key(c) for c in CLASSES]
+        all_outputs = [leaderboard_outputs[k] for k in all_keys]
 
-                        gr.Markdown("### 🏅 Challenge반 리더보드")
-                        leaderboard_output_challenge = gr.Dataframe(
-                            headers=["순위", "사용자명", "점수", "날짜"],
-                            datatype=["number", "str", "number", "str"],
-                            interactive=False
-                        )
-
-                # Challenge반 이벤트 핸들러
-                submit_btn_challenge.click(
-                    fn=process_user_image_challenge,
-                    inputs=[image_input_challenge, username_input_challenge],
-                    outputs=[result_output_challenge, leaderboard_output_challenge]
-                )
-
-            # Originality반 탭
-            with gr.Tab("🎨 Originality반"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📤 이미지 업로드 (Originality반)")
-                        image_input_originality = gr.Image(
-                            label="비교할 이미지를 업로드하세요",
-                            type="pil",
-                            height=300
-                        )
-                        username_input_originality = gr.Textbox(
-                            label="사용자 이름",
-                            placeholder="이름을 입력하세요",
-                            max_lines=1
-                        )
-                        submit_btn_originality = gr.Button("🚀 점수 계산 및 등록", variant="primary", size="lg")
-
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📊 결과 (Originality반)")
-                        result_output_originality = gr.Textbox(
-                            label="계산 결과",
-                            lines=10,
-                            interactive=False
-                        )
-
-                        gr.Markdown("### 🏅 Originality반 리더보드")
-                        leaderboard_output_originality = gr.Dataframe(
-                            headers=["순위", "사용자명", "점수", "날짜"],
-                            datatype=["number", "str", "number", "str"],
-                            interactive=False
-                        )
-
-                # Originality반 이벤트 핸들러
-                submit_btn_originality.click(
-                    fn=process_user_image_originality,
-                    inputs=[image_input_originality, username_input_originality],
-                    outputs=[result_output_originality, leaderboard_output_originality]
-                )
-
-            # BCE반 탭
-            with gr.Tab("🔥 BCE반"):
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📤 이미지 업로드 (BCE반)")
-                        image_input_bce = gr.Image(
-                            label="비교할 이미지를 업로드하세요",
-                            type="pil",
-                            height=300
-                        )
-                        username_input_bce = gr.Textbox(
-                            label="사용자 이름",
-                            placeholder="이름을 입력하세요",
-                            max_lines=1
-                        )
-                        submit_btn_bce = gr.Button("🚀 점수 계산 및 등록", variant="primary", size="lg")
-
-                    with gr.Column(scale=1):
-                        gr.Markdown("### 📊 결과 (BCE반)")
-                        result_output_bce = gr.Textbox(
-                            label="계산 결과",
-                            lines=10,
-                            interactive=False
-                        )
-
-                        gr.Markdown("### 🏅 BCE반 리더보드")
-                        leaderboard_output_bce = gr.Dataframe(
-                            headers=["순위", "사용자명", "점수", "날짜"],
-                            datatype=["number", "str", "number", "str"],
-                            interactive=False
-                        )
-
-                # BCE반 이벤트 핸들러
-                submit_btn_bce.click(
-                    fn=process_user_image_bce,
-                    inputs=[image_input_bce, username_input_bce],
-                    outputs=[result_output_bce, leaderboard_output_bce]
-                )
-        
         # 페이지 로드 시 모든 리더보드 표시
         demo.load(
-            fn=lambda: (
-                get_current_leaderboard_practice(), 
-                get_current_leaderboard_respect(), 
-                get_current_leaderboard_challenge(), 
-                # get_current_leaderboard_originality(), 
-                # get_current_leaderboard_bce()
-            ),
-            outputs=[
-                leaderboard_output_practice, 
-                leaderboard_output_respect,
-                leaderboard_output_challenge, 
-                # leaderboard_output_originality, 
-                # leaderboard_output_bce,
-            ]
+            fn=lambda: tuple(get_current_leaderboard(k) for k in all_keys),
+            outputs=all_outputs
         )
 
         # 실시간 업데이트 (10초마다 - 모든 탭 업데이트)
         timer = gr.Timer(value=10)
         timer.tick(
-            fn=lambda: (
-                get_current_leaderboard_practice(), 
-                get_current_leaderboard_respect(), 
-                get_current_leaderboard_challenge(), 
-                # get_current_leaderboard_originality(), 
-                # get_current_leaderboard_bce()
-            ),
-            outputs=[
-                leaderboard_output_practice, 
-                leaderboard_output_respect,
-                leaderboard_output_challenge, 
-                # leaderboard_output_originality, 
-                # leaderboard_output_bce,
-            ]
+            fn=lambda: tuple(check_and_update_leaderboard(k) for k in all_keys),
+            outputs=all_outputs
         )
     
     return demo
@@ -760,11 +596,10 @@ def main():
     print("🏆 이미지 유사도 리더보드 시스템을 시작합니다...")
     print("📁 참조 이미지: label2.jpg")
     print("💾 데이터 파일:")
-    print("   • Respect반: respect.json")
-    print("   • Challenge반: challenge.json")
-    print("   • Originality반: originality.json")
-    print("   • BCE반: bce.json")
-    print("   • 연습: practice.json")
+    for cls in CLASSES:
+        key = _class_key(cls)
+        print(f"   • {cls}: {key}.json")
+    print(f"   • 연습: practice.json")
 
     # 각 반별 메모리 사용량 체크
     for class_name, lb in leaderboards.items():
@@ -784,7 +619,7 @@ def main():
     print("   • 동시성 제어 활성화")
     print("   • 메모리 정리 자동화")
     print("   • 실시간 업데이트: 10초 간격")
-    print("   • 5개 탭별 독립 리더보드 운영 (4개 반 + 연습)")
+    print(f"   • {len(CLASSES)+1}개 탭별 독립 리더보드 운영 ({len(CLASSES)}개 반 + 연습)")
     
     # 외부 공유 우선 시도
     print("\n🌐 외부 접근 시도 중...")