Create skin.py

MeAI_Maincode/skin.py

@@ -0,0 +1,207 @@
+import requests
+import gradio as gr
+from PIL import Image
+import io
+import numpy as np
+import os
+key = "a8108953dbmsh2baabfa3db731fdp14177bjsnfebbe465261d"
+trans_disease = {
+    "acne": "mụn",
+    "actinic_keratosis": "chứng dày sừng quang hóa",
+    "alopecia_androgenetica": "chứng rụng tóc nội tiết tố androgen",
+    "alopecia_areata": "chứng rụng tóc từng vùng",
+    "bullous_dermatosis": "bệnh da bọng nước",
+    "chloasma": "nám da",
+    "corn": "chứng chai da",
+    "dermatofibroma": "u xơ da",
+    "eczema_dermatitis": "viêm da chàm",
+    "erysipelas": "viêm quầng",
+    "erythema_multiforme": "ban đỏ đa dạng",
+    "folliculitis": "viêm nang lông",
+    "furuncle": "mụn nhọt",
+    "haemangioma": "bệnh u máu",
+    "herpes": "mụn rộp",
+    "herpes_simplex": "nhiễm trùng do virus Herpes Simplex",
+    "iga_vasculitis": "viêm mạch máu Iga",
+    "keloid": "sẹo lồi",
+    "keratosis_follicularism": "bệnh nang lông dày sừng",
+    "lichen_planus": "bệnh lichen phẳng",
+    "lupus_erythematosus": "bệnh ban đỏ",
+    "molluscum_contagiosum": "u mềm lây",
+    "nevus": "nốt ruồi",
+    "paronychia": "viêm quanh móng",
+    "pityriasis_alba": "bệnh vẩy phấn trắng",
+    "pityriasis_rosea": "bệnh vảy phấn hồng",
+    "prurigo_nodularis": "bệnh sẩn ngứa",
+    "psoriasis": "bệnh vẩy nến",
+    "rosacea": "bệnh trứng cá đỏ rosacea",
+    "sebaceous_cyst": "u nang bã nhờn",
+    "sebaceousnevus": "bớt tuyến bã",
+    "seborrheic_dermatitis": "viêm da tiết bã",
+    "seborrheic_keratosis": "chứng dày sừng tiết bã",
+    "skin_tag": "mụn thịt dư",
+    "stasis_dermatitis": "viêm da ứ đọng",
+    "syringoma": "u ống tuyến mồ hôi",
+    "tinea_capitis": "nấm da đầu",
+    "tinea_corporis": "nấm cơ thể",
+    "tinea_cruris": "nấm bẹn",
+    "tinea_manuum": "",
+    "tinea_pedis": "nấm chân",
+    "tinea_unguium": "nấm móng tay móng chân",
+    "tinea_versicolor": "bệnh lang ben",
+    "urticaria": "phát ban",
+    "urticaria_papular": "nổi mề đay",
+    "varicella": "thủy đậu",
+    "verruca_plana": "mụn cóc phẳng",
+    "verruca_vulgaris": "mụn cóc thông thường",
+    "vitiligo": "bệnh bạch biến"
+}
+trans_body = {
+    "head": "đầu",
+    "neck": "cổ",
+    "hand": "tay",
+    "arm": "cánh tay",
+    "leg": "chân",
+    "foot": "bàn chân",
+    "back": "lưng",
+    "chest": "ngực",
+    "abdomen": "bụng",
+    "face": "mặt",
+    "ear": "tai",
+    "eye": "mắt",
+    "nose": "mũi",
+    "mouth": "miệng",
+    "lip": "môi",
+    "cheek": "má",
+    "tongue": "lưỡi",
+    "throat": "cổ họng",
+    "forehead": "trán",
+    "chin": "cằm",
+    "unknown" : "bộ phận chưa rõ"
+}
+trans_description = {
+    "acne": "Tình trạng da phổ biến với mụn trứng cá",
+    "pityriasis_rosea": "Phát ban thường bắt đầu bằng một đốm lớn",
+    "actinic_keratosis": "Các mảng da thô ráp, có vảy do tổn thương từ ánh nắng",
+    "prurigo_nodularis": "Các cục cứng và ngứa trên da",
+    "alopecia_androgenetica": "Rụng tóc theo kiểu mẫu",
+    "alopecia_areata": "Rụng tóc từng mảng",
+    "bullous_dermatosis": "Tình trạng da bị phồng rộp",
+    "chloasma": "Các mảng nâu trên da",
+    "corn": "Vùng da bị dày lên",
+    "dermatofibroma": "Sự phát triển da lành tính phổ biến",
+    "eczema_dermatitis": "Da bị viêm và ngứa",
+    "erysipelas": "Nhiễm trùng da do vi khuẩn",
+    "erythema_multiforme": "Phản ứng trên da",
+    "folliculitis": "Viêm nang lông",
+    "furuncle": "Nhiễm trùng sâu trong nang lông",
+    "haemangioma": "Sự phát triển lành tính của mạch máu",
+    "herpes": "Bệnh zona thần kinh",
+    "herpes_simplex": "Mụn rộp môi",
+    "iga_vasculitis": "Viêm mạch máu",
+    "keloid": "Mô sẹo phát triển quá mức",
+    "keratosis_follicularism": "Các mảng thô ráp và mụn nhỏ",
+    "lichen_planus": "Tình trạng viêm",
+    "lupus_erythematosus": "Bệnh tự miễn",
+    "molluscum_contagiosum": "Nhiễm trùng da do virus",
+    "nevus": "Nốt ruồi",
+    "paronychia": "Nhiễm trùng móng tay/móng chân",
+    "pityriasis_alba": "Vùng da sáng màu",
+    "psoriasis": "Tình trạng da mãn tính",
+    "rosacea": "Mặt đỏ và có mụn nhỏ",
+    "sebaceous_cyst": "U nang không phải ung thư",
+    "sebaceousnevus": "Dấu vết bẩm sinh",
+    "seborrheic_dermatitis": "Da đỏ và có vảy",
+    "seborrheic_keratosis": "Sự phát triển màu nâu và có vảy",
+    "skin_tag": "Sự phát triển nhỏ, mềm trên da",
+    "stasis_dermatitis": "Viêm da",
+    "syringoma": "Các nốt nhỏ trên da",
+    "tinea_capitis": "Nấm da đầu",
+    "tinea_corporis": "Nấm trên cơ thể",
+    "tinea_cruris": "Nấm bẹn",
+    "tinea_manuum": "Nấm tay",
+    "tinea_pedis": "Nấm chân",
+    "tinea_unguium": "Nấm móng tay/móng chân",
+    "tinea_versicolor": "Nhiễm trùng do nấm",
+    "urticaria": "Phát ban mề đay",
+    "urticaria_papular": "Nốt ngứa",
+    "varicella": "Bệnh thủy đậu",
+    "verruca_plana": "Mụn cóc phẳng",
+    "verruca_vulgaris": "Mụn cóc thông thường",
+    "vitiligo": "Mất màu da"
+}
+def detect_skin_disease_new(image, key):
+    try:
+
+        img = Image.fromarray(image.astype('uint8'))  # Tạo ảnh từ numpy array
+        img_byte_array = io.BytesIO()  # Tạo buffer giống tệp
+        img.save(img_byte_array, format='PNG')  # Lưu ảnh dưới dạng PNG
+        img_byte_array.seek(0)  # Quay về đầu buffer
+
+        url = "https://detect-skin-disease1.p.rapidapi.com/skin-disease"
+        files = {"image": ("image.png", img_byte_array, "image/png")}
+        headers = {
+            "X-RapidAPI-Key": key,
+            "X-RapidAPI-Host": "detect-skin-disease1.p.rapidapi.com"
+        }
+        response = requests.post(url, files=files, headers=headers)
+
+        if response.status_code != 200:
+            return f"Lỗi API: HTTP {response.status_code} - {response.reason}"
+        response_json = response.json()
+
+        output = ""
+
+        # Get image type
+        image_type = response_json.get('imageType')
+        if image_type=="normal_skin":
+            return "Da của bạn bình thường\n"
+        if image_type=="non_skin":
+            return "Ảnh của bạn không phải da\n"
+
+        # Get body part
+        body_part = response_json.get('bodyPart')
+        if body_part is not None:
+            vnese_body = trans_body.get(body_part, body_part)
+            output += f"Phần của cơ thể: {vnese_body}\n"
+
+        # Get results and probabilities
+        results = response_json.get('results')
+        if results is not None:
+            output += "Kết quả phân tích:\n"
+
+            # Sort results by probability in descending order
+            sorted_results = sorted(results.items(), key=lambda x: x[1], reverse=True)
+
+            for disease, probability in sorted_results:
+                probability_percent = probability * 100
+                vnese_disease = trans_disease.get(disease, disease)
+                description = trans_description.get(disease, "Không có mô tả")
+                
+                if probability_percent >= 10:
+                    output += f"- {vnese_disease} {probability_percent:.2f}%, mô tả thêm về bệnh: {description} \n"
+
+        return output
+    except Exception as e:
+        return f"Lỗi: {str(e)}"
+
+
+
+
+
+def create_skin_tab_new(skinkey="a8108953dbmsh2baabfa3db731fdp14177bjsnfebbe465261d"):
+    css = """
+    .textboxskin {
+        font-sxxxxize: 50px; !important;
+    }
+    """
+    with gr.Blocks(css=css) as demo:
+        keybox = gr.Text(value=skinkey,visible=False)
+        gr.Markdown("Hãy tải ảnh lên và nhấn **Xử Lý** để chẩn đoán bệnh ngoài da.")
+        with gr.Row():
+          inp = gr.Image(type="numpy",height=512, width=512,
+          value=os.path.join(os.path.dirname(__file__), "../anh/thuydau.jpg"))
+          out = gr.Label(label="Kết Quả Dự Đoán",elem_classes="textboxskin")
+        btn = gr.Button("Xử Lý")
+        btn.click(fn=detect_skin_disease, inputs=[inp,keybox], outputs=out)
+    return demo