Update app.py

app.py

@@ -7,27 +7,23 @@ def process(payload: dict):
     try:
         image_b64 = payload["image_b64"]
 
-        # Base64 → bytes
-        img_bytes = base64.b64decode(image_b64)
+        # 直接寫到 space sandbox
+        tmp_path = "/tmp/tmp.jpg"
 
-        # Save to local file
-        tmp_path = "tmp.jpg"
         with open(tmp_path, "wb") as f:
-            f.write(img_bytes)
-
-        # Check file exists + return success
-        if os.path.exists(tmp_path):
-            size = os.path.getsize(tmp_path)
-            return {
-                "saved": True,
-                "path": tmp_path,
-                "file_size_bytes": size
-            }
-        else:
-            return {"saved": False, "error": "File not found after save."}
+            f.write(base64.b64decode(image_b64))
+
+        # 再讀取確認寫入成功
+        file_size = os.path.getsize(tmp_path)
+
+        return {
+            "saved": True,
+            "file_path": tmp_path,
+            "file_size_bytes": file_size
+        }
 
     except Exception as e:
-        return {"saved": False, "error": str(e)}
+        return {"error": str(e)}
 
 demo = gr.Interface(
     fn=process,