Upload 2 files

gradio_app.py

@@ -0,0 +1,689 @@
+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: share 금지, queue 권장
+        demo.queue(concurrency_count=3).launch(
+            server_name="0.0.0.0",
+            server_port=port,
+            share=False
+        )
+    else:
+        # 로컬/기타 PaaS: 필요 시 share 허용
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=port,
+            share=True,
+            show_error=True
+        )

requirements.txt

@@ -0,0 +1,7 @@
+gradio==4.44.1
+streamlit
+sentence-transformers
+scikit-learn
+numpy
+torch
+transformers