Update app.py

app.py

@@ -8,9 +8,14 @@ import torch, math, pandas as pd
 # ==============================
 # 📦 بارگذاری مدل (یکبار کش میشه)
 # ==============================
-model_id = "prithivMLmods/Minc-Materials-23"
-processor = AutoImageProcessor.from_pretrained(model_id)
-model = AutoModelForImageClassification.from_pretrained(model_id)
+@torch.no_grad()
+def load_model():
+    model_id = "prithivMLmods/Minc-Materials-23"
+    processor = AutoImageProcessor.from_pretrained(model_id)
+    model = AutoModelForImageClassification.from_pretrained(model_id)
+    return processor, model
+
+processor, model = load_model()
 
 # ==============================
 # 📊 پارامترهای مصالح
@@ -34,18 +39,6 @@ material_params = {
     "sky":{"alpha":1.0,"eps":1.0,"I":0},
 }
 
-# دسته‌ها و جایگزین‌ها (خلاصه)
-replacement_text = {
-    "facade": {"brick":"آجر روشن یا نمای سرامیکی روشن","stone":"سنگ روشن یا نمای گچی بازتابی"},
-    "glazing": {"glass":"شیشه دوجداره Low-E","mirror":"شیشه مات یا بازتاب متعادل"},
-    "metallic": {"metal":"آلومینیوم رنگ روشن"},
-    "coverings": {"plastic":"چوب روشن یا سنگ سبک"},
-    "wood_elements": {"wood":"چوب روشن با پوشش بازتابی"},
-    "vegetation": {"foliage":None},
-    "water_bodies": {"water":None},
-    "background": {"sky":None}
-}
-
 material_categories = {
     "facade":["brick","stone","polishedstone","concrete","tile","ceramic","painted"],
     "glazing":["glass","mirror"],
@@ -57,6 +50,23 @@ material_categories = {
     "background":["sky"]
 }
 
+replacement_text = {
+    "facade": {
+        "brick":"آجر روشن یا نمای سرامیکی روشن",
+        "stone":"سنگ روشن یا نمای گچی بازتابی",
+        "tile":"کاشی/سرامیک روشن یا متخلخل",
+        "ceramic":"سرامیک روشن با پوشش بازتابی",
+        "painted":"رنگ بازتابی (cool paint)"
+    },
+    "glazing": {"glass":"شیشه دوجداره Low-E","mirror":"شیشه مات یا بازتاب متعادل"},
+    "metallic":{"metal":"آلومینیوم رنگ روشن"},
+    "coverings":{"plastic":"چوب روشن یا سنگ سبک","paper":"جایگزینی طراحی","fabric":"پارچه بازتابی"},
+    "wood_elements":{"wood":"چوب روشن با پوشش بازتابی"},
+    "vegetation":{"foliage":None},
+    "water_bodies":{"water":None},
+    "background":{"sky":None}
+}
+
 # ==============================
 # محاسبات
 # ==============================
@@ -94,6 +104,7 @@ def get_patches(image,size=224,stride=200):
 def analyze(img,T_air,RH=40,u=2,S=700):
     img=img.convert("RGB")
     patches=get_patches(img)
+    if len(patches)==0: return "⛔ تصویر نامعتبر است یا کوچک است."
 
     all_predictions=[]
     for patch in patches:
@@ -108,13 +119,16 @@ def analyze(img,T_air,RH=40,u=2,S=700):
     total_patches=len(patches)
     materials_found={m for m,c in counter.items() if c>=3 and m in material_params}
 
+    if not materials_found:
+        return "⛔ هیچ مصالح معتبری شناسایی نشد."
+
     results=[]
     for m in materials_found:
         share=counter[m]/total_patches
         dT=calc_deltaT(m,T_air,RH,u,S)
-        results.append(f"{m} | سهم={share*100:.1f}% | ΔT={dT:+.3f}°C")
-
-    if not results: return "⛔ هیچ مصالح معتبری شناسایی نشد."
+        category = next((cat for cat, members in material_categories.items() if m in members), None)
+        suggestion = replacement_text.get(category, {}).get(m, "نیازی به جایگزینی ندارد")
+        results.append(f"{m} | سهم={share*100:.1f}% | ΔT={dT:+.3f}°C | پیشنهاد: {suggestion}")
 
     scene_deltaT=sum((counter[m]/total_patches)*calc_deltaT(m,T_air,RH,u,S) for m in materials_found)
     summary=f"📌 ΔT میانگین وزنی: {scene_deltaT:+.2f} °C\n📌 دمای مؤثر سطح: {T_air+scene_deltaT:.2f} °C"
@@ -132,8 +146,11 @@ demo = gr.Interface(
         gr.Number(label="💨 سرعت باد (m/s)",value=2),
         gr.Number(label="☀️ تابش خورشیدی (W/m²)",value=700)
     ],
-    outputs=gr.Textbox(label="نتایج تحلیل")
+    outputs=gr.Textbox(label="نتایج تحلیل"),
+    title="🏗️ تحلیل مصالح و توصیه اقلیمی",
+    description="این اپلیکیشن مصالح را از تصویر شناسایی کرده و ΔT سطح و پیشنهادهای جایگزینی را نمایش می‌دهد.",
+    allow_flagging="never"
 )
 
 if __name__=="__main__":
-    demo.launch()
+    demo.launch(share=True)  # share=True لینک عمومی موقت می‌دهد