Skip to content

Instantly share code, notes, and snippets.

@willwade
Last active February 2, 2024 07:57
Show Gist options
  • Save willwade/f5d9a2c30906e33fb404437b870ef89a to your computer and use it in GitHub Desktop.
Save willwade/f5d9a2c30906e33fb404437b870ef89a to your computer and use it in GitHub Desktop.
AutoCorrect using OpenAI
from pynput import keyboard, mouse
from pynput.keyboard import Key, Controller as KeyboardController
import requests
import clipman
clipman.init()
keyboard_controller = KeyboardController()
text_buffer = []
def correct_text(text):
api_url = "https://api.openai.com/v1/chat/completions"
api_key = "sk-keyhere" # Replace with your actual API key
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json",
}
data = {
"model": "gpt-3.5-turbo",
"messages": [
{
"role": "system",
"content": "Autocorrect sentence that may have typos in it and capitalization errors as well as bad spacing between words. You will not be verbose. Just return the corrected text if it can be"
},
{
"role": "user",
"content": text
}
]
}
response = requests.post(api_url, json=data, headers=headers)
if response.status_code == 200:
assistant_response = response.json()['choices'][0]['message']['content'].strip()
clipman.set(assistant_response) # Copy corrected text to clipboard
replace_text(len(text))
else:
print(f"Failed to correct text. HTTP Status Code: {response.status_code}")
print(response.text)
def replace_text(length_of_text):
# Simulate Shift + Left Arrow to select the text
with keyboard_controller.pressed(Key.shift):
for _ in range(length_of_text):
keyboard_controller.press(Key.left)
keyboard_controller.release(Key.left)
# Simulate Ctrl+V to paste the corrected text
clipman.paste()
def on_press(key):
try:
# Attempt to capture printable characters
if hasattr(key, 'char') and key.char is not None:
text_buffer.append(key.char)
elif key == keyboard.Key.space:
# Process text upon pressing space, for simplicity
sentence = ''.join(text_buffer)
if sentence.endswith('.'):
correct_text(sentence)
text_buffer.clear() # Clear the buffer after processing
except Exception as e:
print(f"Error processing key press: {e}")
with keyboard.Listener(on_press=on_press) as listener:
listener.join()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment