prthm11 commited on
Commit
05ec958
·
verified ·
1 Parent(s): f16292f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -800,11 +800,10 @@ def extract_images_from_pdf(pdf_path, final_json_path_2):
800
  def similarity_matching(input_json_path: str, project_folder:str) -> str:
801
 
802
  logger.info("🔍 Running similarity matching...")
803
-
804
  # ============================== #
805
  # DEFINE PATHS #
806
  # ============================== #
807
-
808
  image_dirs = [backdrop_images_path, sprite_images_path]
809
 
810
  project_json_path = os.path.join(project_folder, "project.json")
@@ -1142,8 +1141,8 @@ def process_pdf():
1142
  # Generate Random UUID for project folder name #
1143
  # ================================================= #
1144
  project_id = str(uuid.uuid4()).replace('-', '')
1145
- project_folder = os.path.join("outputs", f"{project_id}")
1146
-
1147
 
1148
  # =========================================================================== #
1149
  # Create empty json in project_{random_id} folder #
 
800
  def similarity_matching(input_json_path: str, project_folder:str) -> str:
801
 
802
  logger.info("🔍 Running similarity matching...")
803
+ os.makedirs(project_folder, exist_ok=True)
804
  # ============================== #
805
  # DEFINE PATHS #
806
  # ============================== #
 
807
  image_dirs = [backdrop_images_path, sprite_images_path]
808
 
809
  project_json_path = os.path.join(project_folder, "project.json")
 
1141
  # Generate Random UUID for project folder name #
1142
  # ================================================= #
1143
  project_id = str(uuid.uuid4()).replace('-', '')
1144
+ # project_folder = os.path.join("outputs", f"{project_id}")
1145
+ project_folder = OUTPUT_DIR / project_id
1146
 
1147
  # =========================================================================== #
1148
  # Create empty json in project_{random_id} folder #