Spaces:
Running on Zero
Running on Zero
| import os | |
| import sys | |
| import subprocess | |
| import html | |
| # Tự động cài đặt transformers & accelerate nếu chưa có trên môi trường | |
| try: | |
| import transformers | |
| except ImportError: | |
| print("Installing missing dependencies: transformers, accelerate...") | |
| subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "accelerate"]) | |
| import transformers | |
| import torch | |
| import gradio as gr | |
| from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline | |
| # Hỗ trợ Hugging Face ZeroGPU (nếu Space đang cấu hình phần cứng GPU) | |
| try: | |
| import spaces | |
| except ImportError: | |
| class _MockSpaces: | |
| def GPU(fn=None, *args, **kwargs): | |
| if fn is not None: | |
| return fn | |
| return lambda f: f | |
| spaces = _MockSpaces() | |
| # 1. Quản lý danh sách mô hình & Caching Pipeline | |
| DEFAULT_MODEL = "NhanSoHocCode/xlmr-ner-4labels-v2" | |
| MODEL_OPTIONS = [ | |
| ("✨ XLM-RoBERTa NER 4Labels v2 (Mới nhất / Khuyên dùng)", "NhanSoHocCode/xlmr-ner-4labels-v2"), | |
| ("🏷️ XLM-RoBERTa NER 4Labels v1 (Bản gốc)", "NhanSoHocCode/xlmr-ner-4labels"), | |
| ("⚙️ Tùy chỉnh (Nhập checkpoint Hugging Face khác...)", "custom") | |
| ] | |
| PIPELINE_CACHE = {} | |
| def get_ner_pipeline(model_id: str): | |
| """ | |
| Tải hoặc lấy từ cache pipeline trích xuất thực thể theo model_id. | |
| Hỗ trợ cả checkpoint có sẵn hoặc checkpoint tùy chỉnh có cùng kiến trúc. | |
| """ | |
| model_id = model_id.strip() if model_id else DEFAULT_MODEL | |
| if model_id in PIPELINE_CACHE: | |
| return PIPELINE_CACHE[model_id], None | |
| try: | |
| print(f"Loading model checkpoint: {model_id} ...") | |
| device = 0 if torch.cuda.is_available() else -1 | |
| tokenizer = AutoTokenizer.from_pretrained(model_id) | |
| model = AutoModelForTokenClassification.from_pretrained(model_id) | |
| pipe = pipeline( | |
| "token-classification", | |
| model=model, | |
| tokenizer=tokenizer, | |
| aggregation_strategy="simple", | |
| device=device | |
| ) | |
| PIPELINE_CACHE[model_id] = pipe | |
| print(f"Model loaded successfully: {model_id}") | |
| return pipe, None | |
| except Exception as e: | |
| err_msg = f"Không thể tải mô hình '{model_id}'. Chi tiết lỗi: {str(e)}" | |
| print(err_msg) | |
| return None, err_msg | |
| # Tải trước mô hình mặc định khi khởi động ứng dụng | |
| print(f"Pre-loading default model: {DEFAULT_MODEL} ...") | |
| get_ner_pipeline(DEFAULT_MODEL) | |
| # 2. Cấu hình nhãn & biểu tượng & màu sắc | |
| LABEL_CONFIG = { | |
| "HARD_SKILL": { | |
| "icon": "🛠️", | |
| "name": "HARD_SKILL", | |
| "color": "#ef4444", # Đỏ san hô | |
| "bg": "#ef4444", | |
| "text_color": "#ffffff" | |
| }, | |
| "SOFT_SKILL": { | |
| "icon": "🤝", | |
| "name": "SOFT_SKILL", | |
| "color": "#d97706", # Hổ phách / Vàng cam | |
| "bg": "#d97706", | |
| "text_color": "#ffffff" | |
| }, | |
| "EDUCATION": { | |
| "icon": "🎓", | |
| "name": "EDUCATION", | |
| "color": "#0284c7", # Xanh lam / Cyan | |
| "bg": "#0284c7", | |
| "text_color": "#ffffff" | |
| }, | |
| "MAJOR": { | |
| "icon": "📊", | |
| "name": "MAJOR", | |
| "color": "#16a34a", # Xanh lá lục bảo | |
| "bg": "#16a34a", | |
| "text_color": "#ffffff" | |
| } | |
| } | |
| ORDERED_LABELS = ["HARD_SKILL", "SOFT_SKILL", "EDUCATION", "MAJOR"] | |
| def analyze_ner(title, text, threshold=0.0, model_choice="NhanSoHocCode/xlmr-ner-4labels-v2", custom_model_id=""): | |
| # Xác định checkpoint thực tế được chọn | |
| if model_choice == "custom": | |
| active_model = (custom_model_id or "").strip() | |
| if not active_model: | |
| err_html = """ | |
| <div style="padding: 18px; text-align: center; color: #dc2626; background: #fef2f2; border: 1px solid #fecaca; border-radius: 8px;"> | |
| ⚠️ <b>Chưa nhập thông tin mô hình:</b> Vui lòng điền Hugging Face Repo ID hoặc checkpoint tùy chỉnh để tiếp tục. | |
| </div> | |
| """ | |
| return err_html, "Lỗi: Bạn đã chọn mô hình tùy chỉnh nhưng chưa nhập Hugging Face Repo ID.", [] | |
| else: | |
| active_model = model_choice.strip() if model_choice else DEFAULT_MODEL | |
| if not text or not text.strip(): | |
| empty_msg = """ | |
| <div style="padding: 24px; text-align: center; color: #6b7280; font-family: monospace;"> | |
| Vui lòng nhập văn bản mô tả công việc (Job Description) để bắt đầu phân tích. | |
| </div> | |
| """ | |
| return empty_msg, "Vui lòng nhập văn bản JD.", [] | |
| title = title.strip() if title and title.strip() else "Kết quả phân tích JD" | |
| raw_text = text.strip() | |
| # Lấy pipeline mô hình | |
| ner_pipe, load_err = get_ner_pipeline(active_model) | |
| if load_err or ner_pipe is None: | |
| err_html = f""" | |
| <div style="padding: 18px; color: #dc2626; background: #fef2f2; border: 1px solid #fecaca; border-radius: 8px; font-family: sans-serif;"> | |
| <b>❌ Lỗi tải mô hình ({html.escape(active_model)}):</b><br/> | |
| <span style="font-size: 13.5px;">{html.escape(load_err or 'Không thể khởi tạo mô hình.')}</span><br/><br/> | |
| <small style="color: #6b7280;">Vui lòng kiểm tra lại tên repo trên Hugging Face hoặc đảm bảo mô hình có cùng kiến trúc Token Classification.</small> | |
| </div> | |
| """ | |
| return err_html, f"Lỗi tải mô hình '{active_model}': {load_err}", [] | |
| # Thực hiện trích xuất | |
| try: | |
| raw_results = ner_pipe(raw_text) | |
| except Exception as e: | |
| err_msg = f"Lỗi trong quá trình trích xuất thực thể: {str(e)}" | |
| err_html = f""" | |
| <div style="padding: 18px; color: #dc2626; background: #fef2f2; border: 1px solid #fecaca; border-radius: 8px;"> | |
| <b>❌ Lỗi thực thi mô hình:</b> {html.escape(err_msg)} | |
| </div> | |
| """ | |
| return err_html, err_msg, [] | |
| # Lọc theo ngưỡng tin cậy | |
| entities = [ | |
| item for item in raw_results | |
| if float(item["score"]) >= threshold | |
| ] | |
| # Sắp xếp các thực thể theo thứ tự xuất hiện trong câu | |
| entities = sorted(entities, key=lambda x: x["start"]) | |
| # Xử lý gối vị trí (nếu có chồng lấn start/end) | |
| filtered_entities = [] | |
| last_end = -1 | |
| for ent in entities: | |
| if ent["start"] >= last_end: | |
| filtered_entities.append(ent) | |
| last_end = ent["end"] | |
| # 1. Tạo câu có đánh dấu thực thể dạng HTML màu sắc & dạng văn bản thô | |
| rendered_parts = [] | |
| raw_parts = [] | |
| last_idx = 0 | |
| for ent in filtered_entities: | |
| start = ent["start"] | |
| end = ent["end"] | |
| label = ent["entity_group"] | |
| cfg = LABEL_CONFIG.get(label, { | |
| "icon": "🏷️", | |
| "color": "#6366f1", | |
| "bg": "#6366f1", | |
| "text_color": "#ffffff" | |
| }) | |
| # Đoạn chữ thường phía trước | |
| if start > last_idx: | |
| plain_segment = raw_text[last_idx:start] | |
| rendered_parts.append(html.escape(plain_segment)) | |
| raw_parts.append(plain_segment) | |
| # Lấy từ thực tế trong chuỗi gốc | |
| ent_word = raw_text[start:end] | |
| # Thẻ màu HTML | |
| badge_html = ( | |
| f'<span class="ner-badge" style="background-color: {cfg["bg"]}; color: {cfg["text_color"]};">' | |
| f'[{label}: {html.escape(ent_word)}]' | |
| f'</span>' | |
| ) | |
| rendered_parts.append(badge_html) | |
| raw_parts.append(f"[{label}: {ent_word}]") | |
| last_idx = end | |
| # Đoạn văn bản còn lại sau thực thể cuối cùng | |
| if last_idx < len(raw_text): | |
| plain_segment = raw_text[last_idx:] | |
| rendered_parts.append(html.escape(plain_segment)) | |
| raw_parts.append(plain_segment) | |
| tagged_html = "".join(rendered_parts) | |
| tagged_plain = "".join(raw_parts) | |
| # 2. Thống kê danh sách thực thể theo nhóm | |
| grouped = {lbl: [] for lbl in ORDERED_LABELS} | |
| for ent in filtered_entities: | |
| lbl = ent["entity_group"] | |
| if lbl in grouped: | |
| grouped[lbl].append(ent) | |
| else: | |
| grouped.setdefault(lbl, []).append(ent) | |
| stats_html_lines = [] | |
| stats_plain_lines = [] | |
| table_rows = [] | |
| stt = 1 | |
| for lbl in ORDERED_LABELS: | |
| items = grouped.get(lbl, []) | |
| if not items: | |
| continue | |
| cfg = LABEL_CONFIG.get(lbl, {"icon": "🏷️", "color": "#4b5563"}) | |
| items_detail = [] | |
| for it in items: | |
| word = it["word"] | |
| score_pct = float(it["score"]) * 100 | |
| items_detail.append(f"{word} ({score_pct:.1f}%)") | |
| table_rows.append([stt, f"{cfg['icon']} {lbl}", word, f"{score_pct:.1f}%", f"{it['start']}:{it['end']}"]) | |
| stt += 1 | |
| items_str = ", ".join(items_detail) | |
| # Dòng thống kê HTML | |
| stat_html = ( | |
| f'<div class="stat-row">' | |
| f' <span class="stat-tag" style="color: {cfg["color"]}; font-weight: 700;">' | |
| f' {cfg["icon"]} {lbl} ({len(items)}):' | |
| f' </span> ' | |
| f' <span class="stat-content">{html.escape(items_str)}</span>' | |
| f'</div>' | |
| ) | |
| stats_html_lines.append(stat_html) | |
| # Dòng thống kê thô | |
| stats_plain_lines.append(f" {cfg['icon']} {lbl} ({len(items)}): {items_str}") | |
| stats_html_content = "\n".join(stats_html_lines) | |
| stats_plain_content = "\n".join(stats_plain_lines) | |
| # 3. Kết xuất giao diện thẻ theo đúng hình mẫu | |
| card_html = f""" | |
| <div class="jd-result-card"> | |
| <div class="border-line-double">===============================================================================================</div> | |
| <div class="jd-title">📌 {html.escape(title)}</div> | |
| <div class="model-badge-info">🤖 Mô hình: <code>{html.escape(active_model)}</code></div> | |
| <div class="border-line-single">-----------------------------------------------------------------------------------------------</div> | |
| <div class="section-heading">Văn bản gốc:</div> | |
| <div class="section-body raw-body">{html.escape(raw_text)}</div> | |
| <div class="border-line-single">-----------------------------------------------------------------------------------------------</div> | |
| <div class="section-heading">Kết quả trích xuất trực tiếp trong câu:</div> | |
| <div class="section-body tagged-body">{tagged_html}</div> | |
| <div class="border-line-single">-----------------------------------------------------------------------------------------------</div> | |
| <div class="section-heading">Thống kê danh sách thực thể:</div> | |
| <div class="section-body stats-body"> | |
| {stats_html_content if stats_html_content else '<div style="color: #888;">Không phát hiện thực thể nào vượt qua ngưỡng tin cậy.</div>'} | |
| </div> | |
| <div class="border-line-double">===============================================================================================</div> | |
| </div> | |
| """ | |
| # 4. Văn bản Plain Text hoàn chỉnh (sao chép nhanh) | |
| plain_output = f"""=============================================================================================== | |
| 📌 {title} | |
| 🤖 Mô hình: {active_model} | |
| ----------------------------------------------------------------------------------------------- | |
| Văn bản gốc: | |
| {raw_text} | |
| ----------------------------------------------------------------------------------------------- | |
| Kết quả trích xuất trực tiếp trong câu: | |
| {tagged_plain} | |
| ----------------------------------------------------------------------------------------------- | |
| Thống kê danh sách thực thể: | |
| {stats_plain_content if stats_plain_content else ' (Không phát hiện thực thể nào)'} | |
| ===============================================================================================""" | |
| return card_html, plain_output, table_rows | |
| # Helper tạo link Hugging Face | |
| def render_model_link(model_choice, custom_id): | |
| is_custom = (model_choice == "custom") | |
| target = (custom_id or "").strip() if is_custom else model_choice | |
| if target and not target.startswith("http"): | |
| hf_url = f"https://huggingface.co/{target}" | |
| return ( | |
| f'<div class="hf-link-wrapper">' | |
| f' <span class="hf-icon">🤗</span>' | |
| f' <span>Checkpoint hiện tại: </span>' | |
| f' <a href="{hf_url}" target="_blank" class="hf-link"><b>{html.escape(target)}</b> ↗</a>' | |
| f'</div>' | |
| ) | |
| return ( | |
| '<div class="hf-link-wrapper">' | |
| ' <span class="hf-icon">💡</span>' | |
| ' <span>Nhập Hugging Face Repo ID hoặc checkpoint cục bộ tương thích với 4 nhãn chuẩn.</span>' | |
| '</div>' | |
| ) | |
| def handle_model_change(choice, custom_id): | |
| show_custom = (choice == "custom") | |
| link_html = render_model_link(choice, custom_id) | |
| return gr.update(visible=show_custom), link_html | |
| def handle_custom_id_change(custom_id): | |
| return render_model_link("custom", custom_id) | |
| # Custom CSS cho giao diện | |
| custom_css = """ | |
| /* Font và định dạng tổng quát */ | |
| .container { | |
| max-width: 1050px !important; | |
| margin: 0 auto !important; | |
| font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif; | |
| } | |
| /* Khung chọn mô hình thay thế dòng chữ mô tả */ | |
| .model-selector-panel { | |
| background: linear-gradient(135deg, #f8fafc 0%, #f1f5f9 100%); | |
| border: 1px solid #e2e8f0; | |
| border-radius: 12px; | |
| padding: 14px 18px; | |
| margin: 10px 0 16px 0; | |
| box-shadow: 0 1px 3px rgba(0, 0, 0, 0.04); | |
| } | |
| .hf-link-wrapper { | |
| display: flex; | |
| align-items: center; | |
| gap: 6px; | |
| font-size: 13px; | |
| color: #475569; | |
| margin-top: 6px; | |
| padding-left: 2px; | |
| } | |
| .hf-link { | |
| color: #2563eb !important; | |
| text-decoration: none !important; | |
| transition: color 0.15s ease; | |
| } | |
| .hf-link:hover { | |
| color: #1d4ed8 !important; | |
| text-decoration: underline !important; | |
| } | |
| .model-badge-info { | |
| font-size: 13px; | |
| color: #4b5563; | |
| margin: 4px 0 6px 0; | |
| } | |
| .model-badge-info code { | |
| background-color: #f3f4f6; | |
| color: #1f2937; | |
| padding: 2px 6px; | |
| border-radius: 4px; | |
| font-size: 12.5px; | |
| } | |
| /* Thẻ kết quả mô phỏng chính xác hình mẫu */ | |
| .jd-result-card { | |
| background-color: #ffffff; | |
| border: 1px solid #e5e7eb; | |
| border-radius: 8px; | |
| padding: 20px 24px; | |
| margin-top: 10px; | |
| box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.05), 0 2px 4px -1px rgba(0, 0, 0, 0.03); | |
| font-family: "Consolas", "Courier New", monospace, sans-serif; | |
| color: #1f2937; | |
| line-height: 1.8; | |
| } | |
| /* Dark mode compatibility */ | |
| @media (prefers-color-scheme: dark) { | |
| .model-selector-panel { | |
| background: linear-gradient(135deg, #1e293b 0%, #0f172a 100%); | |
| border-color: #334155; | |
| } | |
| .hf-link-wrapper { | |
| color: #94a3b8; | |
| } | |
| .hf-link { | |
| color: #60a5fa !important; | |
| } | |
| .model-badge-info { | |
| color: #94a3b8; | |
| } | |
| .model-badge-info code { | |
| background-color: #334155; | |
| color: #f1f5f9; | |
| } | |
| .jd-result-card { | |
| background-color: #1e293b; | |
| border-color: #334155; | |
| color: #f1f5f9; | |
| } | |
| .jd-result-card .section-heading { | |
| color: #e2e8f0 !important; | |
| } | |
| .jd-result-card .border-line-double, | |
| .jd-result-card .border-line-single { | |
| color: #64748b !important; | |
| } | |
| } | |
| .border-line-double { | |
| color: #9ca3af; | |
| font-size: 13px; | |
| letter-spacing: 1px; | |
| user-select: none; | |
| overflow: hidden; | |
| white-space: nowrap; | |
| } | |
| .border-line-single { | |
| color: #d1d5db; | |
| font-size: 13px; | |
| margin: 10px 0; | |
| user-select: none; | |
| overflow: hidden; | |
| white-space: nowrap; | |
| } | |
| .jd-title { | |
| font-size: 16px; | |
| font-weight: 700; | |
| color: #111827; | |
| margin: 6px 0 2px 0; | |
| } | |
| @media (prefers-color-scheme: dark) { | |
| .jd-title { color: #f8fafc; } | |
| } | |
| .section-heading { | |
| font-size: 14px; | |
| font-weight: 600; | |
| color: #374151; | |
| margin-bottom: 6px; | |
| } | |
| .section-body { | |
| font-size: 14.5px; | |
| line-height: 1.9; | |
| word-break: break-word; | |
| } | |
| .raw-body { | |
| color: #374151; | |
| } | |
| @media (prefers-color-scheme: dark) { | |
| .raw-body { color: #cbd5e1; } | |
| } | |
| /* Các nhãn thực thể highlight trong câu */ | |
| .ner-badge { | |
| display: inline-block; | |
| padding: 1px 7px; | |
| margin: 1px 2px; | |
| border-radius: 4px; | |
| font-size: 13.5px; | |
| font-weight: 600; | |
| letter-spacing: 0.2px; | |
| box-shadow: 0 1px 2px rgba(0, 0, 0, 0.1); | |
| white-space: nowrap; | |
| } | |
| /* Phần thống kê */ | |
| .stats-body { | |
| margin-top: 4px; | |
| } | |
| .stat-row { | |
| margin-bottom: 6px; | |
| line-height: 1.7; | |
| } | |
| .stat-tag { | |
| font-size: 14px; | |
| } | |
| .stat-content { | |
| font-size: 14px; | |
| color: #374151; | |
| } | |
| @media (prefers-color-scheme: dark) { | |
| .stat-content { color: #cbd5e1; } | |
| } | |
| /* Legend badges trên đầu trang */ | |
| .legend-pill { | |
| display: inline-flex; | |
| align-items: center; | |
| padding: 4px 10px; | |
| border-radius: 9999px; | |
| font-size: 12.5px; | |
| font-weight: 600; | |
| margin-right: 8px; | |
| margin-bottom: 6px; | |
| color: white; | |
| } | |
| """ | |
| # Khởi tạo giao diện Gradio | |
| with gr.Blocks(title="XLM-RoBERTa NER Job Description") as demo: | |
| gr.HTML(f"<style>{custom_css}</style>") | |
| # Tiêu đề chính | |
| gr.Markdown("# 🏷️ Trích xuất Kỹ năng & Thực thể từ Job Description") | |
| # VỊ TRÍ THAY THẾ DÒNG CHỮ CŨ: KHỐI CHỨC NĂNG CHỌN MÔ HÌNH | |
| with gr.Group(elem_classes=["model-selector-panel"]): | |
| with gr.Row(): | |
| model_dropdown = gr.Dropdown( | |
| label="🎯 Chọn mô hình trích xuất (Model Checkpoint)", | |
| choices=MODEL_OPTIONS, | |
| value=DEFAULT_MODEL, | |
| interactive=True, | |
| scale=5, | |
| info="Chuyển đổi giữa 2 checkpoint v1 / v2 đã huấn luyện hoặc tùy chỉnh checkpoint mới." | |
| ) | |
| custom_model_textbox = gr.Textbox( | |
| label="🔗 Hugging Face Repo ID hoặc Checkpoint tùy chỉnh", | |
| placeholder="Ví dụ: NhanSoHocCode/xlmr-ner-4labels hoặc username/model-name", | |
| value="", | |
| interactive=True, | |
| visible=False, | |
| scale=5, | |
| info="Mô hình cần có cùng kiến trúc Token Classification và 4 nhãn thực thể chuẩn." | |
| ) | |
| model_link_html = gr.HTML( | |
| value=render_model_link(DEFAULT_MODEL, ""), | |
| elem_classes=["hf-link-container"] | |
| ) | |
| # Dải hiển thị 4 nhóm nhãn thực thể chuẩn | |
| gr.HTML( | |
| """ | |
| <div style="margin-top: 4px; margin-bottom: 16px;"> | |
| <span class="legend-pill" style="background-color: #ef4444;">🛠️ HARD_SKILL: Kỹ năng chuyên môn / Công nghệ</span> | |
| <span class="legend-pill" style="background-color: #d97706;">🤝 SOFT_SKILL: Kỹ năng mềm / Giao tiếp</span> | |
| <span class="legend-pill" style="background-color: #0284c7;">🎓 EDUCATION: Học vị / Bằng cấp</span> | |
| <span class="legend-pill" style="background-color: #16a34a;">📊 MAJOR: Chuyên ngành đào tạo</span> | |
| </div> | |
| """ | |
| ) | |
| with gr.Row(): | |
| with gr.Column(scale=5): | |
| title_input = gr.Textbox( | |
| label="Tiêu đề JD (Tùy chọn)", | |
| placeholder="Ví dụ: Mẫu JD Senior Golang / Distributed Systems", | |
| value="Mẫu JD Senior Golang / Distributed Systems", | |
| lines=1 | |
| ) | |
| text_input = gr.Textbox( | |
| label="Nội dung Job Description (Văn bản gốc)", | |
| placeholder="Nhập hoặc dán nội dung tuyển dụng vào đây...", | |
| value=( | |
| "Tuyển dụng Senior Backend Engineer yêu cầu tốt nghiệp Cử nhân chuyên ngành Công nghệ thông tin. " | |
| "Có từ 4 năm kinh nghiệm làm việc với Golang, gRPC, Microservices và Event-Driven Architecture. " | |
| "Thành thạo cơ sở dữ liệu PostgreSQL, Redis Caching và Apache Kafka. " | |
| "Kinh nghiệm triển khai trên Kubernetes, Docker, CI/CD. " | |
| "Sử dụng thành thạo Cursor, GitHub Copilot để tăng tốc lập trình. " | |
| "Yêu cầu kỹ năng làm việc nhóm, tư duy giải quyết vấn đề và tiếng Anh giao tiếp tốt." | |
| ), | |
| lines=6 | |
| ) | |
| with gr.Accordion("⚙️ Tùy chỉnh nâng cao", open=False): | |
| threshold_slider = gr.Slider( | |
| minimum=0.0, | |
| maximum=1.0, | |
| value=0.5, | |
| step=0.05, | |
| label="Ngưỡng độ tin cậy tối thiểu (Confidence Threshold)", | |
| info="Chỉ hiển thị các thực thể có độ tin cậy lớn hơn hoặc bằng giá trị này." | |
| ) | |
| with gr.Row(): | |
| submit_btn = gr.Button("🚀 Phân tích & Trích xuất", variant="primary", size="lg") | |
| clear_btn = gr.ClearButton([text_input, title_input], value="🧹 Xóa nội dung", size="lg") | |
| # Các ví dụ mẫu chuẩn bị sẵn | |
| gr.Examples( | |
| examples=[ | |
| [ | |
| "Mẫu JD Senior Golang / Distributed Systems", | |
| "Tuyển dụng Senior Backend Engineer yêu cầu tốt nghiệp Cử nhân chuyên ngành Công nghệ thông tin. Có từ 4 năm kinh nghiệm làm việc với Golang, gRPC, Microservices và Event-Driven Architecture. Thành thạo cơ sở dữ liệu PostgreSQL, Redis Caching và Apache Kafka. Kinh nghiệm triển khai trên Kubernetes, Docker, CI/CD. Sử dụng thành thạo Cursor, GitHub Copilot để tăng tốc lập trình. Yêu cầu kỹ năng làm việc nhóm, tư duy giải quyết vấn đề và tiếng Anh giao tiếp tốt.", | |
| 0.5, | |
| "NhanSoHocCode/xlmr-ner-4labels-v2", | |
| "" | |
| ], | |
| [ | |
| "Mẫu JD AI / Data Science Engineer", | |
| "Tìm kiếm Kỹ sư AI tốt nghiệp Thạc sĩ hoặc Cử nhân ngành Khoa học Máy tính. Có kinh nghiệm với Python, PyTorch, Transformers và mô hình ngôn ngữ lớn (LLM). Thành thạo kỹ thuật RAG, Fine-tuning, triển khai mô hình với FastAPI, Docker trên GCP. Yêu cầu tư duy phản biện, kỹ năng nghiên cứu độc lập và khả năng tiếng Anh chuyên ngành xuất sắc.", | |
| 0.5, | |
| "NhanSoHocCode/xlmr-ner-4labels-v2", | |
| "" | |
| ], | |
| [ | |
| "Mẫu JD Frontend Lead Developer", | |
| "Yêu cầu bằng Kỹ sư hoặc Đại học chuyên ngành Công nghệ phần mềm. Tối thiểu 5 năm kinh nghiệm với JavaScript, TypeScript, ReactJS, Next.js và TailwindCSS. Có kinh nghiệm tối ưu hiệu năng web, quản trị state với Redux Toolkit hoặc Zustand. Cần có kỹ năng lãnh đạo đội ngũ, khả năng giao tiếp và truyền đạt thông tin mạch lạc.", | |
| 0.5, | |
| "NhanSoHocCode/xlmr-ner-4labels", | |
| "" | |
| ] | |
| ], | |
| inputs=[title_input, text_input, threshold_slider, model_dropdown, custom_model_textbox], | |
| label="💡 Nhấp vào các mẫu JD để thử nghiệm nhanh:" | |
| ) | |
| with gr.Column(scale=6): | |
| with gr.Tabs(): | |
| with gr.TabItem("🎨 Giao diện trực quan (Visual Card)"): | |
| card_output = gr.HTML(label="Kết quả trực quan") | |
| with gr.TabItem("📋 Văn bản thô (Plain Text)"): | |
| raw_output = gr.Textbox( | |
| lines=16, | |
| max_lines=30, | |
| label="Văn bản định dạng Console (Sao chép nhanh)" | |
| ) | |
| with gr.TabItem("📊 Bảng dữ liệu chi tiết (Table)"): | |
| table_output = gr.Dataframe( | |
| headers=["STT", "Nhóm nhãn", "Thực thể", "Độ tin cậy", "Vị trí (Start:End)"], | |
| label="Danh sách thực thể chi tiết" | |
| ) | |
| # Sự kiện tương tác bộ chọn mô hình | |
| model_dropdown.change( | |
| fn=handle_model_change, | |
| inputs=[model_dropdown, custom_model_textbox], | |
| outputs=[custom_model_textbox, model_link_html] | |
| ) | |
| custom_model_textbox.input( | |
| fn=handle_custom_id_change, | |
| inputs=[custom_model_textbox], | |
| outputs=[model_link_html] | |
| ) | |
| # Xử lý sự kiện click nút | |
| common_inputs = [title_input, text_input, threshold_slider, model_dropdown, custom_model_textbox] | |
| common_outputs = [card_output, raw_output, table_output] | |
| submit_btn.click( | |
| fn=analyze_ner, | |
| inputs=common_inputs, | |
| outputs=common_outputs | |
| ) | |
| # Chạy lần đầu khi load trang | |
| demo.load( | |
| fn=analyze_ner, | |
| inputs=common_inputs, | |
| outputs=common_outputs | |
| ) | |
| if __name__ == "__main__": | |
| demo.launch() | |