github/bilingual_book_maker
1
1
Fork 0
mirror of https://github.com/yihong0618/bilingual_book_maker.git synced 2025-06-02 09:30:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:Fix translation paragraph count mismatch by explicitly instructing LLM about paragraph requirements

Browse Source
This commit is contained in:
leslie 2025-04-19 23:28:02 +08:00
parent 750ecd7d93
commit 09589c626d
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
book_maker/translator/chatgptapi_translator.py
View File

@ -384,6 +384,7 @@ class ChatGPTAPI(Base):
f"Each paragraph is wrapped in numbered tags like <p1>text</p1>. "
f"DO NOT merge paragraphs. Keep each paragraph separate. "
f"DO NOT combine multiple paragraphs into one. "
f"Preserve numbers at the beginning of paragraphs like '17' or '10x'. "
f"Each original paragraph should become exactly one translated paragraph. "
f"Example output format: <p1>translated text for paragraph 1</p1>\n<p2>translated text for paragraph 2</p2>\n...\n<p{plist_len}>translated text for paragraph {plist_len}</p{plist_len}>"
)
@ -438,7 +439,7 @@ class ChatGPTAPI(Base):
continue
final_result_list.append(paragraph)
# Ensure we have plist_len paragraphs
# Ensure we have exactly plist_len paragraphs
if len(final_result_list) > plist_len:
final_result_list = final_result_list[:plist_len]
elif len(final_result_list) < plist_len:
@ -454,9 +455,9 @@ class ChatGPTAPI(Base):
plist_len, result_list, new_str, "\n", log_path
)
# Del paragraph numbers if any remain
# Remove ONLY the paragraph numbering formats, not all numbers at the start
final_result_list = [
re.sub(r"^(\(\d+\)|\d+\.|(\d+))\s*", "", s) for s in final_result_list
re.sub(r"^(\(\d+\)|\d+\.)\s*", "", s) for s in final_result_list
]
return final_result_list