app.py

# app.py
import gradio as gr
import pickle
import pandas as pd
import numpy as np
import os

# لود مدل (اگه مدل رو آپلود کردی)
if os.path.exists("kaatib_v8_best.pkl"):
    model = pickle.load(open("kaatib_v8_best.pkl", "rb"))
else:
    # اگه مدل نبود، یه پیام بده (موقتاً)
    model = None

def predict(area_type, area, rooms, neighborhood, elevator, parking, warehouse, age):
    if model is None:
        return "مدل در حال لود شدن است، چند ثانیه صبر کنید..."

    if area_type == "متراژ مفید (معمول در دیوار)":
        effective_area = area
    else:
        effective_area = area * 0.87
    
    df = pd.DataFrame([{
        'area': effective_area,
        'log_area': np.log1p(effective_area),
        'rooms': rooms,
        'age': age,
        'is_new': 1 if age <= 5 else 0,
        'neighborhood': neighborhood.strip(),
        'elevator': elevator,
        'parking': parking,
        'warehouse': 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>
    """

css = """
.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);}
"""

with gr.Blocks(css=css, title="کاتب - قیمت آپارتمان تهران ۱۴۰۴") as kaatib:
    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>
    </div>
    """)
    
    area_type = gr.Radio(["متراژ مفید (معمول در دیوار)", "متراژ سندی"], value="متراژ مفید (معمول در دیوار)", label="نوع متراژ وارد شده")
    
    with gr.Row():
        area = gr.Slider(40, 450, value=90, step=1, label="متراژ")
        rooms = gr.Dropdown([1,2,3,4,5,6], value=3, 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="سن بنا (سال)")

    btn = gr.Button("پیش‌بینی قیمت", variant="primary", size="lg")
    output = 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>
    """)

kaatib.launch()