Update app.py

app.py

@@ -657,71 +657,249 @@ Please consult a dermatologist for accurate diagnosis and treatment.</em>
         top_class = next(iter(class_probs))
         top_confidence = class_probs[top_class]
         
-        # Get disease information from database or generate default
-        disease_info = self.disease_info_db.get(top_class.lower(), None)
+        # Get disease information from database - check multiple variations
+        disease_info = None
         
-        # If disease not in database, create default information
+        # Try different case variations to find the disease
+        search_keys = [
+            top_class.lower(),
+            top_class.lower().replace('_', ' '),
+            top_class.lower().replace('-', ' '),
+            top_class.replace('_', ' ').lower(),
+            top_class.replace('-', ' ').lower()
+        ]
+        
+        for key in search_keys:
+            if key in self.disease_info_db:
+                disease_info = self.disease_info_db[key]
+                logger.info(f"Found disease info for: {key}")
+                break
+        
+        # If still not found, create comprehensive default information
         if disease_info is None:
-            disease_info = {
-                "name_ko": self._translate_disease_name(top_class),
-                "severity": self._estimate_severity(top_class),
-                "description_ko": f"{top_class}는 피부 질환의 한 종류로, 전문의 진단이 필요합니다.",
-                "treatment_ko": "정확한 진단을 위해 피부과 전문의와 상담하시기 바랍니다."
-            }
+            logger.warning(f"No database entry for: {top_class}")
+            disease_info = self._create_default_disease_info(top_class)
         
-        # Create comprehensive GPT analysis
+        # Get detailed symptoms if available
+        symptoms_text = disease_info.get('symptoms_ko', '')
+        if symptoms_text:
+            symptoms_html = f"""
+<div class="result-container">
+<h4>🔍 주요 증상 / Key Symptoms</h4>
+<p>{symptoms_text}</p>
+</div>
+"""
+        else:
+            symptoms_html = ""
+        
+        # Create comprehensive GPT analysis with all available information
         analysis = f"""
 <div class="diagnosis-card">
 <h3>🤖 GPT-OSS 상세 분석 / GPT-OSS Detailed Analysis</h3>
 
 <div class="info-box">
-<h4>진단된 질환 정보 / Diagnosed Condition Information</h4>
-<p><strong>질환명 / Disease:</strong> {top_class.title()} ({disease_info.get('name_ko', top_class)})</p>
+<h4>📋 진단된 질환 정보 / Diagnosed Condition Information</h4>
+<p><strong>질환명 / Disease:</strong> {top_class.title()} ({disease_info['name_ko']})</p>
 <p><strong>신뢰도 / Confidence:</strong> {top_confidence:.1%}</p>
-<p><strong>설명 / Description:</strong> {disease_info.get('description_ko', 'AI 분석 중...')}</p>
+<p><strong>중증도 / Severity:</strong> {self._get_severity_badge(disease_info.get('severity', 'unknown'))}</p>
 </div>
 
+<div class="result-container">
+<h4>📖 질환 설명 / Disease Description</h4>
+<p><strong>{disease_info['name_ko']} ({top_class.title()})</strong></p>
+<p>{disease_info['description_ko']}</p>
+</div>
+
+{symptoms_html}
+
 <div class="result-container">
 <h4>📝 권장 치료법 / Recommended Treatment</h4>
-<p>{disease_info.get('treatment_ko', '전문의 상담이 필요합니다.')}</p>
+<p><strong>치료 방법:</strong></p>
+<p>{disease_info['treatment_ko']}</p>
 </div>
 
 <div class="result-container">
 <h4>💊 생활 관리 지침 / Lifestyle Management Guidelines</h4>
-<ul>
-<li>충분한 수분 섭취와 보습 유지 / Maintain adequate hydration and moisturization</li>
-<li>자극적인 화장품 사용 자제 / Avoid irritating cosmetics</li>
-<li>규칙적인 수면과 스트레스 관리 / Regular sleep and stress management</li>
-<li>균형 잡힌 식단 유지 / Maintain a balanced diet</li>
-<li>적절한 자외선 차단 / Adequate sun protection</li>
-</ul>
+{self._get_lifestyle_guidelines(top_class, disease_info.get('severity', 'low'))}
 </div>
 
 <div class="result-container">
 <h4>🏥 의료진 상담 필요 시점 / When to Consult Healthcare Provider</h4>
-<ul>
-<li>증상이 2주 이상 지속될 경우 / If symptoms persist for more than 2 weeks</li>
-<li>통증이나 가려움이 심한 경우 / Severe pain or itching</li>
-<li>병변이 급속히 확산되는 경우 / Rapid spread of lesions</li>
-<li>발열 등 전신 증상이 동반되는 경우 / Accompanied by systemic symptoms like fever</li>
-<li>일상생활에 지장을 줄 정도로 심한 경우 / Severe enough to interfere with daily life</li>
-</ul>
+{self._get_consultation_guidelines(disease_info.get('severity', 'low'))}
 </div>
 
 <div class="result-container">
 <h4>🔬 추가 검사 권장사항 / Recommended Additional Tests</h4>
-<ul>
-<li>피부 조직검사 (필요시) / Skin biopsy (if necessary)</li>
-<li>알레르기 검사 / Allergy testing</li>
-<li>혈액 검사 / Blood tests</li>
-<li>진균 배양 검사 (감염 의심시) / Fungal culture (if infection suspected)</li>
-</ul>
+{self._get_test_recommendations(top_class, disease_info.get('severity', 'low'))}
+</div>
+
+<div class="result-container">
+<h4>⚠️ 주의사항 / Precautions</h4>
+{self._get_precautions(top_class, disease_info.get('severity', 'low'))}
 </div>
 </div>
 """
         
         return analysis
     
+    def _create_default_disease_info(self, disease_name: str) -> Dict:
+        """Create default disease information when not in database."""
+        # Try to infer information from the disease name
+        name_lower = disease_name.lower()
+        
+        # Determine Korean name
+        name_ko = self._translate_disease_name(disease_name)
+        
+        # Determine severity
+        severity = self._estimate_severity(disease_name)
+        
+        # Create appropriate description based on keywords
+        if 'cancer' in name_lower or 'carcinoma' in name_lower or 'melanoma' in name_lower:
+            description = f"악성 종양의 가능성이 있는 피부 병변입니다. 즉시 전문의 진단이 필요합니다."
+            treatment = "즉시 피부과 또는 종양 전문의 상담, 조직검사 필요, 조기 진단과 치료가 중요"
+        elif 'infection' in name_lower or 'bacterial' in name_lower:
+            description = f"세균 감염이 의심되는 피부 질환입니다. 적절한 항생제 치료가 필요할 수 있습니다."
+            treatment = "항생제 치료(국소 또는 경구), 상처 소독, 위생 관리"
+        elif 'fungal' in name_lower:
+            description = f"진균 감염이 의심되는 피부 질환입니다."
+            treatment = "항진균제 치료, 환부 건조 유지, 통풍 개선"
+        elif 'dermatitis' in name_lower or 'eczema' in name_lower:
+            description = f"피부의 염증성 질환으로, 가려움과 발적이 동반될 수 있습니다."
+            treatment = "보습제 사용, 국소 스테로이드, 항히스타민제, 자극 물질 회피"
+        elif 'acne' in name_lower or 'pimple' in name_lower:
+            description = f"모낭과 피지선의 염증성 질환입니다."
+            treatment = "적절한 클렌징, 국소 치료제(레티노이드, 벤조일 퍼옥사이드), 항생제"
+        else:
+            description = f"피부 질환으로 정확한 진단을 위해 전문의 상담이 권장됩니다."
+            treatment = "피부과 전문의 상담을 통한 정확한 진단 후 적절한 치료"
+        
+        return {
+            "name_ko": name_ko,
+            "severity": severity,
+            "description_ko": description,
+            "treatment_ko": treatment,
+            "symptoms_ko": "육안 검사와 추가 검사를 통해 정확한 증상 파악 필요"
+        }
+    
+    def _get_severity_badge(self, severity: str) -> str:
+        """Get HTML badge for severity level."""
+        badges = {
+            "none": '<span class="diagnosis-severity">정상 / Normal</span>',
+            "low": '<span class="diagnosis-severity severity-low">경증 / Mild</span>',
+            "medium": '<span class="diagnosis-severity severity-medium">중등도 / Moderate</span>',
+            "high": '<span class="diagnosis-severity severity-high">중증 / Severe</span>',
+            "unknown": '<span class="diagnosis-severity">미분류 / Unclassified</span>'
+        }
+        return badges.get(severity, badges["unknown"])
+    
+    def _get_lifestyle_guidelines(self, disease: str, severity: str) -> str:
+        """Get lifestyle management guidelines based on disease and severity."""
+        guidelines = "<ul>"
+        
+        # General guidelines for all conditions
+        guidelines += "<li>충분한 수분 섭취 (하루 8잔 이상) / Drink plenty of water (8+ glasses daily)</li>"
+        guidelines += "<li>균형 잡힌 식단 유지 / Maintain a balanced diet</li>"
+        guidelines += "<li>충분한 수면 (7-8시간) / Get adequate sleep (7-8 hours)</li>"
+        
+        # Specific guidelines based on disease type
+        disease_lower = disease.lower()
+        
+        if 'acne' in disease_lower or 'pimple' in disease_lower:
+            guidelines += "<li>저자극성 클렌저 사용 / Use gentle, non-comedogenic cleanser</li>"
+            guidelines += "<li>손으로 얼굴 만지지 않기 / Avoid touching face with hands</li>"
+            guidelines += "<li>유제품과 고당분 식품 제한 / Limit dairy and high-sugar foods</li>"
+        
+        if 'eczema' in disease_lower or 'dermatitis' in disease_lower:
+            guidelines += "<li>하루 2회 이상 보습제 사용 / Apply moisturizer at least twice daily</li>"
+            guidelines += "<li>뜨거운 물 샤워 피하기 / Avoid hot water showers</li>"
+            guidelines += "<li>면 소재 의류 착용 / Wear cotton clothing</li>"
+        
+        if 'psoriasis' in disease_lower:
+            guidelines += "<li>스트레스 관리 (요가, 명상) / Manage stress (yoga, meditation)</li>"
+            guidelines += "<li>알코올 섭취 제한 / Limit alcohol consumption</li>"
+            guidelines += "<li>적절한 햇빛 노출 / Moderate sun exposure</li>"
+        
+        if severity == "high":
+            guidelines += "<li>즉시 전문의 상담 예약 / Schedule immediate specialist consultation</li>"
+            guidelines += "<li>처방된 약물 규칙적 복용 / Take prescribed medications regularly</li>"
+        
+        guidelines += "<li>자외선 차단제 매일 사용 (SPF 30+) / Apply sunscreen daily (SPF 30+)</li>"
+        guidelines += "</ul>"
+        
+        return guidelines
+    
+    def _get_consultation_guidelines(self, severity: str) -> str:
+        """Get consultation guidelines based on severity."""
+        guidelines = "<ul>"
+        
+        if severity == "high":
+            guidelines += "<li><strong style='color: red;'>즉시 의료진 상담 필요 / Immediate medical consultation required</strong></li>"
+            guidelines += "<li>24시간 이내 피부과 방문 권장 / Visit dermatologist within 24 hours</li>"
+        
+        guidelines += "<li>증상이 2주 이상 지속될 경우 / If symptoms persist for more than 2 weeks</li>"
+        guidelines += "<li>통증, 출혈, 분비물이 있을 경우 / If pain, bleeding, or discharge occurs</li>"
+        guidelines += "<li>병변이 급속히 확산되는 경우 / If lesions spread rapidly</li>"
+        guidelines += "<li>발열, 오한 등 전신 증상 동반 시 / If accompanied by fever or chills</li>"
+        guidelines += "<li>일상생활에 지장을 줄 정도로 심한 경우 / If severe enough to affect daily life</li>"
+        guidelines += "<li>기존 치료에 반응하지 않는 경우 / If not responding to current treatment</li>"
+        guidelines += "</ul>"
+        
+        return guidelines
+    
+    def _get_test_recommendations(self, disease: str, severity: str) -> str:
+        """Get test recommendations based on disease type."""
+        tests = "<ul>"
+        
+        disease_lower = disease.lower()
+        
+        # Cancer-related tests
+        if 'cancer' in disease_lower or 'carcinoma' in disease_lower or 'melanoma' in disease_lower:
+            tests += "<li>피부 조직검사 (Biopsy) - <strong>필수</strong></li>"
+            tests += "<li>더모스코피 검사 / Dermoscopy examination</li>"
+            tests += "<li>전신 피부 검진 / Full body skin examination</li>"
+            tests += "<li>림프절 검사 / Lymph node examination</li>"
+        
+        # Infection tests
+        elif 'infection' in disease_lower or 'bacterial' in disease_lower or 'fungal' in disease_lower:
+            tests += "<li>세균 배양 검사 / Bacterial culture</li>"
+            tests += "<li>진균 검사 (KOH test) / Fungal examination</li>"
+            tests += "<li>항생제 감수성 검사 / Antibiotic sensitivity test</li>"
+        
+        # Allergy/Dermatitis tests
+        elif 'dermatitis' in disease_lower or 'eczema' in disease_lower:
+            tests += "<li>패치 테스트 (알레르기 검사) / Patch testing</li>"
+            tests += "<li>혈액 IgE 검사 / Blood IgE test</li>"
+            tests += "<li>피부 단자 검사 / Skin prick test</li>"
+        
+        # General tests
+        else:
+            tests += "<li>피부 확대경 검사 / Magnified skin examination</li>"
+            tests += "<li>Wood's lamp 검사 (필요시) / Wood's lamp examination (if needed)</li>"
+        
+        if severity == "high":
+            tests += "<li>혈액 검사 (CBC, CRP) / Blood tests (CBC, CRP)</li>"
+            tests += "<li>영상 검사 (필요시) / Imaging studies (if needed)</li>"
+        
+        tests += "</ul>"
+        
+        return tests
+    
+    def _get_precautions(self, disease: str, severity: str) -> str:
+        """Get precautions based on disease type."""
+        precautions = "<ul>"
+        
+        if severity == "high":
+            precautions += "<li><strong>자가 치료 금지 - 반드시 전문의 상담</strong></li>"
+        
+        precautions += "<li>처방 없이 스테로이드 연고 장기 사용 금지 / Avoid long-term steroid use without prescription</li>"
+        precautions += "<li>병변 부위 긁거나 자극하지 않기 / Do not scratch or irritate affected areas</li>"
+        precautions += "<li>검증되지 않은 민간요법 피하기 / Avoid unverified home remedies</li>"
+        precautions += "<li>타인과 수건, 의류 공유 금지 / Do not share towels or clothing</li>"
+        precautions += "<li>증상 악화 시 즉시 의료진 상담 / Consult immediately if symptoms worsen</li>"
+        precautions += "</ul>"
+        
+        return precautions
+    
     def _translate_disease_name(self, disease_name: str) -> str:
         """Translate disease name to Korean if not in database."""
         # Common translations for diseases not in main database