Create app.py

app.py

@@ -0,0 +1,56 @@
+import gradio as gr
+from PIL import Image
+import os, time, uuid
+
+# If your Space has "Persistent Storage" enabled, /data is the mount point.
+# If not, it still works but files won't survive a restart.
+SAVE_DIR = os.getenv("SAVE_DIR", "/data/uploads")
+os.makedirs(SAVE_DIR, exist_ok=True)
+
+ALLOWED_FORMATS = {"JPEG", "PNG"}   # strict enforcement
+
+def handle_upload(file_path: str):
+    if not file_path:
+        return "Please choose an image.", None
+
+    # Verify the real file type by reading the header via Pillow.
+    try:
+        with Image.open(file_path) as im:
+            fmt = (im.format or "").upper()
+            if fmt not in ALLOWED_FORMATS:
+                return f"Only JPEG/PNG allowed. Detected: {fmt or 'unknown'}.", None
+
+            # Optional: normalize mode (e.g., ensure RGB)
+            if im.mode not in ("RGB", "RGBA"):
+                im = im.convert("RGB")
+
+            # Save with a unique filename
+            ext = ".jpg" if fmt == "JPEG" else ".png"
+            filename = f"{int(time.time())}_{uuid.uuid4().hex}{ext}"
+            save_path = os.path.join(SAVE_DIR, filename)
+            im.save(save_path)
+
+            info = f"✅ Saved as **{filename}** · {im.width}×{im.height}px · {fmt}"
+            # Return a copy for the preview (so file handle is closed)
+            return info, im.copy()
+    except Exception as e:
+        return f"Failed to read image: {e}", None
+
+with gr.Blocks(title="Image Upload (JPEG/PNG)") as demo:
+    gr.Markdown("### Upload an image (JPEG/PNG). It will be validated and saved to storage.")
+    with gr.Row():
+        inp = gr.Image(
+            label="Drop or click to upload",
+            type="filepath",               # gives us the original temp filepath
+            image_mode="RGB",
+            sources=["upload", "clipboard"]
+        )
+        preview = gr.Image(label="Preview", interactive=False)
+
+    msg = gr.Markdown()
+    # Auto-save on change; also provide an explicit button for clarity.
+    inp.change(handle_upload, inputs=inp, outputs=[msg, preview])
+    gr.Button("Save").click(handle_upload, inputs=inp, outputs=[msg, preview])
+
+if __name__ == "__main__":
+    demo.launch()