llama-cpp-server / replace_hw.py
rbantog's picture
initial commit
4f0cbe4
raw
history blame
1.32 kB
#!/usr/bin/env python3
# filepath: replace_hello_world.py
import os
import urllib.parse
def extract_filename(url):
parsed = urllib.parse.urlsplit(url)
return os.path.basename(parsed.path)
def replace_in_file(file_path, old_text, new_text):
with open(file_path, 'r', encoding='utf8') as file:
content = file.read()
if old_text not in content:
return
updated_content = content.replace(old_text, new_text)
with open(file_path, 'w', encoding='utf8') as file:
file.write(updated_content)
print(f"Updated: {file_path}")
def main():
model_link = os.getenv("MODEL_DOWNLOAD_LINK")
if not model_link:
print("Error: MODEL_DOWNLOAD_LINK environment variable is not set.")
return
download_filename = extract_filename(model_link)
print(f"Extracted filename: {download_filename}")
# Recursively walk thru current directory
for dirpath, _, filenames in os.walk(os.getcwd()):
for filename in filenames:
# Modify the following tuple to include any file extensions you wish to process
if filename.endswith(('.html', '.ts', '.patch')):
full_path = os.path.join(dirpath, filename)
replace_in_file(full_path, "Hello World!", download_filename)
if __name__ == '__main__':
main()