Update app.py

app.py

@@ -226,4 +226,73 @@ def process_image(input_image):
     except Exception as e:
         return None, None, f"❌ خطا: {str(e)}"
 
-# ادامه کد Gradio مشابه قبل...
+# ادامه کد Gradio مشابه قبل...
+
+# ایجاد اینترفیس Gradio
+with gr.Blocks(title="CelebAMask-HQ Face Parsing", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 🎭 CelebAMask-HQ Face Parsing Demo
+    **آپلود یک تصویر صورت و دریافت خروجی Face Parsing**
+    
+    این مدل صورت را به 19 بخش مختلف تقسیم می‌کند (پوست، چشم، ابرو، بینی، دهان، مو و ...)
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(
+                label="📷 تصویر ورودی",
+                type="filepath",
+                sources=["upload"],
+                height=300
+            )
+            process_btn = gr.Button("🚀 پردازش تصویر", variant="primary", size="lg")
+            
+            with gr.Accordion("ℹ️ وضعیت برنامه", open=False):
+                status_display = gr.Markdown(f"""
+                **وضعیت:**
+                - 🎯 مدل: {'✅ لود شده' if success else '❌ خطا در لود'}
+                - 💻 دستگاه: `{device}`
+                - 📦 ماژول‌ها: {'✅ ایمپورت شده' if IMPORT_SUCCESS else '❌ خطا در ایمپورت'}
+                - 🗂️ کلاس‌ها: {len(CELEBA_CLASSES)}
+                """)
+        
+        with gr.Column():
+            output_blended = gr.Image(
+                label="🎨 نتیجه ترکیبی (تصویر + ماسک)",
+                height=300
+            )
+            output_mask = gr.Image(
+                label="🎭 ماسک سگمنتیشن",
+                height=300
+            )
+    
+    with gr.Row():
+        info_output = gr.Textbox(
+            label="📊 اطلاعات پردازش",
+            lines=3,
+            max_lines=6
+        )
+    
+    with gr.Row():
+        gr.HTML(create_legend())
+    
+    # اتصال رویدادها
+    process_btn.click(
+        fn=process_image,
+        inputs=[input_image],
+        outputs=[output_blended, output_mask, info_output]
+    )
+    
+    input_image.upload(
+        fn=process_image,
+        inputs=[input_image],
+        outputs=[output_blended, output_mask, info_output]
+    )
+
+if __name__ == "__main__":
+    print("🚀 Starting Face Parsing Application...")
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )