Update app.py

app.py

@@ -158,36 +158,49 @@ THEME_EXAMPLES = {
 def generate_image_prompts(theme: str, game_type: str) -> dict:
     client = get_groq_client()
     if game_type == "Top-Down Shooter":
-        perspective = "top-down aerial view"
-        char_style  = "top-down aerial view sprite, character seen from directly above"
+        perspective = "bird's eye view directly overhead, isometric top-down game art"
+        char_style  = (
+            "bird's eye view directly overhead, isometric top-down game sprite, "
+            "character seen from above looking down, head visible at top, feet at bottom, "
+            "no perspective distortion, flat top-down angle"
+        )
     else:
-        perspective = "2D side-view flat"
-        char_style  = "2D side-view flat sprite, full body visible from the side"
+        perspective = "2D side-scrolling platformer background, horizontal flat landscape"
+        char_style  = (
+            "2D side-view platformer sprite, character facing right, "
+            "full body visible from the side, flat 2D game art style, "
+            "like classic side-scrolling game characters"
+        )
 
     seeds = {
         "sprite_player.png": (
             f"pixel-art game player character, {char_style}, {theme} theme, "
-            f"vibrant colors, character ONLY with NO background, transparent background, "
-            f"clear silhouette, 64x64 pixel style"
+            f"isolated character on pure solid-color background, "
+            f"strong clear silhouette, vibrant colors, 64x64 pixel style, "
+            f"single character only, no scenery no environment"
         ),
         "sprite_background.png": (
             f"2D game background scene, {perspective}, {theme} theme, "
-            f"wide landscape, atmospheric, game art style, 800x450, no characters"
+            f"wide landscape, atmospheric, game art style, 800x450, "
+            f"no characters no sprites, environment only"
         ),
         "sprite_enemy.png": (
             f"pixel-art enemy character, {char_style}, {theme} theme, "
-            f"menacing, character ONLY with NO background, transparent background, "
-            f"clear silhouette, 64x64 pixel style"
+            f"menacing and threatening, isolated character on pure solid-color background, "
+            f"strong clear silhouette, 64x64 pixel style, "
+            f"single enemy only, no scenery no environment"
         ),
     }
     if game_type == "Platformer":
         seeds["sprite_platform.png"] = (
             f"pixel-art solid platform tile, 2D side-view, {theme} theme, "
-            f"rectangular textured surface, NO background, transparent background, 128x24"
+            f"rectangular stone or wooden surface, isolated on solid-color background, "
+            f"no characters, 128x24 pixel style"
         )
         seeds["sprite_goal.png"] = (
-            f"pixel-art goal treasure chest or star, 2D side-view, {theme} theme, "
-            f"glowing, clearly visible, NO background, transparent background, 40x40"
+            f"pixel-art goal treasure chest or glowing star, 2D side-view, {theme} theme, "
+            f"glowing and clearly visible, isolated on solid-color background, "
+            f"no characters, 40x40 pixel style"
         )
     prompts = {}
     for sprite_name, seed in seeds.items():
@@ -211,6 +224,22 @@ def generate_image_prompts(theme: str, game_type: str) -> dict:
 # Step 2: Generate images via Pollinations.AI
 # ---------------------------------------------------------------------------
 
+def _remove_background(pil_image: Image.Image) -> Image.Image:
+    """Remove background from sprite using rembg library."""
+    try:
+        from rembg import remove
+        buf_in = io.BytesIO()
+        pil_image.save(buf_in, format="PNG")
+        buf_in.seek(0)
+        buf_out = io.BytesIO()
+        remove(buf_in.read(), output=buf_out)
+        buf_out.seek(0)
+        return Image.open(buf_out).convert("RGBA")
+    except Exception as exc:
+        print(f"[rembg] Background removal failed: {exc}")
+        return pil_image
+
+
 def _pil_to_data_uri(pil_image: Image.Image, size: tuple = None) -> str:
     img = pil_image.resize(size, Image.LANCZOS) if size else pil_image
     buf = io.BytesIO()
@@ -253,7 +282,10 @@ def generate_sprites(image_prompts: dict) -> tuple:
                     else:
                         raise retry_exc
             pil_img = Image.open(io.BytesIO(response.content))
-            size    = None if is_bg else (64, 64)
+            # Remove background for character/object sprites (not background scene)
+            if not is_bg:
+                pil_img = _remove_background(pil_img)
+            size = None if is_bg else (64, 64)
             sprite_map[sprite_name] = _pil_to_data_uri(pil_img, size=size)
             print(f"[Pollinations] OK: {sprite_name}")
         except Exception as exc: