💎
اعتبار رایگان امروز تمام شد
شما از ۵ تصویر رایگان امروز استفاده کردهاید.
برای ساخت تصاویر نامحدود و حرفهای، لطفا نسخه خود را ارتقا دهید.
"""
def get_user_record(fingerprint):
"""خواندن رکورد کاربر از دیتابیس بدون اعمال تغییرات"""
global usage_data_cache
if not fingerprint: return None
# Reload to get fresh data
usage_data_cache = load_usage_data()
today_str = date.today().isoformat()
user_record = usage_data_cache.get(fingerprint)
if not user_record or user_record.get("last_reset") != today_str:
return {"count": 0, "last_reset": today_str}
return user_record
def consume_quota(fingerprint):
"""کسر اعتبار کاربر"""
global usage_data_cache
today_str = date.today().isoformat()
usage_data_cache = load_usage_data()
user_record = usage_data_cache.get(fingerprint)
if not user_record or user_record.get("last_reset") != today_str:
user_record = {"count": 0, "last_reset": today_str}
user_record["count"] += 1
usage_data_cache[fingerprint] = user_record
save_usage_data(usage_data_cache)
return user_record["count"]
# --- تابع بررسی اولیه (بدون GPU) برای نمایش/مخفی کردن دکمهها ---
def check_initial_quota(fingerprint, subscription_status):
"""
این تابع به محض لود شدن صفحه توسط جاوا اسکریپت صدا زده میشود.
اگر اعتبار تمام شده باشد، دکمه ساخت را مخفی و دکمه ارتقا را نشان میدهد.
"""
if not fingerprint:
return gr.update(visible=True), gr.update(visible=False), None
if subscription_status == 'paid':
return gr.update(visible=True), gr.update(visible=False), None
user_record = get_user_record(fingerprint)
current_usage = user_record["count"] if user_record else 0
if current_usage >= USAGE_LIMIT:
# اعتبار تمام شده
return gr.update(visible=False), gr.update(visible=True), get_quota_exceeded_html()
else:
# اعتبار دارد
return gr.update(visible=True), gr.update(visible=False), None
@spaces.GPU(duration=30)
def infer(
input_image,
prompt,
lora_adapter_persian,
seed,
randomize_seed,
guidance_scale,
steps,
aspect_ratio_selection,
custom_width,
custom_height,
fingerprint,
subscription_status,
progress=gr.Progress(track_tqdm=True)
):
# بررسی نهایی اعتبار قبل از شروع پردازش سنگین
if subscription_status != 'paid':
user_record = get_user_record(fingerprint)
if user_record and user_record["count"] >= USAGE_LIMIT:
return (
None, seed, get_quota_exceeded_html(),
gr.update(visible=False), gr.update(visible=True)
)
# --- بررسیهای اولیه (تصویر و متن) قبل از کسر اعتبار ---
if input_image is None:
return None, seed, get_error_html("لطفاً ابتدا یک تصویر بارگذاری کنید."), gr.update(visible=True), gr.update(visible=False)
if is_image_nsfw(input_image):
return None, seed, get_error_html("تصویر ورودی دارای محتوای نامناسب است و پردازش نمیشود."), gr.update(visible=True), gr.update(visible=False)
# ترجمه و بررسی متن (مهم: قبل از کسر اعتبار)
english_prompt = translate_prompt(prompt)
if not check_text_safety(english_prompt):
return None, seed, get_error_html("متن درخواست شامل کلمات غیرمجاز، جنسی یا خشونتآمیز است. لطفاً متن را اصلاح کنید."), gr.update(visible=True), gr.update(visible=False)
# اگر همه چیز امن بود، حالا اعتبار را کسر میکنیم
if subscription_status != 'paid':
consume_quota(fingerprint)
adapter_internal_name = LORA_MAPPING.get(lora_adapter_persian)
if adapter_internal_name:
pipe.set_adapters([adapter_internal_name], adapter_weights=[1.0])
if randomize_seed:
seed = random.randint(0, MAX_SEED)
generator = torch.Generator(device=device).manual_seed(seed)
# اضافه کردن کلمات منفی ایمنی به صورت پیشفرض به پرامپت منفی
safety_negative = "nsfw, nude, naked, porn, sexual, xxx, breast, nipple, genital, vagina, penis, ass, lingerie, bikini, swimwear, underwear, fetish, topless, gore, violence, blood, explicit"
base_negative = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
final_negative_prompt = f"{safety_negative}, {base_negative}"
original_image = input_image.convert("RGB")
selection_value = ASPECT_RATIOS_MAP.get(aspect_ratio_selection)
if selection_value == "Custom":
width = (int(custom_width) // 8) * 8
height = (int(custom_height) // 8) * 8
elif selection_value == "Auto" or selection_value is None:
width, height = update_dimensions_on_upload(original_image)
else:
width, height = selection_value
try:
result = pipe(
image=original_image,
prompt=english_prompt,
negative_prompt=final_negative_prompt,
height=height,
width=width,
num_inference_steps=steps,
generator=generator,
true_cfg_scale=guidance_scale,
).images[0]
# بررسی نهایی تصویر تولید شده
if is_image_nsfw(result):
return None, seed, get_error_html("تصویر تولید شده حاوی محتوای نامناسب بود و حذف شد."), gr.update(visible=True), gr.update(visible=False)
# محاسبه اعتبار باقیمانده برای نمایش
user_record = get_user_record(fingerprint)
remaining = USAGE_LIMIT - user_record["count"] if user_record else 0
success_msg = f"تصویر با موفقیت ویرایش شد."
if subscription_status != 'paid':
success_msg += f" (اعتبار باقیمانده امروز: {remaining})"
# اگر با این درخواست اعتبار تمام شد، دکمهها را آپدیت میکنیم
if subscription_status != 'paid' and remaining <= 0:
return result, seed, get_success_html(success_msg), gr.update(visible=False), gr.update(visible=True)
return result, seed, get_success_html(success_msg), gr.update(visible=True), gr.update(visible=False)
except Exception as e:
error_str = str(e)
# -------------------------------------------------------------
# مدیریت خطای GPU QUOTA
# -------------------------------------------------------------
if "quota" in error_str.lower() or "exceeded" in error_str.lower():
raise e
return None, seed, get_error_html(f"خطا در پردازش: {error_str}"), gr.update(visible=True), gr.update(visible=False)
@spaces.GPU(duration=30)
def infer_example(input_image, prompt, lora_adapter):
# مثالها اعتبار کم نمیکنند
res, s, status, btn1, btn2 = infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4, "خودکار (پیشفرض)", 1024, 1024, "example_user", "paid")
return res, s, status
# --- جاوااسکریپت برای دکمه دانلود ---
js_download_func = """
async (image) => {
if (!image) {
alert("لطفاً ابتدا تصویر را تولید کنید.");
return;
}
let fileUrl = image.url;
if (fileUrl && !fileUrl.startsWith('http')) {
fileUrl = window.location.origin + fileUrl;
} else if (!fileUrl && image.path) {
fileUrl = window.location.origin + "/file=" + image.path;
}
window.parent.postMessage({
type: 'DOWNLOAD_REQUEST',
url: fileUrl
}, '*');
}
"""
# --- جاوااسکریپت برای دکمه ارتقا ---
js_upgrade_func = """
() => {
window.parent.postMessage({ type: 'NAVIGATE_TO_PREMIUM' }, '*');
}
"""
# --- جاوااسکریپت سراسری ---
js_global_content = """
"""
# --- CSS Updated ---
css_code = """
"""
# ادغام CSS و JS
combined_html = css_code + js_global_content
# استفاده از gr.Blocks
with gr.Blocks() as demo:
# تزریق کدها به عنوان HTML
gr.HTML(combined_html)
# اینپوتهای مخفی برای انتقال دیتای کاربر از JS به Python
fingerprint_box = gr.Textbox(elem_id="fingerprint_storage", visible=True)
status_box_input = gr.Textbox(elem_id="status_storage", visible=True)
with gr.Column(elem_id="col-container"):
gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
gr.Markdown(
"با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.",
elem_id="main-description"
)
# محل قرارگیری بج (نشانگر) وضعیت اشتراک
gr.HTML('