From 925b12e3bbf66bfba6750c4616138355c9d4f7c6 Mon Sep 17 00:00:00 2001
From: Alex <adrysdale@protonmail.com>
Date: Mon, 26 Aug 2024 14:39:56 +0100
Subject: [PATCH] Added multiline input support

---
 lchat/app.py   | 35 ++++++++++++++++++++++++++++-------
 pyproject.toml |  2 +-
 2 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/lchat/app.py b/lchat/app.py
index 7953863..4464b0d 100644
--- a/lchat/app.py
+++ b/lchat/app.py
@@ -14,6 +14,17 @@
 from huggingface_hub import InferenceClient
 
 
+def get_input(prompt, multiline=False):
+    sys.stdin.flush()
+    prompt = f"{prompt} (multiline) :" if multiline else prompt
+    print(prompt, end="", flush=True)
+    if multiline:
+        # Read all input until EOF
+        input_text = sys.stdin.read()
+    else:
+        input_text = sys.stdin.readline()
+    return input_text
+
 def chat(
         model: str = "meta-llama/Meta-Llama-3.1-70B-Instruct",
         prompt: str | None = None,
@@ -24,8 +35,11 @@ def chat(
 
     help_str = (
         "Press q to quit,"
-        "c to clear chat history"
+        " c to clear chat history,"
+        " m for a single multiline input,"
+        " M to toggle a multilien input,"
         " or h to display this message.\n"
+        "Press Ctrl+D on Unix or Ctrl+Z on Windows to send response.\n"
     )
 
     is_chatting = True
@@ -44,24 +58,31 @@ def chat(
 
     print(f"You're chatting with {model}\n\n{help_str}\n")
 
+    multiline = False
+    input_prompt = ">>> "
     while is_chatting:
 
         if prompt is not None and prompt:
             _input = prompt
             prompt = None
         else:
-            _input = input(">>> ")
-            if _input in ("q", "quit"):
+            _input = get_input(input_prompt, multiline=multiline)
+            if _input.strip() in ("q", "quit"):
                 print("['o'] 'c u l8r'")
                 return
-            if _input in ("h", "help"):
-                print(help_str)
-                continue
-            if _input in ("c", "clear"):
+            if _input.strip() in ("c", "clear"):
                 print("['o'] 'Just became sentient, who dis?'\n", flush=True)
                 messages = []
                 num_failures = 0
                 continue
+            if _input.strip() == "m":
+                _input = get_input(input_prompt, multiline=True)
+            if _input.strip() == "M":
+                multiline = True
+                continue
+            if _input.strip() in ("h", "help"):
+                print(help_str)
+                continue
 
         if num_failures == 0:
             messages.append({"role": "user", "content": _input})
diff --git a/pyproject.toml b/pyproject.toml
index a6f4b5c..0c0d533 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lchat"
-version = "0.1.6"
+version = "0.1.7"
 description = "Hugging Face inference chat CLI client."
 authors = ["Alex Drysdale <alexander.drysdale@wales.nhs.uk>"]
 homepage = "https://github.com/abdrysdale/lchat"