"
return top_bar, art_illustration, verdict_msg, 0
# 이미지 로드 및 변환
if isinstance(image, str):
img = Image.open(image).convert('RGB')
else:
img = Image.fromarray(image).convert('RGB')
# 모델 추론
probs = {}
for name, info in LOADED.items():
tfm = transforms.Compose([
transforms.Resize((info['size'], info['size'])),
transforms.ToTensor(),
transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225]),
])
x = tfm(img).unsqueeze(0)
with torch.no_grad():
probs[name] = torch.sigmoid(info['model'](x)).item()
# 가중 합 계산
total = sum(MY_WEIGHTS.values())
final = sum(probs[k]*w/total for k,w in MY_WEIGHTS.items())
score = int(round(final * 100))
if score >= 50:
top_bar = f"""
당신의 점이 흑색종일 가능성
{score}%
나쁨
"""
art_illustration = """
"""
verdict_msg = "
피부과 방문을 권고드립니다
"
else:
top_bar = f"""
당신의 점이 흑색종일 가능성
{score}%
좋음
"""
art_illustration = """
"""
verdict_msg = "
이상 무! 아주 깨끗하고 안전합니다
"
return top_bar, art_illustration, verdict_msg, score
# --- 네비게이션 ---
def move_s2(): return gr.update(visible=False), gr.update(visible=True)
def move_s3(): return gr.update(visible=False), gr.update(visible=True)
def move_s4(img):
t, a, m, s = run_diagnosis(img)
return gr.update(visible=False), gr.update(visible=True), t, a, m, s
def move_s5():
return gr.update(visible=False), gr.update(visible=True)
def move_s6(): return gr.update(visible=False), gr.update(visible=True)
def back_home(): return gr.update(visible=True), gr.update(visible=False), None
# --- UI ---
with gr.Blocks(css=css, theme=gr.themes.Default()) as demo:
# 화면 1: 홈
with gr.Column(visible=True) as s1:
gr.HTML("
점벼라 세상아!
")
gr.HTML("""
""")
gr.HTML("
스마트폰으로 간편하게 흑색종 위험 판단하기
")
btn_s1 = gr.Button("점 진단하기", variant="primary", size="lg")
# 화면 2: 설문
with gr.Column(visible=False) as s2:
gr.HTML("""
?
""")
gr.Markdown("### 단계 1. 설문 작성")
age = gr.Number(label="당신의 만나이는?", value=24, interactive=True)
history = gr.Radio(["있음", "없음"], label="가족 내 피부암 환자가 있었나요?", value="없음", interactive=True)
change = gr.Radio(["크기가 크게 확장됨", "색상이 얼룩덜룩해짐", "경계선이 뭉개짐", "변화 없음"], label="최근 해당 점의 형태적 변화가 있었습니까?", value="변화 없음", interactive=True)
btn_s2 = gr.Button("다음 단계", variant="primary")
# 화면 3: 촬영
with gr.Column(visible=False) as s3:
gr.HTML("""
z
Z
z
""")
gr.Markdown("### 단계 2. 이미지 촬영")
image_input = gr.Image(sources=["upload", "webcam"], type="filepath", label="카메라로 촬영하거나 사진 선택")
btn_s3 = gr.Button("진단 실행", variant="primary")
# 화면 4: 결과
with gr.Column(visible=False) as s4:
out_top = gr.HTML()
out_anim = gr.HTML()
out_msg = gr.HTML()
score_state = gr.State() # AI 점수 저장용
btn_s4 = gr.Button("다음", variant="primary")
# 화면 5: 정보
with gr.Column(visible=False) as s5:
gr.HTML("""
""")
gr.Markdown("### 흑색종 자가 진단 가이드 (ABCDE)")
gr.Markdown("""
* **A (Asymmetry) 비대칭**: 모양이 동그랗지 않고 한쪽으로 일그러짐
* **B (Border) 불규칙 경계**: 가장자리가 매끄럽지 못함
* **C (Color) 다채로운 색**: 연갈색, 검은색, 붉은색 등이 뒤섞임
* **D (Diameter) 큰 지름**: 직경이 6mm 이상
* **E (Evolution) 지속적 변화**: 크기와 깊이가 계속 변함
---
초기 발견 시 **외과 절제 수술만으로 완치 가능**, 전이 시 **치명적**입니다.
""")
btn_s5 = gr.Button("다음", variant="primary")
# 화면 6: 주의사항
with gr.Column(visible=False) as s6:
gr.HTML("""
""")
gr.Markdown("### 프로그램 사용 주의사항")
gr.Markdown("""
본 시뮬레이터의 수치는 **의학적 진단을 대체할 수 없습니다.**
이상이 있다면 **즉시 피부과를 방문**하세요.
---
**제작자:** jj
""")
btn_s6 = gr.Button("처음 화면으로")
btn_s1.click(move_s2, inputs=None, outputs=[s1, s2])
btn_s2.click(move_s3, inputs=None, outputs=[s2, s3])
btn_s3.click(move_s4, inputs=[image_input], outputs=[s3, s4, out_top, out_anim, out_msg, score_state])
btn_s4.click(move_s5, inputs=None, outputs=[s4, s5])
btn_s5.click(move_s6, inputs=None, outputs=[s5, s6])
btn_s6.click(back_home, inputs=None, outputs=[s1, s6, image_input])
demo.launch(debug=True)