Create app.py

app.py

@@ -0,0 +1,37 @@
+import streamlit as st
+import os
+from PIL import Image
+import subprocess
+
+st.set_page_config(page_title="PaddleGAN Editor", layout="wide")
+st.title("🎨 PaddleGAN Image Editor")
+
+task = st.selectbox("Choose a task", ["Photo2Cartoon", "GPEN Face Enhancement"])
+
+uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png"])
+
+if uploaded_file:
+    input_path = f"input.jpg"
+    with open(input_path, "wb") as f:
+        f.write(uploaded_file.read())
+    st.image(Image.open(input_path), caption="Uploaded Image", use_column_width=True)
+
+    if st.button("Run PaddleGAN"):
+        if task == "Photo2Cartoon":
+            cmd = f"python tools/infer.py --config configs/photo2cartoon.yaml --input_dir {input_path} --output_dir output"
+        elif task == "GPEN Face Enhancement":
+            cmd = f"python tools/infer.py --config configs/gpen.yaml --input_dir {input_path} --output_dir output"
+        else:
+            st.error("Unsupported task.")
+            st.stop()
+
+        with st.spinner("Running PaddleGAN..."):
+            subprocess.run(cmd, shell=True)
+
+        result_path = "output/result.jpg"
+        if os.path.exists(result_path):
+            st.image(Image.open(result_path), caption="Result", use_column_width=True)
+            with open(result_path, "rb") as f:
+                st.download_button("📥 Download Result", f.read(), "edited.jpg", "image/jpeg")
+        else:
+            st.error("No result found.")