44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
|
import sys
|
||
|
|
import subprocess
|
||
|
|
import re
|
||
|
|
import io
|
||
|
|
|
||
|
|
# Fix Windows console encoding
|
||
|
|
if sys.platform == 'win32':
|
||
|
|
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
|
||
|
|
sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8')
|
||
|
|
|
||
|
|
def translate(text):
|
||
|
|
"""Translate text to Russian using ollama translator model"""
|
||
|
|
try:
|
||
|
|
# Run ollama command
|
||
|
|
result = subprocess.run(
|
||
|
|
['ollama', 'run', 'translator', text],
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
encoding='utf-8'
|
||
|
|
)
|
||
|
|
|
||
|
|
# Get output
|
||
|
|
output = result.stdout
|
||
|
|
|
||
|
|
# Remove <think>...</think> tags and their content
|
||
|
|
output = re.sub(r'<think>.*?</think>', '', output, flags=re.DOTALL)
|
||
|
|
|
||
|
|
# Remove empty lines and strip whitespace
|
||
|
|
output = '\n'.join(line.strip() for line in output.split('\n') if line.strip())
|
||
|
|
|
||
|
|
return output
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
return f"Error: {e}"
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
print("Usage: python translate.py \"text to translate\"")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
text = ' '.join(sys.argv[1:])
|
||
|
|
result = translate(text)
|
||
|
|
print(result)
|