Upload localhost/app_v4.py

localhost/app_v4.py

@@ -0,0 +1,170 @@
+import gradio as gr
+import json
+import os
+import re
+from collections import Counter
+
+# ── CONFIG ───────────────────────────────────────────────────────────────────
+
+KB_PATH = "knowledge_base_v4.json"
+
+# Load knowledge base
+with open(KB_PATH, "r", encoding="utf-8") as f:
+    KB = json.load(f)
+
+PATTERNS = KB.get("by_pattern", {})
+FAILURE_KEYS = KB.get("by_failure_key", {})
+LOOKUP = KB.get("lookup", {})
+
+# ── FUZZY SEARCH ───────────────────────────────────────────────────────────────
+
+def find_best_pattern(query):
+    """Tìm pattern phù hợp nhất với query của user"""
+    query = query.upper()
+    
+    # 1. Exact match
+    for pattern in PATTERNS:
+        if pattern.upper() == query:
+            return pattern, 1.0
+    
+    # 2. Substring match
+    for pattern in PATTERNS:
+        if pattern.upper() in query or query in pattern.upper():
+            return pattern, 0.9
+    
+    # 3. Keyword matching
+    query_words = set(re.findall(r'\b[A-Z]{2,}\b', query))
+    best_match = None
+    best_score = 0
+    
+    for pattern in PATTERNS:
+        pattern_words = set(re.findall(r'\b[A-Z]{2,}\b', pattern.upper()))
+        if not pattern_words:
+            continue
+        common = query_words & pattern_words
+        score = len(common) / max(len(query_words), len(pattern_words))
+        if score > best_score:
+            best_score = score
+            best_match = pattern
+    
+    if best_score >= 0.3:
+        return best_match, best_score
+    
+    # 4. Failure key matching
+    for fk in FAILURE_KEYS:
+        if fk.upper() in query:
+            return fk, 0.5  # Return failure key for broad matching
+    
+    return None, 0
+
+# ── RESPONSE BUILDER ─────────────────────────────────────────────────────────
+
+def build_response(query):
+    """Xây dựng phản hồi từ knowledge base"""
+    
+    pattern, score = find_best_pattern(query)
+    
+    if not pattern:
+        return "Không tìm thấy thông tin phù hợp. Vui lòng mô tả lỗi chi tiết hơn (ví dụ: 'lỗi channel 53', 'TRIP PE TEM', 'DDR failed')."
+    
+    # Get entry
+    if pattern in PATTERNS:
+        entry = PATTERNS[pattern]
+    elif pattern in FAILURE_KEYS:
+        entry = FAILURE_KEYS[pattern]
+    else:
+        return "Lỗi: Không tìm thấy dữ liệu."
+    
+    # Build response
+    lines = []
+    lines.append(f"## Phân tích lỗi: {pattern}")
+    lines.append(f"")
+    lines.append(f"- **Tổng cases:** {entry['total_cases']}")
+    lines.append(f"- **Pass rate:** {entry['pass_rate']}%")
+    lines.append(f"- **Pass:** {entry['pass_count']} | **Fail:** {entry['fail_count']}")
+    lines.append(f"- **Board types:** {', '.join(entry.get('board_types', []))}")
+    lines.append(f"")
+    
+    # BKM Procedure
+    bkm_procs = entry.get('bkm_procedures', [])
+    if bkm_procs:
+        lines.append(f"## Quy trình BKM")
+        for proc in bkm_procs[:3]:
+            lines.append(f"- {proc}")
+        lines.append(f"")
+    
+    # BKM Components
+    bkm_comps = entry.get('bkm_components', [])
+    if bkm_comps:
+        lines.append(f"## Linh kiện theo BKM")
+        lines.append(f"- {', '.join(bkm_comps[:10])}")
+        lines.append(f"")
+    
+    # Priority Replace
+    priority = entry.get('priority_replace', [])
+    if priority:
+        lines.append(f"## Thống kê - Linh kiện thay nhiều nhất")
+        lines.append(f"| Rank | Linh kiện | Pass Rate | Số lần thay | Pass | Fail |")
+        lines.append(f"|------|-----------|-----------|-------------|------|------|")
+        for i, p in enumerate(priority[:10], 1):
+            lines.append(f"| {i} | {p['component']} | {p['pass_rate']}% | {p['count']} | {p['pass']} | {p['fail']} |")
+        lines.append(f"")
+    
+    # Best Actions
+    best_actions = entry.get('best_actions', [])
+    if best_actions:
+        lines.append(f"## Hành động hiệu quả nhất")
+        lines.append(f"| Rank | Hành động | Linh kiện | Pass Rate | Số lần | Pass | Fail |")
+        lines.append(f"|------|-----------|-----------|-----------|--------|------|------|")
+        for i, a in enumerate(best_actions[:10], 1):
+            lines.append(f"| {i} | {a['action']} | {a['component']} | {a['pass_rate']}% | {a['count']} | {a['pass']} | {a['fail']} |")
+        lines.append(f"")
+    
+    # Top Technicians
+    techs = entry.get('top_technicians', [])
+    if techs:
+        lines.append(f"## Kỹ thuật viên tốt nhất")
+        lines.append(f"| Rank | Tên | Pass Rate | Tổng cases | Pass |")
+        lines.append(f"|------|-----|-----------|------------|------|")
+        for i, t in enumerate(techs[:5], 1):
+            lines.append(f"| {i} | {t['name']} | {t['pass_rate']}% | {t['total']} | {t['pass']} |")
+        lines.append(f"")
+    
+    # Sample descriptions
+    samples = entry.get('sample_original_descriptions', [])
+    if samples:
+        lines.append(f"## Ví dụ mô tả lỗi tương tự")
+        for s in samples:
+            lines.append(f"- {s}")
+        lines.append(f"")
+    
+    lines.append(f"---")
+    lines.append(f"*Độ chính xác tìm kiếm: {int(score*100)}%*")
+    
+    return "\n".join(lines)
+
+# ── GRADIO UI ────────────────────────────────────────────────────────────────
+
+def chat(message, history):
+    """Chat function for Gradio"""
+    response = build_response(message)
+    return response
+
+# Create interface
+demo = gr.ChatInterface(
+    fn=chat,
+    title="HDMT Debug Assistant v4",
+    description="Chatbot hỗ trợ debug HDMT boards. Mô tả lỗi để nhận phân tích và hướng dẫn.",
+    examples=[
+        "lỗi channel 53",
+        "TRIP PE TEM",
+        "DDR failed",
+        "blue screen",
+        "ADATE error",
+        "FPGA calibration",
+        "MISSING J18",
+    ],
+)
+
+if __name__ == "__main__":
+    demo.launch()