Timothy J. Baek 1 년 전
부모
커밋
989a4d3165
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      backend/apps/images/main.py

+ 2 - 2
backend/apps/images/main.py

@@ -402,7 +402,7 @@ def generate_image(
             for image in res["data"]:
             for image in res["data"]:
                 image_filename = save_b64_image(image["b64_json"])
                 image_filename = save_b64_image(image["b64_json"])
                 images.append({"url": f"/cache/image/generations/{image_filename}"})
                 images.append({"url": f"/cache/image/generations/{image_filename}"})
-                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_id}.json")
+                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_filename}.json")
 
 
                 with open(file_body_path, "w") as f:
                 with open(file_body_path, "w") as f:
                     json.dump(data, f)
                     json.dump(data, f)
@@ -477,7 +477,7 @@ def generate_image(
             for image in res["images"]:
             for image in res["images"]:
                 image_filename = save_b64_image(image)
                 image_filename = save_b64_image(image)
                 images.append({"url": f"/cache/image/generations/{image_filename}"})
                 images.append({"url": f"/cache/image/generations/{image_filename}"})
-                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_id}.json")
+                file_body_path = IMAGE_CACHE_DIR.joinpath(f"{image_filename}.json")
 
 
                 with open(file_body_path, "w") as f:
                 with open(file_body_path, "w") as f:
                     json.dump({**data, "info": res["info"]}, f)
                     json.dump({**data, "info": res["info"]}, f)