import gradio as gr from collections import defaultdict import pandas as pd import os from datetime import datetime # ===================== # DISC Predictor # ===================== class DISCPredictor: def __init__(self, mapping=None): self.mapping = mapping if mapping else {} self.traits = ['D','I','S','C'] def predict_percentage(self, answers): scores = defaultdict(float) for q_idx, ans in enumerate(answers): key = q_idx + 1 ans = str(ans).upper().strip() qmap = self.mapping.get(key, {}) choice_map = qmap.get(ans, {}) for trait, w in choice_map.items(): scores[trait] += float(w) total = sum(scores.values()) if sum(scores.values()) != 0 else 1.0 percentages = {t: round((scores.get(t,0)/total)*100) for t in self.traits} max_score = max(scores.values()) if scores else 0 winners = [t for t, s in scores.items() if s == max_score] main_trait = '/'.join(sorted(winners)) if winners else "" return main_trait, percentages # ===================== # Mapping # ===================== mapping = { 1: {'A': {'D':2}, 'B': {'I':2}, 'C': {'S':2}, 'D': {'C':2}}, 2: {'A': {'C':2}, 'B': {'I':2}, 'C': {'S':2}, 'D': {'D':2}}, 3: {'A': {'S':2}, 'B': {'I':2}, 'C': {'C':2}, 'D': {'D':2}}, 4: {'A': {'S':2}, 'B': {'C':2}, 'C': {'D':2}, 'D': {'I':2}}, 5: {'A': {'D':2}, 'B': {'I':2}, 'C': {'S':2}, 'D': {'C':2}}, 6: {'A': {'I':2}, 'B': {'S':2}, 'C': {'C':2}, 'D': {'D':2}}, 7: {'A': {'S':2}, 'B': {'C':2}, 'C': {'I':2}, 'D': {'D':2}}, 8: {'A': {'C':2}, 'B': {'S':2}, 'C': {'D':2}, 'D': {'I':2}}, 9: {'A': {'D':2}, 'B': {'I':2}, 'C': {'S':2}, 'D': {'C':2}}, 10:{'A': {'I':2}, 'B': {'C':2}, 'C': {'D':2}, 'D': {'S':2}}, } # ===================== # คำถาม DISC # ===================== questions_text = { 1: {"ฉันรู้สึกว่ามันยากที่จะผ่อนคลาย ใช้ชีวิตสบาย ๆ หรืออยู่นิ่ง ๆ ": "A", "ฉันมีเพื่อนเยอะ ": "B", "ฉันพร้อมที่จะช่วยเหลือผู้อื่นเสมอ ": "C", "ฉันชอบความสมบูรณ์แบบ": "D"}, 2: {"ฉันเป็นคนเรียบร้อย มีระเบียบ": "A", "ฉันเป็นคนกระตือรือร้น ทั้ง work hard และ play hard": "B", "ฉันเป็นคนใจเย็นมาก": "C", "ฉันมีทางของตัวเอง": "D"}, 3: {"ฉันค่อนข้างพอใจกับวิธีชีวิตปกติของฉัน": "A", "ฉันมีความคิดสร้างสรรค์และไม่ชอบกฎเกณฑ์": "B", "ฉันชอบความเงียบและความสงบสุข": "C", "ฉันมองโลกในแง่ดี": "D"}, 4: {"ฉันเป็นผู้ฟังที่ดี": "A", "ฉันชอบสิ่งที่แน่นอน คาดเดาได้": "B", "ฉันมั่นใจในตัวเองสูง": "C", "ฉันชอบเรื่องตลกขบขัน": "D"}, 5: {"ฉันชอบการแข่งขัน": "A", "ฉันไม่ชอบการที่ต้องจริงจังกับชีวิตมากเกินไป": "B", "ฉันคิดถึงจิตใจผู้อื่นเสมอ": "C", "ฉันไม่ชอบเสี่ยง ไม่ชอบสิ่งที่ไม่แน่นอน": "D"}, 6: {"ฉันโน้มน้าวใจคนอื่นเก่ง": "A", "ฉันเป็นคนอ่อนโยน ขี้เกรงใจ": "B", "ฉันทำอะไรด้วยความระมัดระวังเสมอ": "C", "ฉันเป็นคนกล้าหาญและเด็ดเดี่ยว": "D"}, 7: {"ฉันให้อภัยผู้อื่นเสมอ": "A", "ฉันเป็นคนมีเหตุผลและเข้มแข็ง": "B", "ฉันเข้าสังคมเก่ง ชอบติดต่อสื่อสาร": "C", "ฉันมักตัดสินใจเร็ว กล้าหาญ": "D"}, 8: {"ฉันปฏิบัติตามกฎระเบียบเสมอ": "A", "ฉันใจเย็น มีน้ำใจ": "B", "ฉันเป็นคนมุ่งมั่น ชอบวิเคราะห์": "C", "ฉันร่าเริง เข้าสังคมเก่ง": "D"}, 9: {"ฉันชอบที่จะเป็นที่ 1": "A", "ฉันเป็นคนสุภาพ อัธยาศัยดี": "B", "ฉันเคารพความคิดเห็นของผู้อื่น": "C", "ฉันไม่ชอบเข้าสังคม": "D"}, 10: {"ฉันคิดว่าฉันเป็นคนมีเสน่ห์": "A", "ฉันวิเคราะห์สถานการณ์ได้ดี": "B", "ฉันเป็นคนใจร้อน ตัดสินใจเร็ว": "C", "เวลาจะทำอะไร ฉันจะคิดก่อนว่ามันจะกระทบผู้อื่นหรือไม่": "D"}, } # ===================== # คำอธิบาย DISC # ===================== disc_descriptions = { "D": "กล้าตัดสินใจ มุ่งมั่น ชอบการแข่งขัน และเป็นผู้นำ", "I": "เข้าสังคมเก่ง ร่าเริง ชอบสื่อสาร โน้มน้าวผู้อื่นได้ดี", "S": "ใจเย็น อดทน สุภาพ มีน้ำใจ ชอบช่วยเหลือผู้อื่น", "C": "รอบคอบ มีเหตุผล ชอบความถูกต้อง และทำงานเป็นระบบ", } # ===================== # ฟังก์ชันตรวจสอบปุ่ม Submit # ===================== def check_all_selected(affiliation, age_group, *args): try: q1 = args[0] q2 = args[1] except Exception: q1 = q2 = None if affiliation and age_group and q1 and q2: return gr.update(interactive=True) else: return gr.update(interactive=False) # ===================== # ฟังก์ชันแสดงผล + บันทึก Excel ใน test/ # ===================== def disc_test(affiliation, age_group, *args): answers_letters = [] for idx, text in enumerate(args): key = idx + 1 ans = questions_text[key].get(text, None) if ans is None: ans = 'A' answers_letters.append(ans) predictor = DISCPredictor(mapping) main_trait, percentages = predictor.predict_percentage(answers_letters) # บันทึกไฟล์ Excel ลง test/ folder_path = "https://huggingface.co/spaces/hongkokokok/save_data" os.makedirs(folder_path, exist_ok=True) file_path = os.path.join(folder_path, "disc_results.xlsx") ts = datetime.now().strftime("%Y-%m-%d %H:%M:%S") row = { "timestamp": ts, "affiliation": affiliation, "age_group": age_group, "Main Trait": main_trait, "D %": percentages.get("D", 0), "I %": percentages.get("I", 0), "S %": percentages.get("S", 0), "C %": percentages.get("C", 0), } for i, a in enumerate(answers_letters, start=1): row[f"Q{i}"] = a try: if os.path.exists(file_path): existing = pd.read_excel(file_path) df = pd.concat([existing, pd.DataFrame([row])], ignore_index=True) else: df = pd.DataFrame([row]) df.to_excel(file_path, index=False) saved_msg = f"บันทึกผลเรียบร้อย: {file_path}" except Exception as e: saved_msg = f"ไม่สามารถบันทึกไฟล์ Excel ได้: {e}" # สร้างผลลัพธ์ HTML result_text = f"""

🧠 ผลลัพธ์ DISC ของคุณ: {main_trait}

สังกัด: {affiliation}   |   ช่วงอายุ: {age_group}
บันทึกเมื่อ: {ts}
""" sorted_traits = sorted(percentages.items(), key=lambda x: x[1], reverse=True) bg_color_map = {"D": "#ffebee", "I": "#fffde7", "S": "#e8f5e9", "C": "#e3f2fd"} text_color_map = {"D": "red", "I": "#fbc02d", "S": "green", "C": "blue"} icon_map = {"D":"🏆", "I":"🎉", "S":"🤝", "C":"📊"} result_text += "
" for t, pct in sorted_traits: desc = disc_descriptions.get(t,"") result_text += f"""
{icon_map.get(t,'')} {t}: {pct}%
{desc}
""" result_text += "
" result_text += f"
{saved_msg}
" return result_text # ===================== # Gradio UI # ===================== with gr.Blocks() as demo: gr.HTML(""" """) gr.HTML("

แบบทดสอบ DISC

") with gr.Row(): with gr.Column(scale=2): affiliation = gr.Dropdown( choices=["งานการพยาบาลผู้ป่วยกุมารเวชศาสตร์","งานการพยาบาลผู้ป่วยทั่วไป จิตเวชและจักษุ โสต ศอ นาสิก","อื่นๆ"], label="สังกัด (โปรดเลือก)", ) with gr.Column(scale=1): age_group = gr.Radio( choices=["น้อยกว่า 20 ปี","21-28 ปี","29-36 ปี","37–44 ปี","45–52 ปี","53-60 ปี"], label="ช่วงอายุ (โปรดเลือก)", ) inputs = [] for i in range(1, 11): with gr.Group(elem_classes=["radio-question"]): gr.HTML(f'
ข้อ {i}
') radio = gr.Radio( list(questions_text[i].keys()), label="", elem_classes=["big-radio"] ) inputs.append(radio) output = gr.HTML(label="ผลลัพธ์ DISC") btn = gr.Button("Submit", elem_classes=["submit-btn"], interactive=False) btn.click(disc_test, inputs=[affiliation, age_group] + inputs, outputs=[output]) for r in inputs: r.change(check_all_selected, inputs=[affiliation, age_group] + inputs, outputs=[btn]) affiliation.change(check_all_selected, inputs=[affiliation, age_group] + inputs, outputs=[btn]) age_group.change(check_all_selected, inputs=[affiliation, age_group] + inputs, outputs=[btn]) if __name__ == "__main__": demo.launch()