Update app.py

app.py

@@ -5,21 +5,20 @@ from io import BytesIO
 import os
 from deep_translator import GoogleTranslator
 
-# آدرس API مدل Stable Diffusion
+# آدرس صحیح API مدل قابل اجرا در Hugging Face
 API_URL = "https://api-inference.huggingface.co/models/runwayml/stable-diffusion-v1-5"
 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"❌ خطا در ترجمه پرامپت:\n{str(e)}"
 
-    # ساخت بدنه درخواست
     payload = {
         "inputs": translated_prompt,
         "options": {"wait_for_model": True}
@@ -29,28 +28,24 @@ def generate_image(prompt):
         # ارسال درخواست به API
         response = requests.post(API_URL, headers=headers, json=payload)
         content_type = response.headers.get("content-type", "")
-
-        # چاپ در لاگ برای بررسی بیشتر
         print("🔍 Content-Type:", content_type)
-        if "image" not in content_type:
-            print("🧾 پاسخ متنی:", response.text[:1000])
 
-        # بررسی اگر پاسخ تصویر بود
         if "image" in content_type:
             image = Image.open(BytesIO(response.content))
             return image
         else:
-            return f"⚠️ پاسخ دریافتی تصویر نبود یا مدل در حال بارگذاری است:\n\n{response.text[:1000]}"
+            print("🧾 پاسخ متنی:", response.text[:1000])
+            return f"⚠️ پاسخ دریافتی تصویر نبود یا مدل هنوز آماده نیست:\n\n{response.text[:1000]}"
     except Exception as e:
         return f"❌ خطا در ارتباط با API:\n{str(e)}"
 
-# رابط گرافیکی Gradio
+# رابط کاربری Gradio
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.Textbox(label="تصویری که می‌خواهی را شرح بده", placeholder="مثلاً: یک گربه داخل جعبه نشسته است..."),
+    inputs=gr.Textbox(label="توضیحت را بنویس", placeholder="مثلاً: یک روباه در جنگل..."),
     outputs="image",
     title="🎨 تولید تصویر دیجی نورون | www.diginoron.com",
-    description="استفاده از مدل Stable Diffusion با یک توضیح کوتاه، تصویر دلخواهت را بساز"
+    description="توضیحی بنویس تا یک تصویر خلق کنیم! از مدل Stable Diffusion نسخه 1.5 استفاده می‌شود."
 )
 
 iface.launch()