Update app.py

app.py

@@ -1,108 +1,73 @@
-# app.py - نسخه نهایی فیکس‌شده برای HF Spaces (Gradio 4.44)
+# app.py - نسخه نهایی و 100٪ کارکردنی
 import gradio as gr
-import pickle
 import pandas as pd
 import numpy as np
-import os
+from catboost import CatBoostRegressor
 
-# لود مدل (چک دقیق‌تر)
-model_path = "kaatib_v8_best.pkl"
-model_loaded = False
-if os.path.exists(model_path):
-    try:
-        model = pickle.load(open(model_path, "rb"))
-        model_loaded = True
-        print("مدل با موفقیت لود شد! ✅")
-    except Exception as e:
-        print(f"خطا در لود مدل: {e}")
-else:
-    print("فایل مدل پیدا نشد! مطمئن شو kaatib_v8_best.pkl آپلود شده باشه.")
+# لود مدل بهینه
+model = CatBoostRegressor()
+model.load_model("kaatib_v8_optimized.cbm")
 
 def predict(area_type, area, rooms, neighborhood, elevator, parking, warehouse, age):
-    if not model_loaded:
-        return "❌ مدل لود نشد! لطفاً فایل kaatib_v8_best.pkl رو چک کن و Space رو Restart کن."
-
-    if area_type == "متراژ مفید (معمول در دیوار)":
-        effective_area = area
-    else:
-        effective_area = area * 0.87
+    effective_area = area if area_type == "متراژ مفید (معمول در دیوار)" else area * 0.87
     
     df = pd.DataFrame([{
         'area': effective_area,
         'log_area': np.log1p(effective_area),
-        'rooms': rooms,
-        'age': age,
+        'rooms': int(rooms),
+        'age': int(age),
         'is_new': 1 if age <= 5 else 0,
         'neighborhood': neighborhood.strip(),
-        'elevator': elevator,
-        'parking': parking,
-        'warehouse': warehouse
+        'elevator': bool(elevator),
+        'parking': bool(parking),
+        'warehouse': bool(warehouse)
     }])
-    
+
     pred = model.predict(df)[0] / 1_000_000_000
     lower = pred * 0.88
     upper = pred * 1.15
-    
+
     return f"""
-<div style="text-align:center; padding:30px; background: linear-gradient(135deg, #0d2b1f, #1a3d2e); border-radius:18px; border: 3px solid #d4af37; box-shadow: 0 10px 30px rgba(212,175,55,0.2);">
-  <h2 style="color:#d4af37; margin:0; font-size:28px;">پیش‌بینی کاتب</h2>
-  <h1 style="color:white; font-size:52px; margin:15px 0 10px;">{pred:.2f} میلیارد تومان</h1>
-  <p style="color:#b8d5cd; font-size:19px; margin:5px;">رنج واقعی بازار: {lower:.2f} – {upper:.2f} میلیارد</p>
+<div style="text-align:center; padding:40px; background:linear-gradient(135deg,#0d2b1f,#1a3d2e); border-radius:20px; border:4px solid #d4af37; box-shadow:0 15px 40px rgba(212,175,55,0.3);">
+  <h1 style="color:#d4af37; font-size:52px; margin:10px 0;">{pred:.2f} میلیارد تومان</h1>
+  <p style="color:#b8d5cd; font-size:22px;">رنج واقعی بازار: {lower:.2f} – {upper:.2f} میلیارد</p>
 </div>
-    """
-
-# CSS کاملاً inline در HTML (بدون head در Blocks)
-css_html = """
-<style>
-.gradio-container {max-width: 920px !important; margin: 20px auto; background: linear-gradient(135deg, #0a1e15, #0f2b1f); border-radius: 28px; box-shadow: 0 25px 70px rgba(0,0,0,0.9); border: 1px solid #1e3d2f;}
-body {background: #040a07;}
-label, .gr-form label, .gr-radio label, .gr-checkbox label, .gr-slider label, .gr-textbox label, .gr-dropdown label {
-    color: black !important; font-weight: bold !important; background: rgba(255,255,255,0.95) !important; padding: 10px 18px !important; border-radius: 14px !important; box-shadow: 0 4px 12px rgba(0,0,0,0.2);}
-.gr-radio, .gr-checkbox-group {background: white !important; border-radius: 14px !important; padding: 15px !important; border: 2px solid #d4af37;}
-.gr-slider .track, .gr-dropdown {background: white !important; color: black !important;}
-.gr-button {background: linear-gradient(45deg, #d4af37, #b8971a) !important; color: black !important; border: none !important; font-weight: bold !important; font-size: 20px !important; padding: 18px !important; border-radius: 16px !important; box-shadow: 0 8px 25px rgba(212,175,55,0.5);}
-.gr-button:hover {transform: translateY(-4px); box-shadow: 0 15px 35px rgba(212,175,55,0.7);}
-</style>
 """
 
-with gr.Blocks(title="کاتب - قیمت آپارتمان تهران ۱۴۰۴") as kaatib:
-    # CSS رو اول لود کن
-    gr.HTML(css_html)
-    
+with gr.Blocks(title="کاتب - قیمت آپارتمان تهران ۱۴۰۴") as app:
     gr.HTML("""
-    <div style="text-align:center; padding:55px 20px; background: linear-gradient(135deg, #0d2b1f, #1e4d38); border-bottom: 5px solid #d4af37;">
-        <h1 style="font-size:62px; margin:0; color:#d4af37; text-shadow: 0 0 30px #d4af3777;">کاتب</h1>
-        <p style="font-size:30px; color:#b8d5cd; margin:15px 0 5px;">هوش مصنوعی پیش‌بینی قیمت مسکن تهران</p>
-        <p style="font-size:21px; color:#8fb8a2;">۸۱,۰۰۰ آگهی واقعی دیوار ۱۴۰۴</p>
+    <style>
+    .gradio-container {max-width: 960px !important; margin: 20px auto; background: linear-gradient(135deg, #0a1e15, #0f2b1f); border-radius: 30px; box-shadow: 0 30px 80px rgba(0,0,0,0.95); padding: 20px;}
+    body {background: #040a07 !important;}
+    label {color: black !important; font-weight: bold !important; background: white !important; padding: 12px 20px !important; border-radius: 16px !important;}
+    .gr-button {background: linear-gradient(45deg, #d4af37, #b8971a) !important; color: black !important; font-size: 24px !important; padding: 20px !important; border-radius: 20px !important;}
+    </style>
+    <div style="text-align:center; padding:60px 20px; background:linear-gradient(135deg,#0d2b1f,#1e4d38); border-bottom:6px solid #d4af37;">
+        <h1 style="font-size:70px; margin:0; color:#d4af37; text-shadow:0 0 40px #d4af3777;">کاتب</h1>
+        <p style="font-size:34px; color:#b8d5cd;">هوش مصنوعی پیش‌بینی قیمت مسکن تهران</p>
+        <p style="font-size:24px; color:#8fb8a2;">۸۱,۰۰۰ آگهی واقعی دیوار ۱۴۰۴</p>
     </div>
     """)
-    
-    area_type = gr.Radio(["متراژ مفید (معمول در دیوار)", "متراژ سندی"], value="متراژ مفید (معمول در دیوار)", label="نوع متراژ وارد شده")
-    
+
+    area_type = gr.Radio(["متراژ مفید (معمول در دیوار)", "متراژ سندی"], value="متراژ مفید (معمول در دیوار)", label="نوع متراژ")
     with gr.Row():
-        area = gr.Slider(40, 450, value=90, step=1, label="متراژ")
+        area = gr.Slider(40, 450, 90, step=1, label="متراژ")
         rooms = gr.Dropdown([1,2,3,4,5,6], value=3, label="تعداد خواب")
-    
-    neighborhood = gr.Textbox(placeholder="مثل: سعادت آباد، نیاوران، زعفرانیه، جردن، پونک...", label="محله")
-    
+    neighborhood = gr.Textbox(placeholder="مثل: سعادت آباد، پونک، نیاوران، زعفرانیه...", label="محله")
     with gr.Row():
         elevator = gr.Checkbox(label="آسانسور", value=True)
         parking = gr.Checkbox(label="پارکینگ", value=True)
         warehouse = gr.Checkbox(label="انباری", value=True)
-    
-    age = gr.Slider(0, 50, value=8, step=1, label="سن بنا (سال)")
+    age = gr.Slider(0, 50, 8, step=1, label="سن بنا (سال)")
 
-    btn = gr.Button("پیش‌بینی قیمت", variant="primary", size="lg")
-    output = gr.Markdown()
+    gr.Button("پیش‌بینی قیمت").click(predict,
+        inputs=[area_type, area, rooms, neighborhood, elevator, parking, warehouse, age],
+        outputs=gr.Markdown())
 
-    btn.click(predict, inputs=[area_type, area, rooms, neighborhood, elevator, parking, warehouse, age], outputs=output)
-    
     gr.HTML("""
-    <div style="text-align:center; margin:50px 20px 30px; padding:30px; background: rgba(212,175,55,0.12); border-radius:18px; border: 2px dashed #d4af37;">
-        <p style="color:#d4af37; font-size:20px; margin:8px;">ساخته شده با افتخار در ایران 🇮🇷</p>
-        <p style="color:#b8d5cd; font-size:16px;">اولین مدل تماماً فارسی برای بازار مسکن تهران</p>
+    <div style="text-align:center; margin:60px; padding:40px; background:rgba(212,175,55,0.15); border-radius:25px; border:3px dashed #d4af37;">
+        <p style="color:#d4af37; font-size:26px;">ساخته شده با افتخار در ایران</p>
     </div>
     """)
 
-# لانچ ساده
-kaatib.launch()
+app.launch()