Update app.py

app.py

@@ -61,6 +61,17 @@ def load_css(path: str) -> str:
     return Path(path).read_text(encoding="utf-8")
 
 
+
+def to_rgb(img: Image.Image) -> Image.Image:
+    if img.mode == "RGB":
+        return img
+    if img.mode in ("RGBA", "LA"):
+        bg = Image.new("RGB", img.size, (255, 255, 255))
+        bg.paste(img, mask=img.split()[-1])
+        return bg
+    return img.convert("RGB")
+
+
 class I18N:
     """
     Simple i18n manager backed by a JSON string.
@@ -148,10 +159,8 @@ def as_pil_list(gallery_value: Any) -> List[Image.Image]:
         return []
     imgs: List[Image.Image] = []
     for item in gallery_value:
-        if isinstance(item, tuple) and len(item) >= 1:
-            imgs.append(item[0])
-        else:
-            imgs.append(item)
+        im = item[0] if isinstance(item, tuple) and len(item) >= 1 else item
+        imgs.append(to_rgb(im))
     return imgs
 
 
@@ -166,9 +175,9 @@ def ensure_pil(x: Any) -> Image.Image:
     :raises TypeError: If unsupported type.
     """
     if isinstance(x, Image.Image):
-        return x
+        return to_rgb(x)
     if isinstance(x, torch.Tensor):
-        return to_pil_image(x.clamp(0, 1))
+        return to_rgb(to_pil_image(x.clamp(0, 1)))
     raise TypeError(f"Unsupported output type: {type(x)}")
 
 
@@ -1734,23 +1743,15 @@ class TTPApp:
             return demo
 
 
-def main() -> None:
-    """
-    Gradio entrypoint.
 
-    :return: None
-    :rtype: None
-    """
-    texts_json = load_texts_json(DEFAULT_I18N_PATH)
-    i18n = I18N(texts_json, default_lang="EN")
-    factory = TransformFactory()
-    codegen = CodeGenerator()
-    engine = TransformationEngine(factory)
-    app = TTPApp(i18n=i18n, engine=engine, codegen=codegen)
-
-    demo = app.build()
-    demo.launch(css=load_css(DEFAULT_CSS_PATH), ssr_mode=False, debug=False)
+texts_json = load_texts_json(DEFAULT_I18N_PATH)
+i18n = I18N(texts_json, default_lang="EN")
+factory = TransformFactory()
+codegen = CodeGenerator()
+engine = TransformationEngine(factory)
+app = TTPApp(i18n=i18n, engine=engine, codegen=codegen)
 
+demo = app.build()
 
 if __name__ == "__main__":
-    main()
+    demo.launch(css=load_css(DEFAULT_CSS_PATH), ssr_mode=False, debug=False)