Update app.py

app.py

@@ -8,22 +8,21 @@ import math
 
 # ============================== (همان پارامترها و توابع قبلی)
 material_params = {
-    "brick": {"alpha": 0.3, "eps": 0.9, "I": 1600},
-    "stone": {"alpha": 0.25, "eps": 0.92, "I": 2000},
-    "polishedstone": {"alpha": 0.2, "eps": 0.9, "I": 2100},
-    "concrete": {"alpha": 0.35, "eps": 0.9, "I": 1800},
-    "metal": {"alpha": 0.5, "eps": 0.2, "I": 4000},
-    "glass": {"alpha": 0.1, "eps": 0.85, "I": 1500},
-    "wood": {"alpha": 0.35, "eps": 0.9, "I": 800},
-    "tile": {"alpha": 0.4, "eps": 0.9, "I": 1200},
-    "ceramic": {"alpha": 0.45, "eps": 0.92, "I": 1300},
-    "painted": {"alpha": 0.3, "eps": 0.9, "I": 1000},
-    "plastic": {"alpha": 0.1, "eps": 0.95, "I": 800},
-    "paper": {"alpha": 0.6, "eps": 0.95, "I": 500},
-    "mirror": {"alpha": 0.7, "eps": 0.1, "I": 2000},
-    "foliage": {"alpha": 0.25, "eps": 0.98, "I": 900},
-    "water": {"alpha": 0.06, "eps": 0.98, "I": 4200},
-   
+    "brick": {"alpha": 0.3, "eps": 0.9, "I": 1600, "name": "آجر"},
+    "stone": {"alpha": 0.25, "eps": 0.92, "I": 2000, "name": "سنگ"},
+    "polishedstone": {"alpha": 0.2, "eps": 0.9, "I": 2100, "name": "سنگ صیقلی"},
+    "concrete": {"alpha": 0.35, "eps": 0.9, "I": 1800, "name": "بتن"},
+    "metal": {"alpha": 0.5, "eps": 0.2, "I": 4000, "name": "فلز"},
+    "glass": {"alpha": 0.1, "eps": 0.85, "I": 1500, "name": "شیشه"},
+    "wood": {"alpha": 0.35, "eps": 0.9, "I": 800, "name": "چوب"},
+    "tile": {"alpha": 0.4, "eps": 0.9, "I": 1200, "name": "کاشی"},
+    "ceramic": {"alpha": 0.45, "eps": 0.92, "I": 1300, "name": "سرامیک"},
+    "painted": {"alpha": 0.3, "eps": 0.9, "I": 1000, "name": "سطح رنگ‌شده"},
+    "plastic": {"alpha": 0.1, "eps": 0.95, "I": 800, "name": "پلاستیک"},
+    "paper": {"alpha": 0.6, "eps": 0.95, "I": 500, "name": "کاغذ"},
+    "mirror": {"alpha": 0.7, "eps": 0.1, "I": 2000, "name": "آینه"},
+    "foliage": {"alpha": 0.25, "eps": 0.98, "I": 900, "name": "گیاهان"},
+    "water": {"alpha": 0.06, "eps": 0.98, "I": 4200, "name": "آب"},
 }
 
 material_categories = {
@@ -35,27 +34,25 @@ material_categories = {
     "wood_elements": {"members": ["wood"], "candidates": ["wood"]},
     "vegetation": {"members": ["foliage"], "candidates": ["foliage"]},
     "water_bodies": {"members": ["water"], "candidates": ["water"]},
-   
 }
 
 replacement_text = {
-    "facade": {"brick": "آجر روشن یا نمای سرامیکی/تایل روشن با پوشش بازتابی (cool coating)",
+    "facade": {"brick": "آجر روشن یا نمای سرامیکی/تایل روشن با پوشش بازتابی",
                "stone": "سنگ روشن یا سنگ با پوشش بازتابی",
                "polishedstone": "سنگ مات روشن یا سرامیک نما روشن",
-               "concrete": "بتن روشن با پوشش بازتابی یا موزاییک نما روشن",
+               "concrete": "بتن روشن با پوشش بازتابی",
                "tile": "کاشی/سرامیک روشن یا متخلخل",
                "ceramic": "سرامیک روشن با نمای بازتابی",
                "painted": "رنگ بازتابی (cool paint) یا پوشش نانو بازتابی"},
-    "glazing": {"glass": "شیشه دو جداره با پوشش Low-E یا شیشه بازتابی کنترل‌شده",
+    "glazing": {"glass": "شیشه دو جداره با پوشش Low-E",
                 "mirror": "شیشه مات یا شیشه Low-E با فریم عایق"},
     "metallic": {"metal": "آلومینیوم رنگ روشن یا پوشش پودری با بازتاب بالا"},
-    "coverings": {"plastic": "سنگ سبک یا چوب روکش‌دار روشن (بسته به کاربرد)",
-                  "paper": "در نما کاربرد معمول ندارد - بررسی بهینه‌سازی طراحی",
-                  "fabric": "پارچه با روکش بازتابی یا سایه‌انداز طبیعی"},
-    "wood_elements": {"wood": "چوب رنگ روشن یا چوب با روکش بازتابی/محافظ"},
-    "vegetation": {"foliage": None},
-    "water_bodies": {"water": None},
-    
+    "coverings": {"plastic": "سنگ سبک یا چوب روکش‌دار روشن",
+                  "paper": "مواد پوششی بازتابی",
+                  "fabric": "پارچه با روکش بازتابی"},
+    "wood_elements": {"wood": "چوب رنگ روشن یا چوب با روکش بازتابی"},
+    "vegetation": {"foliage": "حفظ پوشش گیاهی طبیعی"},
+    "water_bodies": {"water": "حفظ منابع آبی"},
 }
 
 # ============================== (توابع کمکی)
@@ -112,7 +109,7 @@ def analyze_image(image, T_air=32.0, RH=40, u=2.0, S=700):
     counter = Counter(all_predictions)
     total_patches = len(patches)
     MIN_COUNT = 3
-    ignore_classes = ["food", "sky","skin", "other", "wallpaper", "carpet"]
+    ignore_classes = ["food", "skin", "other", "wallpaper", "carpet"]
     materials_found = {label for label, count in counter.items() if count >= MIN_COUNT and label not in ignore_classes}
     
     if len(materials_found) == 0:
@@ -124,63 +121,93 @@ def analyze_image(image, T_air=32.0, RH=40, u=2.0, S=700):
         share = count / total_patches
         dT = calc_deltaT(label, T_air, RH, u, S)
         material_info[label] = {"count": count, "share": share, "deltaT": dT}
-    # ============================== (مقایسه درون‌دسته‌ای و توصیه)
-IMPROVEMENT_THRESHOLD = 0.02
-SHARE_IMPORTANCE_THRESHOLD = 0.03
-recommendations = []
-candidate_delta_cache = {}
-
-# محاسبه ΔT برای تمام candidate های هر دسته
-for cat, info in material_categories.items():
-    # نادیده گرفتن دسته‌های با کمتر از 2 عضو
-    if len(info["candidates"]) < 2:
-        continue
-    for candidate in info["candidates"]:
-        if candidate not in candidate_delta_cache:
-            candidate_delta_cache[candidate] = calc_deltaT(candidate, T_air, RH, u, S)
-
-# برای هر دسته، بهترین ΔT (خنک‌ترین) را پیدا می‌کنیم
-for cat, info in material_categories.items():
-    # دسته‌های با کمتر از 2 عضو را نادیده می‌گیریم
-    if len(info["candidates"]) < 2:
-        continue
-
-    # ΔT هر candidate
-    cand_list = [(c, candidate_delta_cache[c]) for c in info["candidates"]]
-    cand_list.sort(key=lambda x: x[1])  # کمترین ΔT اول
-
-    best_candidate, best_dT = cand_list[0]  # خنک‌ترین عضو
-    for label in info["members"]:
-        if label not in material_info:
+    
+    # مقایسه درون‌دسته‌ای و توصیه - بخش اصلاح شده
+    IMPROVEMENT_THRESHOLD = 0.02
+    SHARE_IMPORTANCE_THRESHOLD = 0.03
+    recommendations = []
+    
+    # سرتیتر نتایج
+    recommendations.append("📋 نتایج تحلیل مصالح:\n")
+    
+    for label, info in material_info.items():
+        material_name = material_params[label]["name"]
+        share_pct = info["share"] * 100
+        current_dT = info["deltaT"]
+        
+        recommendations.append(f"• {material_name}: سهم={share_pct:.1f}% | ΔT={current_dT:+.2f}°C")
+    
+    recommendations.append("\n💡 توصیه‌های بهینه‌سازی:\n")
+    
+    candidate_delta_cache = {}
+    for cat, info in material_categories.items():
+        for candidate in info["candidates"]:
+            if candidate not in candidate_delta_cache:
+                candidate_delta_cache[candidate] = calc_deltaT(candidate, T_air, RH, u, S)
+    
+    for label, info in material_info.items():
+        material_name = material_params[label]["name"]
+        current_dT = info["deltaT"]
+        share = info["share"]
+        
+        # پیدا کردن دسته
+        found_category = None
+        for cat, cinfo in material_categories.items():
+            if label in cinfo["members"]:
+                found_category = cat
+                break
+        
+        if found_category is None:
+            recommendations.append(f"• {material_name}: در دسته‌های پیش‌تعریف قرار ندارد")
             continue
-        current_dT = material_info[label]["deltaT"]
+        
+        # پیدا کردن بهترین جایگزین
+        candidates = material_categories[found_category]["candidates"]
+        cand_list = [(c, candidate_delta_cache.get(c, calc_deltaT(c, T_air, RH, u, S))) for c in candidates]
+        cand_list.sort(key=lambda x: x[1])
+        
+        best_candidate, best_dT = cand_list[0]
         improvement = current_dT - best_dT
-        share_pct = material_info[label]["share"] * 100
-
-        if improvement >= IMPROVEMENT_THRESHOLD and label != best_candidate:
-            importance = "High" if material_info[label]["share"] >= SHARE_IMPORTANCE_THRESHOLD else "Optional"
-            suggestion_text = replacement_text.get(cat, {}).get(best_candidate, f"Consider replacing with {best_candidate}")
+        
+        if improvement >= IMPROVEMENT_THRESHOLD and best_candidate != label:
+            best_candidate_name = material_params[best_candidate]["name"]
+            importance = "بالا" if share >= SHARE_IMPORTANCE_THRESHOLD else "اختیاری"
+            suggestion = replacement_text.get(found_category, {}).get(best_candidate, "جایگزین بهینه")
+            
             recommendations.append(
-                f"{label} ({cat}): ΔT={current_dT:+.2f}°C → جایگزین: {best_candidate} (ΔT={best_dT:+.2f}°C) | بهبود: {improvement:+.2f}°C | اهمیت: {importance} | پیشنهاد: {suggestion_text}"
+                f"• برای {material_name}:\n"
+                f"  → پیشنهاد: {best_candidate_name} (ΔT={best_dT:+.2f}°C)\n"
+                f"  → بهبود: {improvement:.2f}°C | اهمیت: {importance}\n"
+                f"  → توصیه: {suggestion}"
             )
         else:
-            recommendations.append(f"{label}: ΔT={current_dT:+.2f}°C → نیازی به جایگزینی ن��ارد.")
-
- 
+            recommendations.append(f"• {material_name}: از نظر حرارتی بهینه است")
+    
+    # خلاصه نتایج
+    scene_deltaT = sum([info["share"] * info["deltaT"] for info in material_info.values()])
+    
+    recommendations.append("\n📊 خلاصه نتایج:")
+    recommendations.append(f"• ΔT میانگین وزنی: {scene_deltaT:+.2f}°C")
+    recommendations.append(f"• دمای مؤثر سطح: {T_air + scene_deltaT:.2f}°C")
+    recommendations.append(f"• تعداد پچ‌های تحلیل شده: {total_patches}")
+    
+    return "\n".join(recommendations)
 
 # ============================== (راه‌اندازی رابط Gradio)
 iface = gr.Interface(
     fn=analyze_image,
     inputs=[
-        gr.Image(type="pil", label="آپلود تصویر"),
-        gr.Number(value=32.0, label="دمای هوا T_air (°C)"),
-        gr.Number(value=40, label="رطوبت نسبی RH (%)"),
-        gr.Number(value=2.0, label="سرعت باد u (m/s)"),
-        gr.Number(value=700, label="تابش خورشیدی S (W/m²)")
+        gr.Image(type="pil", label="📷 تصویر نمای ساختمان"),
+        gr.Number(value=32.0, label="🌡️ دمای هوا (°C)"),
+        gr.Number(value=40, label="💧 رطوبت نسبی (%)"),
+        gr.Number(value=2.0, label="💨 سرعت باد (m/s)"),
+        gr.Number(value=700, label="☀️ تابش خورشیدی (W/m²)")
     ],
-    outputs=gr.Textbox(label="خروجی ΔT و توصیه‌ها"),
-    title="تحلیل مصالح و ΔT سطحی",
-    description="آپلود تصویر ساختمان/محیط → نمایش ΔT مصالح و توصیه جایگزینی منطقی."
+    outputs=gr.Textbox(label="نتایج تحلیل", lines=20),
+    title="🏗️ تحلیل مصالح و توصیه اقلیمی",
+    description="این اپلیکیشن مصالح را از تصویر شناسایی کرده و ΔT سطح و پیشنهادهای جایگزینی را نمایش می‌دهد.",
+    allow_flagging="never"
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch(share=True, server_name="0.0.0.0", server_port=7860)