diff --git a/docs/build_docs.py b/docs/build_docs.py index 95a86cde..914f2fe8 100644 --- a/docs/build_docs.py +++ b/docs/build_docs.py @@ -83,6 +83,21 @@ def update_html_links(): print(f'Total number of links updated: {total_updated_links}') +def update_page_title(file_path: Path, new_title: str): + """Update the title of an HTML file.""" + + # Read the content of the file + with open(file_path, encoding='utf-8') as file: + content = file.read() + + # Replace the existing title with the new title + updated_content = re.sub(r'.*?', f'{new_title}', content) + + # Write the updated content back to the file + with open(file_path, 'w', encoding='utf-8') as file: + file.write(updated_content) + + def main(): # Build the docs build_docs() @@ -93,6 +108,9 @@ def main(): # Show command to serve built website print('Serve site at http://localhost:8000 with "python -m http.server --directory site"') + # Update titles + update_page_title(SITE / '404.html', new_title='Ultralytics Docs - Not Found') + if __name__ == '__main__': main()