clipshell.py

from dotenv import load_dotenv; load_dotenv()
from subprocess import Popen, PIPE
import google.generativeai as genai
import clipboard
import os, time

sh = lambda cmd: Popen(cmd,stdout=PIPE,shell=True).communicate()[0].strip()

read = clipboard.paste
write = clipboard.copy

genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

def fdecode(x):
	try:
		return x.decode()
	except:
		return str(x)[2:-1]

class ClipShell(object):
	def __init__(self, read=read, write=write, sh=sh):
		self.read = read
		self.write = write
		self.sh = sh

	def on_change(self, data):
		if data.startswith("clipshell::"):
			cmd = data[11:]
			res = fdecode(self.sh(cmd))
			self.write(res)
		elif data.startswith('ai::'):
			cmd = data[7:]
			model = genai.GenerativeModel('gemini-1.0-pro-latest')
			resp = model.generate_content(cmd)
			self.write(resp.text)

	def run(self, wait=0):
		old = ""
		while True:
			time.sleep(wait)
			new = self.read()
			if not old == new:
				self.on_change(new)
				old = new

ClipShell().run() if __name__ == "__main__" else None