Skip to content

Commit 1477f15

Browse files
committed
+ universalize filename
1 parent 63481b8 commit 1477f15

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

text_extract_api/extract/tasks.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,10 +23,10 @@ def ocr_task(
2323
filename: str,
2424
file_hash: str,
2525
ocr_cache: bool,
26-
prompt: str,
27-
model: str,
28-
language: str,
29-
storage_profile: str,
26+
prompt: Optional[str] = None,
27+
model: Optional[str] = None,
28+
language: Optional[str] = None,
29+
storage_profile: Optional[str] = None,
3030
storage_filename: Optional[str] = None,
3131
):
3232
"""
@@ -85,7 +85,7 @@ def ocr_task(
8585

8686
if storage_profile:
8787
if not storage_filename:
88-
storage_filename = filename.replace('.pdf', '.md')
88+
storage_filename = filename.replace('.', '_') + '.pdf'
8989

9090
storage_manager = StorageManager(storage_profile)
9191
storage_manager.save(filename, storage_filename, extracted_text)

0 commit comments

Comments
 (0)