Update app.py

app.py

@@ -51,6 +51,7 @@ custom_role_conversions=None,
 
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
 @tool
 def image_open(prompt: str) -> None:
     """Prompta göre görüntü üretir ve resmi açar.
@@ -59,21 +60,21 @@ def image_open(prompt: str) -> None:
        prompt:Bir prompt alır(örnek:a cat image)
     
     """
-    # Resmi üret
-    result = image_generation_tool(prompt=prompt)
-
-    # Base64 string al
-    img_base64 = result["images"][0]
-
-    # Base64 → byte
-    img_bytes = base64.b64decode(img_base64)
+    
+    # Tool’u çağır → zaten bir PIL Image döndürüyor
+    img = image_generation_tool(prompt=prompt)
 
-    # Byte → PIL Image
-    img = Image.open(BytesIO(img_bytes))
+    # Eğer tool bir dict gibi dönerse ona göre ele alalım
+    if isinstance(img, dict) and "images" in img:
+        img = img["images"][0]  # ama büyük ihtimalle buna gerek kalmaz
 
-    # resmi aç
-    img.show()
+    # Eğer img gerçekten bir PIL.Image ise:
+    if hasattr(img, "show"):
+        img.show()
+        return None
 
+    # Eğer hâlâ değilse güvenli bir hata mesajı
+    print("image_generation_tool beklenmeyen bir format döndürdü:", type(img))
     return None