Update app.py

app.py

@@ -1,37 +1,205 @@
 import gradio as gr
 from transformers import pipeline
 from PIL import Image
+import numpy as np
 
 # Załaduj model klasyfikacji zmian skórnych
 skin_model = pipeline("image-classification", model="Anwarkh1/Skin_Cancer-Image_Classification")
 
+# Definicje typów zmian skórnych z interpretacją medyczną
+SKIN_CONDITIONS = {
+    "Actinic keratoses": {
+        "name": "Rogowacenie słoneczne",
+        "risk": "medium",
+        "description": "Przedrakowе zmiany wywołane długotrwałym narażeniem na słońce",
+        "recommendation": "Wymaga kontroli dermatologicznej - może prowadzić do raka płaskonabłonkowego"
+    },
+    "Basal cell carcinoma": {
+        "name": "Rak podstawnokomórkowy",
+        "risk": "high",
+        "description": "Najczęstszy typ raka skóry, rzadko daje przerzuty",
+        "recommendation": "PILNIE skonsultuj się z onkologiem/dermatologiem"
+    },
+    "Dermatofibroma": {
+        "name": "Włókniak skóry",
+        "risk": "low",
+        "description": "Łagodny guzek składający się z tkanki włóknistej",
+        "recommendation": "Zwykle nie wymaga leczenia, kontrola dermatologiczna"
+    },
+    "Melanoma": {
+        "name": "Czerniak",
+        "risk": "high",
+        "description": "Najgroźniejszy typ raka skóry, może szybko dawać przerzuty",
+        "recommendation": "NATYCHMIAST skonsultuj się z onkologiem!"
+    },
+    "Nevus": {
+        "name": "Znamię",
+        "risk": "low",
+        "description": "Zwykle łagodna zmiana pigmentowa",
+        "recommendation": "Obserwuj zmiany według reguły ABCDE"
+    },
+    "Pigmented benign keratosis": {
+        "name": "Łagodne rogowacenie barwnikowe",
+        "risk": "low",
+        "description": "Łagodna zmiana wieku, często ciemno zabarwiona",
+        "recommendation": "Zwykle nie wymaga leczenia"
+    },
+    "Seborrheic keratosis": {
+        "name": "Rogowacenie łojotokowe",
+        "risk": "low",
+        "description": "Łagodna zmiana starcza, często brodawkowata",
+        "recommendation": "Nie wymaga leczenia, możliwość usunięcia kosmetycznego"
+    },
+    "Squamous cell carcinoma": {
+        "name": "Rak płaskonabłonkowy",
+        "risk": "high",
+        "description": "Drugi najczęstszy typ raka skóry",
+        "recommendation": "PILNIE skonsultuj się z onkologiem/dermatologiem"
+    }
+}
+
+def get_risk_indicator(risk_level):
+    """Zwraca kolorowy wskaźnik ryzyka"""
+    if risk_level == "high":
+        return "🔴 WYSOKIE RYZYKO"
+    elif risk_level == "medium":
+        return "🟡 ŚREDNIE RYZYKO"
+    else:
+        return "🟢 NISKIE RYZYKO"
+
 def analyze_skin(image):
     if image is None:
-        return "❌ Proszę wgrać zdjęcie", None
-    
-    # Bezpośrednia analiza zmiany skórnej
-    result = skin_model(image)
-    
-    # Formatuj wynik
-    output = "✅ Wynik klasyfikacji:\n\n"
-    for r in result:
-        output += f"🔹 **{r['label']}** – {r['score']:.2%}\n"
-    
-    return output, image
-
-# Utwórz interfejs Gradio
-demo = gr.Interface(
-    fn=analyze_skin,
-    inputs=gr.Image(type="pil", label="📸 Wgraj zdjęcie zmiany skórnej"),
-    outputs=[
-        gr.Textbox(label="🔎 Wynik analizy", lines=5),
-        gr.Image(label="Przesłane zdjęcie")
-    ],
-    title="🧴 Klasyfikator zmian skórnych (melanoma etc.)",
-    description="Wgraj zdjęcie zmiany skórnej, a aplikacja przeanalizuje rodzaj zmiany. Upewnij się, że zdjęcie przedstawia fragment ludzkiej skóry.",
-    theme="soft",
-    allow_flagging="never"
-)
+        return "❌ Proszę wgrać zdjęcie", None, ""
+    
+    try:
+        # Analiza zmiany skórnej
+        result = skin_model(image)
+        
+        # Formatuj wynik z kolorowymi wskaźnikami
+        output = "# 🔬 WYNIK ANALIZY ZMIANY SKÓRNEJ\n\n"
+        
+        # Główny wynik (najwyższe prawdopodobieństwo)
+        top_result = result[0]
+        top_condition = SKIN_CONDITIONS.get(top_result['label'], {
+            'name': top_result['label'],
+            'risk': 'medium',
+            'description': 'Nieznana zmiana skórna',
+            'recommendation': 'Skonsultuj się z dermatologiem'
+        })
+        
+        risk_indicator = get_risk_indicator(top_condition['risk'])
+        
+        output += f"## 🎯 NAJBARDZIEJ PRAWDOPODOBNE:\n"
+        output += f"**{top_condition['name']}** ({top_result['label']})\n"
+        output += f"**Prawdopodobieństwo:** {top_result['score']:.1%}\n"
+        output += f"**Poziom ryzyka:** {risk_indicator}\n\n"
+        
+        output += f"### 📋 Opis medyczny:\n"
+        output += f"{top_condition['description']}\n\n"
+        
+        output += f"### 💡 Zalecenia:\n"
+        output += f"{top_condition['recommendation']}\n\n"
+        
+        # Wszystkie wyniki
+        output += "## 📊 SZCZEGÓŁOWE PRAWDOPODOBIEŃSTWA:\n\n"
+        for i, r in enumerate(result, 1):
+            condition = SKIN_CONDITIONS.get(r['label'], {'name': r['label'], 'risk': 'medium'})
+            risk_emoji = "🔴" if condition['risk'] == 'high' else "🟡" if condition['risk'] == 'medium' else "🟢"
+            output += f"{i}. {risk_emoji} **{condition['name']}** – {r['score']:.2%}\n"
+        
+        # Ostrzeżenia bezpieczeństwa
+        safety_warning = """
+---
+## ⚠️ WAŻNE OSTRZEŻENIA BEZPIECZEŃSTWA:
+
+🩺 **Ta aplikacja NIE ZASTĘPUJE konsultacji lekarskiej!**
+- Wyniki mają charakter orientacyjny
+- Ostateczną diagnozę może postawić tylko lekarz
+- W przypadku podejrzanych zmian skórnych - skonsultuj się z dermatologiem
+
+🔍 **Zasada ABCDE dla znamion:**
+- **A**symetria
+- **B**rzegi nieregularne  
+- **C**olor niejednolity
+- **D**średnica > 6mm
+- **E**wolucja (zmiany)
+
+📞 **Kiedy natychmiast do lekarza:**
+- Zmiany rosnące, krwawiące, swędzące
+- Nowe znamiona po 30. roku życia
+- Zmiany wyglądu istniejących znamion
+"""
+        
+        return output, image, safety_warning
+        
+    except Exception as e:
+        return f"❌ Błąd podczas analizy: {str(e)}", None, ""
+
+# Utwórz zaawansowany interfejs Gradio
+with gr.Blocks(theme=gr.themes.Soft(), title="🔬 Klasyfikator Raka Skóry") as demo:
+    gr.Markdown("""
+    # 🔬 Klasyfikator Zmian Skórnych - Detekcja Raka Skóry
+    
+    **Zaawansowana analiza AI wykrywająca 8 typów zmian skórnych**
+    
+    Wgraj zdjęcie zmiany skórnej, a aplikacja przeanalizuje:
+    - Czerniak (Melanoma) 🔴
+    - Rak podstawnokomórkowy 🔴  
+    - Rak płaskonabłonkowy 🔴
+    - Rogowacenie słoneczne 🟡
+    - Znamiona i inne łagodne zmiany 🟢
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(
+                type="pil", 
+                label="📸 Wgraj zdjęcie zmiany skórnej",
+                height=400
+            )
+            
+            analyze_btn = gr.Button(
+                "🔍 ANALIZUJ ZMIANĘ SKÓRNĄ", 
+                variant="primary",
+                size="lg"
+            )
+            
+        with gr.Column():
+            result_output = gr.Markdown(
+                label="🔎 Wynik analizy",
+                value="Wgraj zdjęcie i kliknij 'Analizuj' aby rozpocząć"
+            )
+    
+    with gr.Row():
+        uploaded_image = gr.Image(
+            label="📷 Przesłane zdjęcie", 
+            height=300
+        )
+        
+    safety_output = gr.Markdown(
+        label="⚠️ Ostrzeżenia bezpieczeństwa"
+    )
+    
+    # Obsługa kliknięcia
+    analyze_btn.click(
+        fn=analyze_skin,
+        inputs=[image_input],
+        outputs=[result_output, uploaded_image, safety_output]
+    )
+    
+    # Przykłady użycia
+    gr.Markdown("""
+    ## 📋 Instrukcje użycia:
+    1. **Wgraj zdjęcie** - najlepiej wysokiej jakości, dobrze oświetlone
+    2. **Kliknij 'Analizuj'** - poczekaj na wynik AI
+    3. **Sprawdź wynik** - poziom ryzyka i prawdopodobieństwa
+    4. **Przeczytaj zalecenia** - czy potrzebna konsultacja lekarska
+    
+    ## 🎯 Aplikacja analizuje 8 typów zmian:
+    - 🔴 **Wysokie ryzyko:** Czerniak, Rak podstawnokomórkowy, Rak płaskonabłonkowy
+    - 🟡 **Średnie ryzyko:** Rogowacenie słoneczne  
+    - 🟢 **Niskie ryzyko:** Znamiona, Włókniak skóry, Rogowacenie łojotokowe
+    """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)