Update MeAI_Maincode/Detectskindisease.py

MeAI_Maincode/Detectskindisease.py

@@ -79,52 +79,117 @@ trans_body = {
     "chin": "cằm",
     "unknown" : "bộ phận chưa rõ"
 }
-def detect_skin_disease(image,key):
+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(image, key):
     try:
-
         # Convert NumPy array to image file-like object
         img = Image.fromarray((image).astype('uint8'))
         img_byte_array = io.BytesIO()
         img.save(img_byte_array, format='PNG')
         img_byte_array.seek(0)
 
-        url = "https://detect-skin-disease.p.rapidapi.com/facebody/analysis/detect-skin-disease"
-        # files = {"image": img_byte_array}
+
+        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-disease.p.rapidapi.com"
+            "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 = ""
 
-        if 'data' in response_json:
-            body_part = response_json['data'].get('body_part')
-            results = response_json['data'].get('results_english')
+        # 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"
 
-            if body_part is not None:
-                vnese_body = trans_body[body_part]
-                output += f"Phần của cơ thể: {vnese_body} ({body_part})\n"
-                
-            if results is not None:
-              output += " Kết quả phân tích:      "
+        # 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 the results by probability percentage in descending order
-              sorted_results = sorted(results.items(), key=lambda x: x[1], reverse=True)
+            # 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:
+            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}%\n"
+                    output += f"- {vnese_disease} {probability_percent:.2f}%, mô tả thêm về bệnh: {description} \n"
 
-            return output
-        else:
-            return "Không có dữ liệu phản hồi từ API."
+        return output
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Lỗi: {str(e)}"
+
+
+
+
+
 
 def create_skin_tab(skinkey="b71d1f41camsh51cbf0a2df5facap1b9e25jsn2ce94a61f252"):
     css = """