text-correction-groq.py

from pynput import keyboard
from pynput.keyboard import Key, Controller
import pyperclip
import time

from groq import Groq

client = Groq(api_key="your-api-key-here")


controller = Controller()


def fix_text(text):
    message = text

    chat_completion = client.chat.completions.create(
    messages=[
        {
            "role": "system",
            "content": "you are an ai system designed to fix typos in a provided text while preserving all lines and punctuations. If provided with a text, eg: 'helloooo there, howww are yoou', you will return 'hello there, how are you' and nothing else. Do not respond with a preamble or any other information except for the corrected text. Do not even put the corrected text in quotation marks."
        },
        {
            "role": "user",
            "content": message,
        }
    ],
    model="mixtral-8x7b-32768",
)
    return chat_completion.choices[0].message.content

def fix_selection():
    with controller.pressed(Key.ctrl):
        controller.tap('c')
    
    time.sleep(0.1)

    text = pyperclip.paste()

    fixed_text = fix_text(text)

    pyperclip.copy(fixed_text)

    time.sleep(0.1)

    with controller.pressed(Key.ctrl):
        controller.tap('v')
    

def on_f9():
    fix_selection()



with keyboard.GlobalHotKeys({
    '<65478>': on_f9}) as h:
    h.join()