Update app.py

app.py

@@ -1,34 +1,168 @@
+import tensorflow as tf
+from tensorflow.keras.models import load_model
 import gradio as gr
+import cv2
+import numpy as np
+from tensorflow.keras.applications.vgg19 import preprocess_input
+import gdown
 
-# Fungsi untuk tab "Home"
-def home_content():
-    return "Selamat datang di halaman Home!"
-
-# Fungsi untuk tab "Coba sekarang"
-def coba_sekarang_content():
-    return "Ini adalah halaman 'Coba sekarang'!"
-
-# Membuat antarmuka dengan tab
-with gr.Blocks() as demo:
-    with gr.Tab("Home"):
-        with gr.Column():
-            gr.Markdown("### Home Page")
-            gr.Textbox(home_content, label="Home", elem_id="home_box")
-
-    with gr.Tab("Coba sekarang"):
-        with gr.Column():
-            gr.Markdown("### Coba sekarang")
-            gr.Textbox(coba_sekarang_content, label="Coba sekarang", elem_id="coba_sekarang_box")
-
-    # Menggunakan CSS untuk mengganti warna latar belakang tiap tab
-    demo.css = """
-    #home_box {
-        background-color: lightblue;
-    }
-    #coba_sekarang_box {
-        background-color: lightgreen;
-    }
-    """
-
-# Menjalankan aplikasi Gradio
-demo.launch()
+# Download model dari Google Drive
+file_id = '1YUV4xVBwXfUMwBiCfZ53G32vsZ2XJtp6'
+gdown.download(f"https://drive.google.com/uc?export=download&id={file_id}", "AiCNE_Model_Akhir.h5", quiet=False)
+
+# Load model
+model = load_model("AiCNE_Model_Akhir.h5")
+
+# Labels dan saran pengobatan untuk deteksi acne
+acne_labels = {
+    0: 'Acne',
+    1: 'Clear',
+    2: 'Comedo'
+}
+
+acne_treatment = {
+    0: '- Topical anti-acne agents, such as benzoyl peroxide, azelaic acid, and tretinoin or adapalene gel and some antibiotics (clindamycin)\n- New bioactive proteins may also prove successful\n- Newer topical agents such as clascoterone\n- Low-dose combined oral contraceptive\n- Antiseptic or keratolytic washes containing salicylic acid\n- Light/laser therapy',
+    1: 'Keep up the good work! ',
+    2: '- Benzoyl peroxide\n- Azelaic acid\n- Salicylic acid +/- sulfur and resorcinol\n- Glycolic acid\n- Retinoids such as tretinoin, isotretinoin, adapalene (these require a doctors prescription)'
+}
+
+# Fungsi untuk mendeteksi acne
+def detect_acne(image, threshold=0.4):
+    # Resize gambar menjadi 224x224 piksel agar sesuai dengan ukuran input model
+    image_resized = cv2.resize(image, (224, 224))
+    
+    # Proses gambar dengan preprocess_input untuk menyesuaikan format input model
+    input_data = preprocess_input(np.expand_dims(image_resized, axis=0))  # Menambah dimensi untuk batch
+
+    # Mendapatkan prediksi dari model
+    predictions = model.predict(input_data)
+
+    # Menemukan indeks kelas dengan probabilitas tertinggi
+    max_index = np.argmax(predictions[0])  # Indeks dengan probabilitas tertinggi
+    max_prob = predictions[0][max_index]  # Nilai probabilitas tertinggi
+
+    # Inisialisasi variabel untuk hasil deteksi dan saran pengobatan
+    detections = []  # Daftar untuk menyimpan label deteksi (misalnya 'Acne', 'Clear', 'Comedo')
+    treatment_suggestion = ""  # Saran pengobatan yang sesuai
+
+    # Jika probabilitas tertinggi lebih besar dari threshold, deteksi berhasil
+    if max_prob > threshold:
+        detections.append(acne_labels[max_index])  # Menambahkan label deteksi ke dalam daftar
+        treatment_suggestion = acne_treatment[max_index]  # Mendapatkan saran pengobatan berdasarkan deteksi
+
+    # Mengembalikan hasil: gambar yang sudah dianotasi, hasil deteksi, dan saran pengobatan
+    return f"Detected face: {detections}", treatment_suggestion
+
+
+# CSS untuk mempercantik tampilan antarmuka
+custom_css = """
+@import url('https://fonts.googleapis.com/css2?family=Itim&display=swap');
+@import url('https://fonts.googleapis.com/css2?family=Instrument+Sans:wght@400;700&display=swap');
+body {
+    background: linear-gradient(to bottom, #FFA500, #1E90FF); 
+    color: #ffffff !important; 
+    margin: 0;
+    padding: 0;
+    overflow: hidden; 
+}
+
+.gradio-container {
+    max-height: 90vh; 
+    overflow-y: auto; 
+    background: transparent !important;
+    color: #ffffff !important;
+    border-radius: 12px;
+    text-align: center;
+    box-sizing: border-box;
+    padding: 10px;
+}
+
+.gradio-container .wrap h1 {
+    font-size: 100px !important; 
+    font-family: 'Itim', cursive !important; 
+    font-weight: bold !important;
+    text-shadow: 2px 2px 4px rgba(0, 0, 0, 0.5); 
+    color: #000000 !important; 
+}
+
+h2 {
+    font-size: 40px;
+    font-family: 'Instrument Sans', sans-serif;
+    font-weight: bold;
+    color: #000000;
+    text-align: center;
+}
+
+p {
+    font-size: 30px;
+    font-family: 'Instrument Sans', sans-serif;
+    font-weight: bold;
+    color: #000000;
+    text-align: center;
+}
+
+.gradio-container .wrap {
+    font-size: 40px !important; 
+    line-height: 1.6;
+    padding: 20px
+}
+
+.output-textbox textarea {
+    background-color: #1e1e1e !important;
+    color: #ffffff !important;
+    border: 2px solid #000000; 
+    font-weight: bold;
+    font-size: 16px;
+    padding: 10px;
+    border-radius: 8px;
+}
+
+.gradio-container::-webkit-scrollbar {
+    width: 12px;
+}
+
+.gradio-container::-webkit-scrollbar-track {
+    background: rgba(255, 255, 255, 0.2);
+    border-radius: 6px; 
+}
+
+.gradio-container::-webkit-scrollbar-thumb {
+    background: #D3D3D3;
+    border-radius: 6px;
+    border: #555555; 
+}
+
+.gradio-container::-webkit-scrollbar-thumb:active {
+    background: #555555; 
+    border: #555555; 
+}
+"""
+
+interface = gr.Interface(
+    fn=detect_acne,
+    inputs=gr.Image(type="numpy", label="Upload an image"),
+    outputs=[
+        gr.Textbox(label="Detection Result"),
+        gr.Textbox(label="Treatment Suggestion")
+    ],
+    title="🌟 AiCNE 🌟",
+    description=(
+        "<div style='text-align: center;'>"
+        "<h2>Welcome to <b>AiCNE</b>, your AI-powered assistant for acne detection!</h2>"
+        "<h2>Upload a clear image of your face to analyze and classify acne types.</h2>"
+        "<h2><b>Get instant results</b> and take a step closer to understanding your skin!</h2>"
+        "<br>"
+        "<br>"
+        "<h2>What is AiCNE?</h2>"
+        "<p>AiCNE is an AI-Powered Acne Detection tool, a smart solution to understand your skin condition with AI Technology.</p>"
+        "<br>"
+        "<h2>Why use AiCNE?</h2>"
+        "<p>AiCNE detects acne within seconds with high accuracy, offering a user-friendly interface for your convenience.</p>"
+        "<br>"
+        "<br>"
+        "</div>"
+    ),
+    css=custom_css,
+)
+
+interface.launch()