Update app.py

app.py

@@ -36,7 +36,7 @@ def generate_image(
     from huggingface_hub import InferenceClient
     from PIL import Image
     import io
-    import os, tempfile, sys
+    import os, sys
     
     # 🔍 Отладка
     print(f"🔍 generate_image: prompt='{prompt[:50]}...', size={width}x{height}", file=sys.stderr)
@@ -65,15 +65,11 @@ def generate_image(
             # Попытка обработать как поток байтов
             image = Image.open(io.BytesIO(result))
         
-        # 🔑 Валидация
-        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
-            image.save(tmp.name, format="PNG")
-            file_path = tmp.name
+        print(f"✅ generate_image: изображение успешно сгенерировано!", file=sys.stderr)
         
-        print(f"✅ generate_image: сохранено в {file_path}", file=sys.stderr)
-        
-        # 🔑 Возвращаем ПУТЬ как строку (не PIL.Image!)
-        return file_path
+        # 🔑 ВОЗВРАЩАЕМ САМ ОБЪЕКТ PIL.Image!
+        # GradioUI сам поймет, что это картинка, и правильно отобразит её в чате
+        return image
         
     except Exception as e:
         print(f"❌ generate_image exception: {e}", file=sys.stderr)