keannu125/DocManagerBackend
1
0
Fork 0
mirror of https://github.com/lemeow125/DocManagerBackend.git synced 2025-04-27 18:21:23 +08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add in requested changes

Browse source
This commit is contained in:
Keannu Christian Bernasol 2024-12-16 14:58:50 +08:00
parent d81319c8ec
commit 844113d44f
7 changed files with 54 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

7
docmanager_backend/config/management/commands/start_watcher.py
View file

@ -124,7 +124,7 @@ class PDFHandler(FileSystemEventHandler):
# A few safety checks if the model does not follow through with output instructions
if len(document_type) > 16:
self.logger.warning(
f"Ollama API gave incorrect document category: {response["message"]["content"]}. Retrying...")
f"Ollama API gave incorrect document category: {response['message']['content']}. Retrying...")
break
# If that fails, just use regular OCR read the title as a dirty fix/fallback
@ -146,7 +146,7 @@ class PDFHandler(FileSystemEventHandler):
# Open the file for instance creation
DOCUMENT, created = Document.objects.get_or_create(
name=filename,
name=filename.replace(".pdf", ""),
defaults={
"number_pages": num_pages,
"ocr_metadata": metadata,
@ -158,8 +158,7 @@ class PDFHandler(FileSystemEventHandler):
DOCUMENT.file.save(
name=filename, content=File(open(file_path, "rb")))
self.logger.info(
f"Document '{filename}' created successfully with type '{
document_type}'."
f"Document '{filename}' created successfully with type '{document_type}'."
)
else: