Update app.py

app.py

@@ -6,23 +6,23 @@ from PIL import Image
 import torch
 import math
 
-# ============================== (همان پارامترها و توابع قبلی)
+# ============================== پارامترهای مصالح
 material_params = {
-    "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": "آب"},
+    "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},
 }
 
 material_categories = {
@@ -37,25 +37,25 @@ material_categories = {
 }
 
 replacement_text = {
-    "facade": {"brick": "آجر روشن یا نمای سرامیکی/تایل روشن با پوشش بازتابی",
+    "facade": {"brick": "آجر روشن یا نمای سرامیکی/تایل روشن با پوشش بازتابی (cool coating)",
                "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": "حفظ پوشش گیاهی طبیعی"},
-    "water_bodies": {"water": "حفظ منابع آبی"},
+    "coverings": {"plastic": "سنگ سبک یا چوب روکش‌دار روشن (بسته به کاربرد)",
+                  "paper": "در نما کاربرد معمول ندارد - بررسی بهینه‌سازی طراحی",
+                  "fabric": "پارچه با روکش بازتابی یا سایه‌انداز طبیعی"},
+    "wood_elements": {"wood": "چوب رنگ روشن یا چوب با روکش بازتابی/محافظ"},
+    "vegetation": {"foliage": None},
+    "water_bodies": {"water": None},
 }
 
-# ============================== (توابع کمکی)
+# ============================== توابع کمکی
 def ET_proxy(T, RH):
     es = 0.6108 * math.exp((17.27 * T) / (T + 237.3))
     return es * (1 - RH / 100.0)
@@ -72,7 +72,7 @@ def calc_deltaT(material, T_air, RH=40, u=2, S=700):
     gamma = S / max(h_c, 1e-6)
     return gamma * C_m / 1000.0
 
-# ============================== (بارگذاری مدل)
+# ============================== بارگذاری مدل
 model_id = "prithivMLmods/Minc-Materials-23"
 processor = AutoImageProcessor.from_pretrained(model_id)
 model = AutoModelForImageClassification.from_pretrained(model_id)
@@ -93,7 +93,7 @@ def get_patches(image, size=224, stride=100):
                     patches.append(patch)
     return patches
 
-# ============================== (تابع اصلی Gradio)
+# ============================== تابع اصلی Gradio
 def analyze_image(image, T_air=32.0, RH=40, u=2.0, S=700):
     patches = get_patches(image, size=patch_size)
     all_predictions = []
@@ -122,89 +122,65 @@ def analyze_image(image, T_air=32.0, RH=40, u=2.0, S=700):
         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 = []
-    
-    # سرتیتر نتایج
-    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 = {}
+    
+    # محاسبه ΔT برای تمام candidate های هر دسته
     for cat, info in material_categories.items():
+        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)
     
-    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}: در دسته‌های پیش‌تعریف قرار ندارد")
+    # بررسی هر دسته
+    for cat, info in material_categories.items():
+        if len(info["candidates"]) < 2:
             continue
-        
-        # پیدا کردن بهترین جایگزین
-        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
-        
-        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"• برای {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"• {material_name}: از نظر حرارتی بهینه است")
+        # لیست Δ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:
+                continue
+            current_dT = material_info[label]["deltaT"]
+            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}")
+                recommendations.append(
+                    f"{label} ({cat}): ΔT={current_dT:+.2f}°C → جایگزین: {best_candidate} (ΔT={best_dT:+.2f}°C) | بهبود: {improvement:+.2f}°C | اهمیت: {importance} | پیشنهاد: {suggestion_text}"
+                )
+            else:
+                recommendations.append(f"{label}: ΔT={current_dT:+.2f}°C → نیازی به جایگزینی ندارد.")
     
-    # خلاصه نتایج
+    # ΔT میانگین وزنی کل صحنه
     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}")
+    recommendations.append(f"ΔT میانگین وزنی کل صحنه: {scene_deltaT:+.2f}°C")
+    recommendations.append(f"دمای مؤثر سطح: {T_air + scene_deltaT:.2f}°C")
     
     return "\n".join(recommendations)
 
-# ============================== (راه‌اندازی رابط Gradio)
-demo = gr.Interface(
-    fn=analyze,
+# ============================== راه‌اندازی رابط Gradio
+iface = gr.Interface(
+    fn=analyze_image,
     inputs=[
-        gr.Image(type="pil",label="📷 تصویر را بارگذاری کنید"),
-        gr.Number(label="🌡️ دمای هوا (°C)",value=32),
-        gr.Number(label="💧 رطوبت نسبی (%)",value=40),
-        gr.Number(label="💨 سرعت باد (m/s)",value=2),
-        gr.Number(label="☀️ تابش خورشیدی (W/m²)",value=700)
+        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²)")
     ],
-    outputs=gr.Textbox(label="نتایج تحلیل")
+    outputs=gr.Textbox(label="خروجی ΔT و توصیه‌ها"),
+    title="تحلیل مصالح و ΔT سطحی",
+    description="آپلود تصویر ساختمان/محیط → نمایش ΔT مصالح و توصیه جایگزینی منطقی."
 )
 
-if __name__=="__main__":
-    demo.launch()
+iface.launch()