Update app.py

app.py

@@ -10,9 +10,23 @@ from gradio.themes import Soft
 from gradio.themes.utils import colors, fonts, sizes
 from deep_translator import GoogleTranslator
 
-# --- تعریف متغیرها ---
+# --- تعریف تم ---
+colors.steel_blue = colors.Color(
+    name="steel_blue",
+    c50="#EBF3F8",
+    c100="#D3E5F0",
+    c200="#A8CCE1",
+    c300="#7DB3D2",
+    c400="#529AC3",
+    c500="#4682B4",
+    c600="#3E72A0",
+    c700="#36638C",
+    c800="#2E5378",
+    c900="#264364",
+    c950="#1E3450",
+)
+
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-dtype = torch.bfloat16
 
 # --- بارگذاری مدل ---
 from diffusers import FlowMatchEulerDiscreteScheduler
@@ -20,6 +34,9 @@ from qwenimage.pipeline_qwenimage_edit_plus import QwenImageEditPlusPipeline
 from qwenimage.transformer_qwenimage import QwenImageTransformer2DModel
 from qwenimage.qwen_fa3_processor import QwenDoubleStreamAttnProcessorFA3
 
+dtype = torch.bfloat16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 print("Loading pipeline...")
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "Qwen/Qwen-Image-Edit-2509",
@@ -33,27 +50,35 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
 ).to(device)
 
 # بارگذاری LoRA ها
-loras = {
-    "anime": ("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime", "Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors"),
-    "multiple-angles": ("dx8152/Qwen-Edit-2509-Multiple-angles", "镜头转换.safetensors"),
-    "light-restoration": ("dx8152/Qwen-Image-Edit-2509-Light_restoration", "移除光影.safetensors"),
-    "relight": ("dx8152/Qwen-Image-Edit-2509-Relight", "Qwen-Edit-Relight.safetensors"),
-    "multi-angle-lighting": ("dx8152/Qwen-Edit-2509-Multi-Angle-Lighting", "多角度灯光-251116.safetensors"),
-    "edit-skin": ("tlennon-ie/qwen-edit-skin", "qwen-edit-skin_1.1_000002750.safetensors"),
-    "next-scene": ("lovis93/next-scene-qwen-image-lora-2509", "next-scene_lora-v2-3000.safetensors"),
-    "upscale-image": ("vafipas663/Qwen-Edit-2509-Upscale-LoRA", "qwen-edit-enhance_64-v3_000001000.safetensors")
-}
-
-for name, (repo, weight) in loras.items():
-    try:
-        pipe.load_lora_weights(repo, weight_name=weight, adapter_name=name)
-    except Exception as e:
-        print(f"Error loading LoRA {name}: {e}")
+pipe.load_lora_weights("autoweeb/Qwen-Image-Edit-2509-Photo-to-Anime",
+                       weight_name="Qwen-Image-Edit-2509-Photo-to-Anime_000001000.safetensors",
+                       adapter_name="anime")
+pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multiple-angles",
+                       weight_name="镜头转换.safetensors",
+                       adapter_name="multiple-angles")
+pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Light_restoration",
+                       weight_name="移除光影.safetensors",
+                       adapter_name="light-restoration")
+pipe.load_lora_weights("dx8152/Qwen-Image-Edit-2509-Relight",
+                       weight_name="Qwen-Edit-Relight.safetensors",
+                       adapter_name="relight")
+pipe.load_lora_weights("dx8152/Qwen-Edit-2509-Multi-Angle-Lighting",
+                       weight_name="多角度灯光-251116.safetensors",
+                       adapter_name="multi-angle-lighting")
+pipe.load_lora_weights("tlennon-ie/qwen-edit-skin",
+                       weight_name="qwen-edit-skin_1.1_000002750.safetensors",
+                       adapter_name="edit-skin")
+pipe.load_lora_weights("lovis93/next-scene-qwen-image-lora-2509",
+                       weight_name="next-scene_lora-v2-3000.safetensors",
+                       adapter_name="next-scene")
+pipe.load_lora_weights("vafipas663/Qwen-Edit-2509-Upscale-LoRA",
+                       weight_name="qwen-edit-enhance_64-v3_000001000.safetensors",
+                       adapter_name="upscale-image")
 
 pipe.transformer.set_attn_processor(QwenDoubleStreamAttnProcessorFA3())
 MAX_SEED = np.iinfo(np.int32).max
 
-# --- نگاشت نام‌های فارسی ---
+# --- نگاشت نام‌های فارسی به نام‌های داخلی مدل ---
 LORA_MAPPING = {
     "تبدیل عکس به انیمه": "anime",
     "تغییر زاویه دید": "multiple-angles",
@@ -65,47 +90,72 @@ LORA_MAPPING = {
     "افزایش کیفیت (Upscale)": "upscale-image"
 }
 
-# --- توابع کمکی ---
 def translate_prompt(text):
-    if not text: return ""
+    """ترجمه متن فارسی به انگلیسی"""
+    if not text:
+        return ""
     try:
-        return GoogleTranslator(source='auto', target='en').translate(text)
-    except:
+        translated = GoogleTranslator(source='auto', target='en').translate(text)
+        return translated
+    except Exception as e:
+        print(f"Translation Error: {e}")
         return text
 
 def update_dimensions_on_upload(image):
-    if image is None: return 1024, 1024
-    w, h = image.size
-    if w > h:
-        new_w = 1024
-        new_h = int(new_w * (h / w))
+    if image is None:
+        return 1024, 1024
+    
+    original_width, original_height = image.size
+    
+    if original_width > original_height:
+        new_width = 1024
+        aspect_ratio = original_height / original_width
+        new_height = int(new_width * aspect_ratio)
     else:
-        new_h = 1024
-        new_w = int(new_h * (w / h))
-    return (new_w // 8) * 8, (new_h // 8) * 8
+        new_height = 1024
+        aspect_ratio = original_width / original_height
+        new_width = int(new_height * aspect_ratio)
+        
+    new_width = (new_width // 8) * 8
+    new_height = (new_height // 8) * 8
+    
+    return new_width, new_height
 
 @spaces.GPU(duration=30)
-def infer(input_image, prompt, lora_adapter_persian, seed, randomize_seed, guidance_scale, steps, progress=gr.Progress(track_tqdm=True)):
+def infer(
+    input_image,
+    prompt,
+    lora_adapter_persian,
+    seed,
+    randomize_seed,
+    guidance_scale,
+    steps,
+    progress=gr.Progress(track_tqdm=True)
+):
     if input_image is None:
-        raise gr.Error("لطفاً تصویر را بارگذاری کنید.")
+        raise gr.Error("لطفاً یک تصویر برای ویرایش بارگذاری کنید.")
 
     english_prompt = translate_prompt(prompt)
-    adapter_name = LORA_MAPPING.get(lora_adapter_persian)
-    if adapter_name:
-        pipe.set_adapters([adapter_name], adapter_weights=[1.0])
+    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)
-    w, h = update_dimensions_on_upload(input_image)
+    negative_prompt = "worst quality, low quality, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry"
+
+    original_image = input_image.convert("RGB")
+    width, height = update_dimensions_on_upload(original_image)
 
     result = pipe(
-        image=input_image.convert("RGB"),
+        image=original_image,
         prompt=english_prompt,
-        negative_prompt="worst quality, low quality, blurry, bad anatomy",
-        height=h,
-        width=w,
+        negative_prompt=negative_prompt,
+        height=height,
+        width=width,
         num_inference_steps=steps,
         generator=generator,
         true_cfg_scale=guidance_scale,
@@ -113,205 +163,313 @@ def infer(input_image, prompt, lora_adapter_persian, seed, randomize_seed, guida
 
     return result, seed
 
+@spaces.GPU(duration=30)
 def infer_example(input_image, prompt, lora_adapter):
-    return infer(input_image, prompt, lora_adapter, 0, True, 1.0, 4)
+    input_pil = input_image.convert("RGB")
+    guidance_scale = 1.0
+    steps = 4
+    result, seed = infer(input_pil, prompt, lora_adapter, 0, True, guidance_scale, steps)
+    return result, seed
 
-# --- کدهای سمت کلاینت (JS/CSS) ---
 
+# --- جاوااسکریپت برای دکمه دانلود (PostMessage) ---
 js_post_message = """
 async (image) => {
-    if (!image) { alert("تصویر هنوز ساخته نشده است!"); return; }
+    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 }, '*');
+    console.log("Sending download request for:", fileUrl);
+    window.parent.postMessage({
+        type: 'DOWNLOAD_REQUEST',
+        url: fileUrl
+    }, '*');
 }
 """
 
-html_style_and_script = """
+# --- تنظیمات HTML (CSS زیبا و مدرن + JS اجبار تم روشن) ---
+html_code = """
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Vazirmatn:wght@300;400;500;700&display=swap');
 
-/* --- تنظیمات کلی و اجبار تم روشن --- */
+/* --- 1. تنظیمات کلی و اجبار تم روشن --- */
 :root, .dark, body, .gradio-container {
-    --body-background-fill: #f0f2f5 !important;
+    --body-background-fill: #f5f7fa !important;
     --body-text-color: #1f2937 !important;
     --background-fill-primary: #ffffff !important;
+    --background-fill-secondary: #f3f4f6 !important;
     --border-color-primary: #e5e7eb !important;
+    --block-background-fill: #ffffff !important;
     --block-label-text-color: #374151 !important;
     --block-title-text-color: #111827 !important;
-    font-family: 'Vazirmatn', sans-serif !important;
+    --input-background-fill: #ffffff !important;
+    color-scheme: light !important;
 }
 
 body {
-    background-color: #f0f2f5 !important;
+    font-family: 'Vazirmatn', sans-serif !important;
+    background-color: #f5f7fa !important;
     margin: 0;
-    padding: 0;
+    padding: 10px;
 }
 
-/* --- کانتینر اصلی --- */
+/* --- 2. کانتینر اصلی --- */
 #col-container {
-    margin: 20px auto;
-    max-width: 1000px;
-    background: #ffffff !important;
-    border-radius: 24px;
-    padding: 30px;
-    box-shadow: 0 10px 40px rgba(0,0,0,0.06);
-    border: 1px solid #ffffff;
+    margin: 0 auto;
+    max-width: 980px;
     direction: rtl;
     text-align: right;
+    padding: 30px;
+    background: #ffffff !important;
+    border-radius: 24px;
+    box-shadow: 0 10px 40px -10px rgba(0,0,0,0.08); 
+    border: 1px solid rgba(255,255,255,0.8);
 }
 
-/* --- هدر و توضیحات --- */
+/* --- 3. هدر و توضیحات --- */
 #main-title h1 {
-    font-weight: 800 !important;
-    color: #111827 !important;
+    font-size: 2.4em !important;
     text-align: center;
-    margin-bottom: 10px;
-    font-size: 2rem !important;
+    color: #1a202c !important;
+    margin-bottom: 15px;
+    font-weight: 800;
+    background: -webkit-linear-gradient(45deg, #2563eb, #1e40af);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
 }
+
 #main-description {
     text-align: center;
-    color: #6b7280 !important;
-    margin-bottom: 30px;
-    font-size: 1.1rem;
+    font-size: 1.15em;
+    color: #4b5563 !important;
+    margin-bottom: 40px;
+    line-height: 1.6;
 }
 
-/* --- دکمه‌ها --- */
-button.primary {
+/* --- 4. استایل ورودی‌ها و لیبل‌ها --- */
+.gr-input-label, span.label-wrap, label span {
+    font-weight: 700 !important;
+    color: #374151 !important;
+    font-size: 0.95em !important;
+    margin-bottom: 8px !important;
+}
+
+textarea, input[type="text"] {
+    border: 2px solid #e2e8f0 !important;
+    border-radius: 12px !important;
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    padding: 12px !important;
+    font-family: 'Vazirmatn', sans-serif !important;
+}
+
+textarea:focus, input[type="text"]:focus {
+    border-color: #3b82f6 !important;
+    box-shadow: 0 0 0 4px rgba(59, 130, 246, 0.1) !important;
+    outline: none;
+}
+
+.gr-dropdown {
+    background: #ffffff !important;
+    border-radius: 12px !important;
+}
+
+/* --- 5. دکمه‌ها --- */
+.primary-btn, button.primary {
     background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
-    color: white !important;
     border: none !important;
-    border-radius: 12px !important;
-    padding: 12px 20px !important;
+    color: white !important;
     font-weight: 700 !important;
-    font-size: 1rem !important;
+    font-size: 1.1em !important;
+    padding: 14px 28px !important;
+    border-radius: 14px !important;
     box-shadow: 0 4px 15px rgba(16, 185, 129, 0.3) !important;
-    transition: transform 0.2s, box-shadow 0.2s !important;
+    transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1) !important;
+    cursor: pointer !important;
+    width: 100%;
+    margin-top: 15px;
 }
-button.primary:hover {
+
+.primary-btn:hover, button.primary:hover {
     transform: translateY(-2px);
-    box-shadow: 0 6px 20px rgba(16, 185, 129, 0.4) !important;
+    box-shadow: 0 8px 25px rgba(16, 185, 129, 0.45) !important;
 }
+
+.primary-btn:active, button.primary:active {
+    transform: translateY(1px);
+}
+
 #download-btn {
     background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%) !important;
     box-shadow: 0 4px 15px rgba(59, 130, 246, 0.3) !important;
-    margin-top: 15px;
-    width: 100%;
 }
-
-/* --- ورودی‌ها و لیبل‌ها --- */
-.gr-input-label {
-    font-weight: 600;
-    color: #374151 !important;
-    margin-bottom: 8px;
-}
-textarea, input[type="text"], .gr-dropdown {
-    background-color: #f9fafb !important;
-    border: 1px solid #e5e7eb !important;
-    border-radius: 10px !important;
-    color: #111827 !important;
-    box-shadow: none !important;
-}
-textarea:focus, input:focus {
-    border-color: #10b981 !important;
-    background-color: #ffffff !important;
-    box-shadow: 0 0 0 3px rgba(16, 185, 129, 0.1) !important;
+#download-btn:hover {
+    box-shadow: 0 8px 25px rgba(59, 130, 246, 0.45) !important;
 }
 
-/* --- رفع مشکل جدول نمونه‌ها --- */
-.gr-samples-table {
+/* --- 6. بخش نمونه‌ها (Examples) - رفع کامل سیاهی --- */
+.gradio-container .prose table, 
+.gradio-container table {
+    background-color: #ffffff !important;
+    color: #111827 !important;
+    border: 1px solid #e5e7eb !important;
     border-radius: 12px !important;
     overflow: hidden !important;
-    border: 1px solid #e5e7eb !important;
+    width: 100% !important;
+    margin-top: 20px !important;
+}
+
+.gradio-container thead th {
+    background-color: #f3f4f6 !important;
+    color: #374151 !important;
+    font-weight: 700 !important;
+    border-bottom: 2px solid #e5e7eb !important;
+    padding: 12px !important;
+    text-align: right !important;
 }
-.gr-samples-table tr, .gr-samples-table th, .gr-samples-table td {
+
+.gradio-container tbody tr {
     background-color: #ffffff !important;
-    color: #1f2937 !important;
     border-bottom: 1px solid #f3f4f6 !important;
 }
-.gr-samples-table tr:nth-child(even) {
-    background-color: #f9fafb !important; /* راه‌راه خیلی ملایم */
+
+.gradio-container tbody tr:hover {
+    background-color: #f9fafb !important;
 }
-.gr-samples-table th {
-    background-color: #f3f4f6 !important;
-    font-weight: 700 !important;
+
+.gradio-container tbody td {
+    background-color: #ffffff !important;
+    color: #374151 !important;
+    padding: 10px !important;
+}
+
+.gradio-container tbody td span, 
+.gradio-container tbody td p,
+.gradio-container tbody td div {
+    color: #374151 !important;
 }
 
-/* --- مخفی کردن اضافات --- */
-footer, .flagging { display: none !important; }
+/* --- 7. مخفی‌سازی‌ها --- */
+footer { display: none !important; }
+.flagging { display: none !important; }
+
+@media (prefers-color-scheme: dark) {
+    body, .gradio-container, .prose, table, tr, td, th {
+        background-color: #ffffff !important;
+        color: #333333 !important;
+    }
+}
 </style>
 
 <script>
-// اسکریپت اجباری برای حذف کلاس Dark از بدنه اصلی گرادیو
-function forceLight() {
-    document.body.classList.remove('dark');
-    document.body.style.backgroundColor = '#f0f2f5';
+function forceLightMode() {
+    const body = document.querySelector('body');
+    const gradioApp = document.querySelector('gradio-app');
+    
+    if (body) {
+        body.classList.remove('dark');
+        body.style.backgroundColor = '#f5f7fa';
+        body.style.color = '#333333';
+    }
+    if (gradioApp) {
+        gradioApp.classList.remove('dark');
+    }
+    document.querySelectorAll('.dark').forEach(el => {
+        el.classList.remove('dark');
+    });
 }
-document.addEventListener('DOMContentLoaded', forceLight);
-const observer = new MutationObserver(() => forceLight());
-observer.observe(document.body, { attributes: true });
+forceLightMode();
+setInterval(forceLightMode, 1000);
+document.addEventListener('DOMContentLoaded', forceLightMode);
 </script>
 """
 
-# --- رابط کاربری Gradio ---
 with gr.Blocks() as demo:
-    gr.HTML(html_style_and_script)
-
+    # تزریق HTML (استایل و اسکریپت)
+    gr.HTML(html_code)
+    
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **ویرایشگر هوشمند آلفا**", elem_id="main-title")
-        gr.Markdown("با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", elem_id="main-description")
+        gr.Markdown(
+            "با هوش مصنوعی آلفا تصاویر تونو به مدل های مختلف ویرایش کنید.", 
+            elem_id="main-description"
+        )
 
         with gr.Row(equal_height=True):
-            # ستون چپ (ورودی)
             with gr.Column():
-                input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=300)
+                input_image = gr.Image(label="بارگذاری تصویر", type="pil", height=320)
+                
                 prompt = gr.Text(
                     label="دستور ویرایش (به فارسی)",
+                    show_label=True,
                     placeholder="مثال: تصویر را به سبک انیمه تبدیل کن...",
-                    rtl=True, lines=3
+                    rtl=True,
+                    lines=3
                 )
-                run_btn = gr.Button("✨ شروع پردازش", variant="primary")
 
-            # ستون راست (خروجی)
+                run_button = gr.Button("✨ شروع پردازش و ساخت تصویر", variant="primary", elem_classes="primary-btn")
+
             with gr.Column():
-                output_image = gr.Image(label="تصویر نهایی", interactive=False, height=350)
-                download_btn = gr.Button("📥 دانلود و ذخیره تصویر", elem_id="download-btn", variant="primary")
+                output_image = gr.Image(label="تصویر نهایی", interactive=False, format="png", height=380)
                 
-                with gr.Row():
-                    adapter = gr.Dropdown(label="سبک ویرایش", choices=list(LORA_MAPPING.keys()), value="تبدیل عکس به انیمه")
+                # دکمه دانلود اختصاصی برای ارسال به Iframe
+                download_button = gr.Button("📥 دانلود و ذخیره تصویر", variant="secondary", elem_id="download-btn", elem_classes="primary-btn")
                 
-                with gr.Accordion("تنظیمات پیشرفته", open=False):
-                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
-                    random_seed = gr.Checkbox(label="Seed تصادفی", value=True)
-                    scale = gr.Slider(label="Guidance Scale", 1.0, 10.0, step=0.1, value=1.0)
-                    steps = gr.Slider(label="Steps", 1, 50, step=1, value=4)
-
+                with gr.Row():
+                    lora_adapter = gr.Dropdown(
+                        label="انتخاب سبک ویرایش (LoRA)",
+                        choices=list(LORA_MAPPING.keys()),
+                        value="تبدیل عکس به انیمه"
+                    )
+                with gr.Accordion("تنظیمات پیشرفته", open=False, visible=True):
+                    seed = gr.Slider(label="دانه تصادفی (Seed)", minimum=0, maximum=MAX_SEED, step=1, value=0)
+                    randomize_seed = gr.Checkbox(label="استفاده از Seed تصادفی", value=True)
+                    # اصلاح شده: استفاده از آرگومان‌های با نام
+                    guidance_scale = gr.Slider(label="میزان وفاداری به متن (Guidance Scale)", minimum=1.0, maximum=10.0, step=0.1, value=1.0)
+                    steps = gr.Slider(label="تعداد مراحل پردازش (Steps)", minimum=1, maximum=50, step=1, value=4)
+        
+        # تعریف نمونه‌ها (Examples)
         gr.Examples(
             examples=[
                 ["examples/1.jpg", "تبدیل به انیمه کن.", "تبدیل عکس به انیمه"],
                 ["examples/5.jpg", "سایه‌ها را حذف کن و نورپردازی نرم به تصویر بده.", "اصلاح نور و سایه"],
                 ["examples/4.jpg", "از فیلتر ساعت طلایی با پخش نور ملایم استفاده کن.", "نورپردازی مجدد (Relight)"],
                 ["examples/2.jpeg", "دوربین را ۴۵ درجه به سمت چپ بچرخان.", "تغییر زاویه دید"],
-                ["examples/9.jpg", "سبک سینمایی واقعی با درخشش نرم.", "صحنه بعدی (سینمایی)"],
+                ["examples/7.jpg", "منبع نور را از سمت راست عقب قرار بده.", "نورپردازی چند زاویه‌ای"],
+                ["examples/10.jpeg", "کیفیت تصویر را افزایش بده (Upscale).", "افزایش کیفیت (Upscale)"],
+                ["examples/7.jpg", "منبع نور را از پایین بتابان.", "نورپردازی چند زاویه‌ای"],
+                ["examples/2.jpeg", "زاویه دوربین را به نمای بالا گوشه راست تغییر بده.", "تغییر زاویه دید"],
+                ["examples/9.jpg", "دوربین کمی به جلو حرکت می‌کند در حالی که نور خورشید از میان ابرها می‌تابد و درخششی نرم اطراف شبح شخصیت در مه ایجاد می‌کند. سبک سینمایی واقعی.", "صحنه بعدی (سینمایی)"],
+                ["examples/8.jpg", "جزئیات پوست سوژه را برجسته‌تر و طبیعی‌تر کن.", "روتوش پوست"],
+                ["examples/6.jpg", "دوربین را به نمای پایین به بالا تغییر بده.", "تغییر زاویه دید"],
             ],
-            inputs=[input_image, prompt, adapter],
+            inputs=[input_image, prompt, lora_adapter],
             outputs=[output_image, seed],
             fn=infer_example,
             cache_examples=False,
-            label="نمونه‌های آماده"
+            label="نمونه‌ها (برای تست کلیک کنید)"
         )
 
-    run_btn.click(
+    # اتصال دکمه اجرا
+    run_button.click(
         fn=infer,
-        inputs=[input_image, prompt, adapter, seed, random_seed, scale, steps],
+        inputs=[input_image, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps],
         outputs=[output_image, seed]
     )
-    
-    download_btn.click(None, [output_image], None, js=js_post_message)
+
+    # اتصال دکمه دانلود
+    download_button.click(
+        fn=None,
+        inputs=[output_image],
+        outputs=None,
+        js=js_post_message
+    )
 
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(show_error=True)