github-actions Claude Sonnet 4.6 commited on
Commit
d4dbbd5
·
1 Parent(s): 4cd4e97

fix: don't delete pre-existing files in output folder when not resetting

Browse files
comic_panel_extractor/llm_panel_extractor.py CHANGED
@@ -165,6 +165,9 @@ def extract_panel_via_llm(input_image_path, config=None, reset=True):
165
  if Path(extractor_config.output_folder).exists():
166
  shutil.rmtree(extractor_config.output_folder, ignore_errors=True)
167
  Path(extractor_config.output_folder).mkdir(exist_ok=True)
 
 
 
168
 
169
  # Initialize extractor
170
  panel_extractor = LLMPanelExtractor(extractor_config)
@@ -248,7 +251,7 @@ def extract_panel_via_llm(input_image_path, config=None, reset=True):
248
 
249
  if not extractor_config.debug:
250
  for file in os.listdir(extractor_config.output_folder):
251
- if "_panel_" not in file:
252
  os.remove(os.path.join(extractor_config.output_folder, file))
253
 
254
  print(f"Processing complete. Final result saved to: {extractor_config.output_folder}")
 
165
  if Path(extractor_config.output_folder).exists():
166
  shutil.rmtree(extractor_config.output_folder, ignore_errors=True)
167
  Path(extractor_config.output_folder).mkdir(exist_ok=True)
168
+ pre_existing_files = set()
169
+ else:
170
+ pre_existing_files = set(os.listdir(extractor_config.output_folder)) if Path(extractor_config.output_folder).exists() else set()
171
 
172
  # Initialize extractor
173
  panel_extractor = LLMPanelExtractor(extractor_config)
 
251
 
252
  if not extractor_config.debug:
253
  for file in os.listdir(extractor_config.output_folder):
254
+ if "_panel_" not in file and file not in pre_existing_files:
255
  os.remove(os.path.join(extractor_config.output_folder, file))
256
 
257
  print(f"Processing complete. Final result saved to: {extractor_config.output_folder}")