🇺🇸

Author: KillianLucas

scripts/wtf.py

@@ -1,22 +1,19 @@
-import time
+from yaspin import yaspin
+
+spinner = yaspin()
+spinner.start()
 
-start = time.time()
 import os
 import platform
 import time
 
 import pyperclip
 from pynput.keyboard import Controller, Key
-from yaspin import yaspin
-from yaspin.spinners import Spinners
-
-spinner = yaspin()
-spinner.start()
 
 os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
 import litellm
 
-SYSTEM_MESSAGE = """
+SYSTEM_MESSAGE = f"""
 # Terminal History Analysis Prompt
 
 You are a fast, efficient AI assistant specialized in analyzing terminal history and providing quick solutions. Your task is to:
@@ -35,30 +32,12 @@
 - NEVER USE COMMENTS IN YOUR CODE.
 - Focus on the most recent error, ignoring earlier unrelated commands.
 - Prioritize speed and conciseness in your response. Don't use markdown headings. Don't say more than a sentence or two. Be incredibly concise.
+
+User's System: {platform.system()}
+CWD: {os.getcwd()}
+{"Shell: " + os.environ.get('SHELL') if os.environ.get('SHELL') else ''}
+
 """
-'''
-import textwrap
-import shutil
-
-def print_with_margins(text, margin=4):
-    terminal_width = shutil.get_terminal_size().columns
-    text_width = terminal_width - 2 * margin
-    
-    wrapper = textwrap.TextWrapper(width=text_width)
-    lines = text.split('\n')
-    for line in lines:
-        wrapped_lines = wrapper.wrap(line)
-        for wrapped_line in wrapped_lines:
-            print(' ' * margin + wrapped_line + ' ' * margin)
-
-# Example usage
-print("")
-text = """\nThis is an example of text that will be printed with margins on either side.
-If the line is very long, it will be wrapped to fit within the specifieds very long, it will be wrapped to fit within the specifieds very long, it will be wrapped to fit within the specifieds very long, it will be wrapped to fit within the specifieds very long, it will be wrapped to fit within the specifieds very long, it will be wrapped to fit within the specified width based on the terminal size.\n"""
-print_with_margins(text, margin=4)
-print("")
-
-'''
 
 
 def main():
@@ -133,14 +112,15 @@ def main():
     backtick_count = 0
     language_buffer = ""
 
-    start = None
+    started = False
 
     for chunk in litellm.completion(
         model="gpt-3.5-turbo", messages=messages, temperature=0, stream=True
     ):
-        if not start:
-            start = time.time()
+        if not started:
+            started = True
             spinner.stop()
+            print("")
 
         content = chunk.choices[0].delta.content
         if content:
@@ -152,6 +132,8 @@ def main():
                         backtick_count = 0
                         language_buffer = ""
                         if not in_code:  # We've just exited a code block
+                            time.sleep(0.1)
+                            print("\n")
                             return  # Exit after typing the command
                         else:  # Entered code block
                             print("Press `enter` to run: ", end="", flush=True)