Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -12,6 +12,7 @@ import requests
|
|
12 |
import mimetypes
|
13 |
import tempfile
|
14 |
import subprocess
|
|
|
15 |
|
16 |
|
17 |
from PIL import Image
|
@@ -160,7 +161,7 @@ def load_image(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', frame_
|
|
160 |
@app.post("/save")
|
161 |
async def save(image_data: RequestModel):
|
162 |
os.makedirs(BASE_DIR, exist_ok=True)
|
163 |
-
filename = os.path.join(BASE_DIR, f"{image_data.originId}_{image_data.assetCode}.json")
|
164 |
|
165 |
img1 = load_image(image_data.originSource, f"origin {image_data.originSource}")
|
166 |
img2 = load_image(image_data.source, image_data.assetCode, image_data.contentType)
|
|
|
12 |
import mimetypes
|
13 |
import tempfile
|
14 |
import subprocess
|
15 |
+
import uuid
|
16 |
|
17 |
|
18 |
from PIL import Image
|
|
|
161 |
@app.post("/save")
|
162 |
async def save(image_data: RequestModel):
|
163 |
os.makedirs(BASE_DIR, exist_ok=True)
|
164 |
+
filename = os.path.join(BASE_DIR, f"{image_data.originId}_{image_data.assetCode}_{uuid.uuid4().hex[:8]}.json")
|
165 |
|
166 |
img1 = load_image(image_data.originSource, f"origin {image_data.originSource}")
|
167 |
img2 = load_image(image_data.source, image_data.assetCode, image_data.contentType)
|