Spaces:
Running
Running
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}")
|