Update app.py

app.py

@@ -3,16 +3,25 @@ import requests
 from PIL import Image
 from io import BytesIO
 import os
+from deep_translator import GoogleTranslator  # برای ترجمه
 
-# آدرس API مدل Stable Diffusion از Hugging Face
+# آدرس API مدل Stable Diffusion در Hugging Face
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
 headers = {
-    "Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"  # توی Secrets ست میشه
+    "Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"
 }
 
 def generate_image(prompt):
+    try:
+        # ترجمه خودکار پرامپت فارسی به انگلیسی
+        translated_prompt = GoogleTranslator(source='auto', target='en').translate(prompt)
+        print(f"🈯 Prompt ترجمه‌شده: {translated_prompt}")
+    except Exception as e:
+        return f"❌ خطا در ترجمه پرامپت: {str(e)}"
+
+    # ارسال درخواست به API
     payload = {
-        "inputs": prompt,
+        "inputs": translated_prompt,
         "options": {"wait_for_model": True}
     }
 
@@ -22,14 +31,15 @@ def generate_image(prompt):
         image = Image.open(BytesIO(response.content))
         return image
     except Exception as e:
-        return f"❌ Error: {str(e)}\n\nRaw content:\n{response.content[:500]}"
+        return f"❌ خطا در پردازش تصویر: {str(e)}\n\n🧾 پاسخ:\n{response.content[:500]}"
 
+# رابط Gradio
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="Prompt", placeholder="e.g. A cat sitting in a cardboard box, black and white sketch..."),
+    inputs=gr.Textbox(label="پرامپت فارسی", placeholder="مثلاً: یک گربه داخل جعبه نشسته است..."),
     outputs="image",
-    title="🎨 Stable Diffusion Image Generator",
-    description="Generate images from text prompts using Stable Diffusion via Hugging Face Inference API."
+    title="🎨 تولید تصویر با پرامپت فارسی + Stable Diffusion",
+    description="پرامپت فارسی بده، ترجمه می‌کنیم و با مدل Stable Diffusion تصویر تولید می‌کنیم. (با استفاده از Hugging Face Inference API)"
 )
 
 iface.launch()