Upload app.py with huggingface_hub

app.py

@@ -1,100 +1,54 @@
 #!/usr/bin/env python3
-"""KAIdol A/B Test Arena - Gradio App"""
-
-import os
-import sys
-
-# 현재 디렉토리를 path에 추가
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+"""KAIdol A/B Test Arena - Simple Version"""
 
 import gradio as gr
 
-from config import get_config
-from ui.arena_tab import create_arena_tab
-from ui.chat_tab import create_chat_tab
-from ui.leaderboard_tab import create_leaderboard_tab
-from ui.history_tab import create_history_tab
-
-
-def create_app():
-    """Gradio 앱 생성"""
-
-    config = get_config()
-    use_mock = config["model"]["use_mock"]
-
-    # 모델 매니저 (Mock 모드가 아닐 때만 로드)
-    model_manager = None
-    if not use_mock:
-        try:
-            from models import get_model_manager
-            model_manager = get_model_manager(
-                max_cached_models=config["model"]["max_cached_models"],
-                use_4bit=config["model"]["use_4bit"],
-            )
-        except Exception as e:
-            print(f"Warning: Could not load model manager: {e}")
-            print("Running in mock mode.")
-            use_mock = True
-
-    # CSS 스타일
-    css = """
-    .response-box { min-height: 200px; }
-    .thinking-box { background-color: #f5f5f5; padding: 10px; border-radius: 5px; }
-    .vote-button { min-width: 100px; }
-    .gr-button-primary { background-color: #6366f1 !important; }
-    """
-
-    # Gradio Blocks
-    with gr.Blocks(
-        title=config["app"]["title"],
-        theme=gr.themes.Soft(),
-        css=css,
-    ) as demo:
-
-        gr.Markdown(f"# {config['app']['title']}")
-        gr.Markdown(config["app"]["description"])
-
-        if use_mock:
-            gr.Markdown("**Mock 모드**: 실제 모델 없이 테스트 응답을 생성합니다.")
-
-        with gr.Tabs():
-            # A/B Arena 탭
-            with gr.Tab("A/B Arena"):
-                create_arena_tab(
-                    model_manager=model_manager,
-                    use_mock=use_mock,
-                )
-
-            # Single Chat 탭
-            with gr.Tab("Single Chat"):
-                create_chat_tab(
-                    model_manager=model_manager,
-                    use_mock=use_mock,
-                )
-
-            # Leaderboard 탭
-            with gr.Tab("Leaderboard"):
-                refresh_leaderboard = create_leaderboard_tab()
-
-            # History 탭
-            with gr.Tab("History"):
-                refresh_history = create_history_tab()
-
-    return demo
-
-
-def main():
-    """메인 함수"""
-    # 환경 변수로 Mock 모드 강제 설정 가능
-    # USE_MOCK=true python app.py
-
-    demo = create_app()
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
+def chat_response(message, character):
+    """Mock 응답 생성"""
+    thinking = f"<think>{character}의 입장에서 생각해보면... 이 메시지에 어떻게 반응해야 할까?</think>"
+    response = f"안녕~ 반가워! 나는 {character}야~"
+    return f"{thinking}\n\n{response}"
+
+# 캐릭터 목록
+CHARACTERS = ["강율", "서이안", "이지후", "차도하", "최민"]
+
+# 모델 목록
+MODELS = [
+    "hyperclovax-32b-dpo-v5",
+    "qwen2.5-14b-dpo-v5",
+    "qwen2.5-7b-dpo-v5",
+    "exaone-7.8b-dpo-v5",
+]
+
+with gr.Blocks(title="KAIdol A/B Test Arena") as demo:
+    gr.Markdown("# KAIdol A/B Test Arena")
+    gr.Markdown("K-pop 아이돌 롤플레이 모델 A/B 비교 평가")
+    gr.Markdown("**Mock 모드**: 실제 모델 없이 테스트 응답을 생성합니다.")
+
+    with gr.Row():
+        character = gr.Dropdown(choices=CHARACTERS, value="강율", label="캐릭터")
+        model_a = gr.Dropdown(choices=MODELS, value=MODELS[0], label="Model A")
+        model_b = gr.Dropdown(choices=MODELS, value=MODELS[1], label="Model B")
+
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### Model A")
+            response_a = gr.Textbox(label="응답", lines=5)
+        with gr.Column():
+            gr.Markdown("### Model B")
+            response_b = gr.Textbox(label="응답", lines=5)
+
+    user_input = gr.Textbox(label="메시지", placeholder="아이돌에게 메시지를 보내세요...")
+    submit_btn = gr.Button("전송", variant="primary")
+
+    def generate(msg, char, ma, mb):
+        return chat_response(msg, char), chat_response(msg, char)
+
+    submit_btn.click(
+        fn=generate,
+        inputs=[user_input, character, model_a, model_b],
+        outputs=[response_a, response_b]
     )
 
-
 if __name__ == "__main__":
-    main()
+    demo.launch(server_name="0.0.0.0", server_port=7860)