Issue prompt from last copied text

Simatwa · Dec 25, 2023 · 26c0ac6 · 26c0ac6
1 parent 0cb3d4c
commit 26c0ac6
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/WebChatGPT/console.py b/WebChatGPT/console.py
@@ -242,11 +242,13 @@ def do_h(self, text):
 ├────┼────────────────────────┼─────────────────────────────────────┤
 │ 14 │ set_theme              │ Set theme for displaying codes      │
 ├────┼────────────────────────┼─────────────────────────────────────┤
-│ 15 │ ./<command>            │ Run system command                  │
+│ 15 │ from_copied            │ Use last copied text as prompt      │
 ├────┼────────────────────────┼─────────────────────────────────────┤
-│ 16 │ <any other>            │ Interact with ChatGPT               │
+│ 16 │ ./<command>            │ Run system command                  │
 ├────┼────────────────────────┼─────────────────────────────────────┤
-│ 17 │ exit                   │ Quits Program                       │
+│ 17 │ <any other>            │ Interact with ChatGPT               │
+├────┼────────────────────────┼─────────────────────────────────────┤
+│ 18 │ exit                   │ Quit Program                        │
 ╘════╧════════════════════════╧═════════════════════════════════════╛
 
 Submit any bug at : {__repo__}/issues/new
@@ -496,6 +498,15 @@ def generate_title(self):
             return resp["message"]
         return "Untitled"
 
+    def do_from_copied(self, line):
+        """Use last copied text as prompt"""
+        issued_prompt = (
+            f"{line}\n{pyperclip.paste()}" if bool(line.strip()) else pyperclip.paste()
+        )
+        click.secho(issued_prompt, fg="yellow")
+        if click.confirm("Do you wish to proceed"):
+            self.default()
+
     # @busy_bar.run()
     def default(self, line):
         """Chat with ChatGPT"""