gradio_app.py

import gradio as gr
from enhanced_rag_system import EnhancedImmunoRAGSystem
import json

# RAG 시스템 초기화
rag_system = None

def initialize_rag_system():
    global rag_system
    try:
        rag_system = EnhancedImmunoRAGSystem("merged_qa_dataset.json")
        return "✅ 시스템 준비 완료"
    except Exception as e:
        return f"❌ 로딩 실패: {str(e)}"

def answer_question(question, history):
    if rag_system is None:
        return history + [[question, "⚠️ 시스템을 먼저 초기화해주세요."]]
    
    if not question.strip():
        return history + [[question, "💭 질문을 입력해주세요."]]
    
    try:
        result = rag_system.answer_question(question)
        
        # 깔끔한 응답 포맷
        confidence = result['confidence']
        confidence_bar = "█" * int(confidence * 10) + "░" * (10 - int(confidence * 10))
        confidence_color = "🟢" if confidence > 0.7 else "🟡" if confidence > 0.4 else "🔴"
        
        response = f"{result['answer']}\n\n"
        response += f"{confidence_color} **신뢰도** `{confidence:.0%}` {confidence_bar}\n\n"
        
        if result['matched_questions'] and confidence > 0.3:
            top_match = result['matched_questions'][0]
            response += f"💡 **관련 질문:** {top_match['question']}"
        
        return history + [[question, response]]
        
    except Exception as e:
        return history + [[question, f"❌ **오류**\n```\n{str(e)}\n```"]]

def clear_chat():
    return []

# 세련된 CSS 스타일
custom_css = """
@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700&display=swap');

:root {
    --primary: #6366f1;
    --primary-light: #818cf8;
    --primary-dark: #4f46e5;
    --secondary: #64748b;
    --accent: #06b6d4;
    --success: #10b981;
    --warning: #f59e0b;
    --error: #ef4444;
    --background: #fafbfc;
    --surface: #ffffff;
    --surface-2: #f8fafc;
    --border: #e1e5e9;
    --border-light: #f1f3f4;
    --text: #1e293b;
    --text-light: #64748b;
    --text-muted: #94a3b8;
    --shadow-sm: 0 1px 2px 0 rgb(0 0 0 / 0.05);
    --shadow: 0 1px 3px 0 rgb(0 0 0 / 0.1), 0 1px 2px -1px rgb(0 0 0 / 0.1);
    --shadow-md: 0 4px 6px -1px rgb(0 0 0 / 0.1), 0 2px 4px -2px rgb(0 0 0 / 0.1);
    --shadow-lg: 0 10px 15px -3px rgb(0 0 0 / 0.1), 0 4px 6px -4px rgb(0 0 0 / 0.1);
    --radius: 12px;
    --radius-lg: 16px;
    --radius-xl: 20px;
}

/* 전역 리셋 및 기본 스타일 */
* {
    font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif !important;
    box-sizing: border-box;
}

/* 전역 텍스트 색상 */
*:not(.app-header):not(.app-header *):not(.user-message):not(.user-message *) {
    color: var(--text) !important;
}

/* 메인 컨테이너 */
.gradio-container {
    max-width: 1000px !important;
    margin: 0 auto !important;
    padding: 24px !important;
    background: var(--background) !important;
    min-height: 100vh;
}

/* 헤더 - 미니멀 디자인 */
.app-header {
    text-align: center;
    padding: 48px 32px;
    background: linear-gradient(135deg, var(--primary) 0%, var(--primary-light) 100%);
    border-radius: var(--radius-xl);
    margin-bottom: 32px;
    box-shadow: var(--shadow-lg);
    position: relative;
    overflow: hidden;
}

.app-header::before {
    content: '';
    position: absolute;
    top: 0;
    left: 0;
    right: 0;
    bottom: 0;
    background: linear-gradient(135deg, rgba(255,255,255,0.1) 0%, rgba(255,255,255,0.05) 100%);
    pointer-events: none;
}

.app-header * {
    color: white !important;
    position: relative;
    z-index: 1;
}

.app-header h1 {
    font-size: 2.75rem;
    font-weight: 700;
    margin: 0 0 12px 0;
    letter-spacing: -0.02em;
    line-height: 1.1;
}

.app-header p {
    font-size: 1.125rem;
    opacity: 0.9;
    margin: 0;
    font-weight: 400;
    line-height: 1.5;
}

/* 채팅 컨테이너 - 깔끔한 카드 디자인 */
.chat-container {
    background: var(--surface);
    border-radius: var(--radius-xl);
    padding: 0;
    margin-bottom: 24px;
    border: 1px solid var(--border-light);
    box-shadow: var(--shadow-sm);
    overflow: hidden;
}

/* 채팅박스 */
.chatbot {
    border: none !important;
    border-radius: 0 !important;
    background: transparent !important;
    box-shadow: none !important;
    min-height: 500px !important;
    padding: 24px !important;
}

.chatbot .message {
    border-radius: var(--radius) !important;
    padding: 20px !important;
    margin: 16px 0 !important;
    box-shadow: none !important;
    border: none !important;
    max-width: 85% !important;
}

/* 사용자 메시지 */
.chatbot .message.user,
.user-message {
    background: var(--primary) !important;
    margin-left: auto !important;
    margin-right: 0 !important;
    border-radius: var(--radius) var(--radius) 4px var(--radius) !important;
}

.chatbot .message.user *,
.user-message * {
    color: white !important;
}

/* 봇 메시지 */
.chatbot .message.bot {
    background: var(--surface-2) !important;
    margin-left: 0 !important;
    margin-right: auto !important;
    border: 1px solid var(--border-light) !important;
    border-radius: var(--radius) var(--radius) var(--radius) 4px !important;
}

.chatbot .message.bot * {
    color: var(--text) !important;
}

/* 입력 영역 - 플로팅 디자인 */
.input-container {
    background: var(--surface);
    border-radius: var(--radius-lg);
    padding: 20px;
    border: 1px solid var(--border-light);
    box-shadow: var(--shadow-sm);
    margin-top: 24px;
}

.input-row {
    background: transparent;
    border-radius: 0;
    padding: 0;
    border: none;
    box-shadow: none;
    margin: 0;
}

.input-row .textbox,
.input-row input,
.input-row textarea {
    border: 1px solid var(--border) !important;
    background: var(--surface) !important;
    border-radius: var(--radius) !important;
    padding: 16px 20px !important;
    font-size: 16px !important;
    line-height: 1.5 !important;
    transition: all 0.2s ease !important;
    color: var(--text) !important;
    box-shadow: var(--shadow-sm) !important;
}

.input-row .textbox::placeholder,
.input-row input::placeholder,
.input-row textarea::placeholder {
    color: var(--text-muted) !important;
    opacity: 1 !important;
}

.input-row .textbox:focus,
.input-row input:focus,
.input-row textarea:focus {
    border-color: var(--primary) !important;
    box-shadow: 0 0 0 3px rgba(99, 102, 241, 0.1) !important;
    outline: none !important;
}

/* 버튼 - 모던 디자인 */
.btn-primary,
button[variant="primary"] {
    background: var(--primary) !important;
    border: none !important;
    border-radius: var(--radius) !important;
    padding: 16px 24px !important;
    font-weight: 600 !important;
    font-size: 15px !important;
    line-height: 1 !important;
    transition: all 0.2s ease !important;
    box-shadow: var(--shadow-sm) !important;
    color: white !important;
    cursor: pointer !important;
}

.btn-primary:hover,
button[variant="primary"]:hover {
    background: var(--primary-dark) !important;
    transform: translateY(-1px) !important;
    box-shadow: var(--shadow-md) !important;
}

.btn-secondary {
    background: var(--surface) !important;
    border: 1px solid var(--border) !important;
    border-radius: var(--radius) !important;
    padding: 16px 24px !important;
    font-weight: 500 !important;
    font-size: 15px !important;
    color: var(--text-light) !important;
    transition: all 0.2s ease !important;
    cursor: pointer !important;
}

.btn-secondary:hover {
    border-color: var(--primary) !important;
    color: var(--primary) !important;
    transform: translateY(-1px) !important;
    box-shadow: var(--shadow-sm) !important;
}

/* 예시 질문 - 카드 그리드 */
.examples-section {
    margin: 32px 0;
}

.examples-title {
    font-size: 1.125rem;
    font-weight: 600;
    color: var(--text);
    margin-bottom: 16px;
    text-align: center;
}

.example-grid {
    display: grid;
    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
    gap: 16px;
    margin-top: 20px;
}

.example-btn {
    background: var(--surface) !important;
    border: 1px solid var(--border-light) !important;
    border-radius: var(--radius) !important;
    padding: 20px !important;
    text-align: left !important;
    font-size: 14px !important;
    font-weight: 500 !important;
    color: var(--text) !important;
    transition: all 0.2s ease !important;
    box-shadow: var(--shadow-sm) !important;
    cursor: pointer !important;
    line-height: 1.4 !important;
}

.example-btn:hover {
    border-color: var(--primary) !important;
    background: var(--surface) !important;
    transform: translateY(-2px) !important;
    box-shadow: var(--shadow-md) !important;
    color: var(--primary) !important;
}

/* 탭 - 미니멀 디자인 */
.tab-nav {
    background: var(--surface) !important;
    border-radius: var(--radius-lg) !important;
    border: 1px solid var(--border-light) !important;
    overflow: hidden !important;
    box-shadow: var(--shadow-sm) !important;
    margin-bottom: 24px !important;
}

.tab-nav .tab-item {
    padding: 20px 32px !important;
    font-weight: 500 !important;
    font-size: 15px !important;
    transition: all 0.2s ease !important;
    color: var(--text-light) !important;
    border: none !important;
}

.tab-nav .tab-item.selected {
    background: var(--primary) !important;
    color: white !important;
}

.tab-nav .tab-item:hover:not(.selected) {
    background: var(--surface-2) !important;
    color: var(--text) !important;
}

/* 시스템 정보 섹션 */
.system-info {
    background: var(--surface);
    border-radius: var(--radius-lg);
    padding: 32px;
    border: 1px solid var(--border-light);
    box-shadow: var(--shadow-sm);
}

/* 마크다운 스타일링 */
.markdown,
.gr-markdown {
    color: var(--text) !important;
    line-height: 1.6;
}

.markdown h1, .markdown h2, .markdown h3,
.gr-markdown h1, .gr-markdown h2, .gr-markdown h3 {
    color: var(--text) !important;
    font-weight: 600;
    margin-top: 24px;
    margin-bottom: 12px;
}

.markdown h3, .gr-markdown h3 {
    font-size: 1.125rem;
    color: var(--text) !important;
}

.markdown p, .gr-markdown p {
    color: var(--text) !important;
    margin-bottom: 16px;
}

.markdown table, .gr-markdown table {
    border-collapse: collapse;
    width: 100%;
    margin: 20px 0;
    border-radius: var(--radius);
    overflow: hidden;
    box-shadow: var(--shadow-sm);
}

.markdown th, .gr-markdown th {
    background: var(--surface-2) !important;
    color: var(--text) !important;
    font-weight: 600;
    padding: 16px;
    text-align: left;
    border-bottom: 1px solid var(--border);
}

.markdown td, .gr-markdown td {
    color: var(--text) !important;
    padding: 16px;
    border-bottom: 1px solid var(--border-light);
}

.markdown strong, .gr-markdown strong {
    color: var(--text) !important;
    font-weight: 600;
}

.markdown code, .gr-markdown code {
    background: var(--surface-2);
    color: var(--text) !important;
    padding: 2px 6px;
    border-radius: 4px;
    font-size: 0.875em;
}

/* 상태 표시 */
.status-output {
    background: var(--surface-2);
    border: 1px solid var(--border-light);
    border-radius: var(--radius);
    padding: 16px;
    font-family: 'SF Mono', Monaco, 'Cascadia Code', monospace;
    font-size: 14px;
    color: var(--text) !important;
}

/* 모바일 최적화 */
@media (max-width: 768px) {
    .gradio-container {
        padding: 16px !important;
    }
    
    .app-header {
        padding: 32px 24px !important;
        margin-bottom: 24px !important;
    }
    
    .app-header h1 {
        font-size: 2.25rem !important;
    }
    
    .chatbot {
        padding: 16px !important;
        min-height: 400px !important;
    }
    
    .chatbot .message {
        max-width: 90% !important;
        padding: 16px !important;
    }
    
    .input-container {
        padding: 16px !important;
    }
    
    .example-grid {
        grid-template-columns: 1fr !important;
        gap: 12px !important;
    }
    
    .example-btn {
        padding: 16px !important;
    }
}

/* 다크모드 지원 */
@media (prefers-color-scheme: dark) {
    :root {
        --background: #0f172a;
        --surface: #1e293b;
        --surface-2: #334155;
        --border: #475569;
        --border-light: #374151;
        --text: #f1f5f9;
        --text-light: #cbd5e0;
        --text-muted: #94a3b8;
    }
}

/* 부드러운 애니메이션 */
@keyframes slideUp {
    from {
        opacity: 0;
        transform: translateY(20px);
    }
    to {
        opacity: 1;
        transform: translateY(0);
    }
}

.fade-in {
    animation: slideUp 0.4s ease-out;
}

/* 스크롤바 스타일링 */
::-webkit-scrollbar {
    width: 8px;
}

::-webkit-scrollbar-track {
    background: var(--surface-2);
    border-radius: 4px;
}

::-webkit-scrollbar-thumb {
    background: var(--border);
    border-radius: 4px;
}

::-webkit-scrollbar-thumb:hover {
    background: var(--text-muted);
}
"""

# 예시 질문들 (더 간결하게)
example_questions = [
    "HBs Ag 참고치는?",
    "CEA 정상범위는?", 
    "PIVKA 검사 원리는?",
    "CA19-9 언제 나와요?",
    "AFP 기준치는?",
    "PSA 정상값은?"
]

# Gradio 인터페이스
with gr.Blocks(
    title="면역검사 AI 챗봇",
    theme=gr.themes.Default(
        primary_hue="blue",
        secondary_hue="slate",
        neutral_hue="slate",
        font=gr.themes.GoogleFont("Inter")
    ),
    css=custom_css
) as demo:
    
    # 헤더
    gr.HTML("""
    <div class="app-header fade-in">
        <h1>🧬 면역검사 AI</h1>
        <p>정확하고 빠르은 면역검사 정보를 제공합니다</p>
    </div>
    """)
    
    with gr.Tabs(elem_classes="tab-nav") as tabs:
        
        # 채팅 탭
        with gr.Tab("💬 채팅"):
            
            with gr.Group(elem_classes="chat-container"):
                gr.Markdown("💬 아래 채팅창에 질문을 입력해 주세요. 처음 로딩 시 모델 준비에 시간이 걸릴 수 있습니다.")
                chatbot = gr.Chatbot(
                    height=500,
                )
                
                with gr.Row():
                    msg = gr.Textbox(
                        placeholder="예: HBs Ag 참고치는 얼마인가요?",
                        container=False,
                        scale=4,
                        show_label=False,
                        elem_classes="textbox"
                    )
                    with gr.Column(scale=1, min_width=120):
                        submit_btn = gr.Button(
                            "전송", 
                            variant="primary",
                            elem_classes="btn-primary"
                        )
                        clear_btn = gr.Button(
                            "초기화",
                            elem_classes="btn-secondary"
                        )
            
            # 예시 질문들
            gr.Markdown("### 💡 빠른 질문")
            gr.HTML('<div class="example-grid">')
            
            with gr.Row():
                for i, question in enumerate(example_questions):
                    if i % 3 == 0 and i > 0:
                        gr.HTML('</div><div class="example-grid">')
                    
                    example_btn = gr.Button(
                        f"💭 {question}",
                        elem_classes="example-btn"
                    )
                    example_btn.click(
                        lambda x=question: x,
                        outputs=msg
                    )
            
            gr.HTML('</div>')
        
        # 시스템 정보 탭
        with gr.Tab("⚙️ 시스템"):
            
            with gr.Row():
                init_btn = gr.Button(
                    "🔄 시스템 초기화", 
                    variant="primary",
                    elem_classes="btn-primary"
                )
            
            init_output = gr.Textbox(
                label="상태", 
                interactive=False,
                lines=1
            )
            
            # 시스템 정보
            gr.Markdown("""
            ### 📋 지원하는 검사 항목
            
            | 분류 | 검사 항목 |
            |------|----------|
            | **간염 검사** | HBs Ag, HBs Ab, HBc Ab, HCV Ab |
            | **종양 표지자** | CEA, AFP, CA19-9, CA125, PSA, PIVKA-II |
            | **호르몬 검사** | TSH, T3, T4, 인슐린 |
            | **감염 검사** | HIV, 매독, 결핵 |
            
            ### 🎯 사용법
            1. 자연어로 질문하세요 (예: "HBs Ag 정상치는?")
            2. 동의어도 인식합니다 (예: "비형간염항원", "B형간염표면항원")
            3. 신뢰도가 높을수록 정확한 답변입니다
            """)
    
    # 이벤트 핸들러
    msg.submit(
        answer_question, 
        [msg, chatbot], 
        [chatbot]
    ).then(
        lambda: "", 
        outputs=msg
    )
    
    submit_btn.click(
        answer_question, 
        [msg, chatbot], 
        [chatbot]
    ).then(
        lambda: "", 
        outputs=msg
    )
    
    clear_btn.click(clear_chat, outputs=chatbot)
    init_btn.click(initialize_rag_system, outputs=init_output)
    
    # 자동 초기화
    demo.load(initialize_rag_system, outputs=init_output)

if __name__ == "__main__":
    import os
    # Hugging Face Spaces/일반 환경 모두 호환
    port = int(os.environ.get("PORT", "7860"))
    is_spaces = bool(os.environ.get("SPACE_ID"))
    
    if is_spaces:
        # Hugging Face Spaces: 간단한 실행
        demo.launch()
    else:
        # 로컬/기타 PaaS: 필요 시 share 허용
        demo.launch(
            server_name="0.0.0.0",
            server_port=port,
            share=True,
            show_error=True
        )
    
    # Hugging Face Spaces에서 여전히 두 가지 오류가 발생하고 있습니다:
    # 1. **JSON 스키마 오류**: Gradio 내부 오류로 보입니다
    # 2. **localhost 접근 오류**: 여전히 네트워크 설정 문제입니다
    # 
    # 최종 해결책: gradio_app.py 수정
    # 
    # Hugging Face Spaces에서는 가장 기본적인 설정만 사용해야 합니다:
    import os
    
    # Hugging Face Spaces 감지
    is_spaces = "SPACE_ID" in os.environ
    
    try:
        if is_spaces:
            # Hugging Face Spaces: 기본 설정만 사용
            demo.launch(enable_queue=True)
        else:
            # 로컬 환경
            port = int(os.environ.get("PORT", "7860"))
            demo.launch(
                server_name="0.0.0.0",
                server_port=port,
                share=True
            )
    except Exception as e:
        print(f"Launch error: {e}")
        # 최후의 수단: 가장 기본적인 실행
        demo.launch()