Skip to content

Commit

Permalink
feat: started working on ncurses ui
Browse files Browse the repository at this point in the history
  • Loading branch information
@ErikBjare
ErikBjare committed Oct 20, 2024
1 parent 756e420 commit d3413ea
Showing 1 changed file with 202 additions and 0 deletions.
202 changes: 202 additions & 0 deletions gptme/ncurses.py
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,202 @@
import curses
import textwrap

class Message:
def __init__(self, content, role="user"):
self.content = content
self.expanded = False
self.role = role

class MessageApp:
def __init__(self, stdscr):
self.stdscr = stdscr
self.messages = []
self.input_buffer = ""
self.cursor_y = 0
self.cursor_x = 0
self.scroll_offset = 0
self.mode = "normal"
self.selected_message = None
self.current_role = "user"

def add_message(self, content):
self.messages.append(Message(content, self.current_role))

def draw(self):
self.stdscr.clear()
height, width = self.stdscr.getmaxyx()

# Draw messages
for i, message in enumerate(self.messages[self.scroll_offset:]):
if i >= height - 3:
break
if message == self.selected_message:
self.stdscr.attron(curses.A_REVERSE)

role_color = curses.COLOR_GREEN if message.role == "user" else curses.COLOR_BLUE if message.role == "assistant" else curses.COLOR_RED
self.stdscr.attron(curses.color_pair(role_color))
self.stdscr.addstr(i, 1, f"[{message.role}] ")
self.stdscr.attroff(curses.color_pair(role_color))

wrapped_lines = textwrap.wrap(message.content, width - 12) # Adjusted for role prefix
for j, line in enumerate(wrapped_lines[:3 if not message.expanded else None]):
self.stdscr.addstr(i + j, 11, line) # Adjusted for role prefix
if not message.expanded and len(wrapped_lines) > 3:
self.stdscr.addstr(i + 2, width - 5, "...")
if message == self.selected_message:
self.stdscr.attroff(curses.A_REVERSE)

# Draw input box
self.stdscr.addstr(height - 2, 0, "-" * width)
role_color = curses.COLOR_GREEN if self.current_role == "user" else curses.COLOR_BLUE if self.current_role == "assistant" else curses.COLOR_RED
self.stdscr.attron(curses.color_pair(role_color))
input_prefix = f"[{self.current_role}]> "
self.stdscr.addstr(height - 1, 0, input_prefix)
self.stdscr.attroff(curses.color_pair(role_color))

max_input_width = width - len(input_prefix) - 1 # Leave 1 character for cursor
if len(self.input_buffer) > max_input_width:
visible_input = self.input_buffer[-max_input_width:]
self.stdscr.addstr(height - 1, len(input_prefix), visible_input)
else:
self.stdscr.addstr(height - 1, len(input_prefix), self.input_buffer)

# Draw mode indicator
self.stdscr.addstr(0, width - 10, f"[{self.mode.upper()}]")

# Position cursor
if self.mode == "input" or self.mode == "edit":
cursor_x = min(max_input_width, self.cursor_x)
self.stdscr.move(height - 1, len(input_prefix) + cursor_x)

self.stdscr.refresh()

def run(self):
curses.curs_set(1)
curses.start_color()
curses.init_pair(curses.COLOR_GREEN, curses.COLOR_GREEN, curses.COLOR_BLACK)
curses.init_pair(curses.COLOR_BLUE, curses.COLOR_BLUE, curses.COLOR_BLACK)
curses.init_pair(curses.COLOR_RED, curses.COLOR_RED, curses.COLOR_BLACK)

while True:
self.draw()
key = self.stdscr.getch()

if self.mode == "normal":
if key == ord('q'):
break
elif key == ord('i'):
self.mode = "input"
self.cursor_x = len(self.input_buffer)
elif key == ord('s'):
self.mode = "select"
self.selected_message = self.messages[0] if self.messages else None
elif key == ord('r'):
self.mode = "role"
elif key == curses.KEY_UP:
self.scroll_offset = max(0, self.scroll_offset - 1)
elif key == curses.KEY_DOWN:
self.scroll_offset = min(len(self.messages) - 1, self.scroll_offset + 1)

elif self.mode == "input":
if key == 27: # ESC
self.mode = "normal"
elif key == 10: # Enter
if self.input_buffer:
self.add_message(self.input_buffer)
self.input_buffer = ""
self.cursor_x = 0
elif key == curses.KEY_BACKSPACE or key == 127:
if self.cursor_x > 0:
self.input_buffer = self.input_buffer[:self.cursor_x-1] + self.input_buffer[self.cursor_x:]
self.cursor_x -= 1
elif key == curses.KEY_DC: # Delete key
if self.cursor_x < len(self.input_buffer):
self.input_buffer = self.input_buffer[:self.cursor_x] + self.input_buffer[self.cursor_x+1:]
elif key == curses.KEY_LEFT:
self.cursor_x = max(0, self.cursor_x - 1)
elif key == curses.KEY_RIGHT:
self.cursor_x = min(len(self.input_buffer), self.cursor_x + 1)
elif key == curses.KEY_HOME:
self.cursor_x = 0
elif key == curses.KEY_END:
self.cursor_x = len(self.input_buffer)
elif 32 <= key <= 126: # Printable ASCII characters
self.input_buffer = self.input_buffer[:self.cursor_x] + chr(key) + self.input_buffer[self.cursor_x:]
self.cursor_x += 1

elif self.mode == "select":
if key == 27: # ESC
self.mode = "normal"
self.selected_message = None
elif key == ord('e'):
self.mode = "edit"
self.input_buffer = self.selected_message.content
self.cursor_x = len(self.input_buffer)
elif key == ord('x'):
self.selected_message.expanded = not self.selected_message.expanded
elif key == ord('d'):
self.messages.remove(self.selected_message)
if self.messages:
self.selected_message = self.messages[0]
else:
self.selected_message = None
self.mode = "normal"
elif key == curses.KEY_UP and self.messages:
idx = self.messages.index(self.selected_message)
self.selected_message = self.messages[max(0, idx - 1)]
elif key == curses.KEY_DOWN and self.messages:
idx = self.messages.index(self.selected_message)
self.selected_message = self.messages[min(len(self.messages) - 1, idx + 1)]

elif self.mode == "edit":
if key == 27: # ESC
self.mode = "select"
self.input_buffer = ""
self.cursor_x = 0
elif key == 10: # Enter
self.selected_message.content = self.input_buffer
self.mode = "select"
self.input_buffer = ""
self.cursor_x = 0
elif key == curses.KEY_BACKSPACE or key == 127:
if self.cursor_x > 0:
self.input_buffer = self.input_buffer[:self.cursor_x-1] + self.input_buffer[self.cursor_x:]
self.cursor_x -= 1
elif key == curses.KEY_DC: # Delete key
if self.cursor_x < len(self.input_buffer):
self.input_buffer = self.input_buffer[:self.cursor_x] + self.input_buffer[self.cursor_x+1:]
elif key == curses.KEY_LEFT:
self.cursor_x = max(0, self.cursor_x - 1)
elif key == curses.KEY_RIGHT:
self.cursor_x = min(len(self.input_buffer), self.cursor_x + 1)
elif key == curses.KEY_HOME:
self.cursor_x = 0
elif key == curses.KEY_END:
self.cursor_x = len(self.input_buffer)
elif 32 <= key <= 126: # Printable ASCII characters
self.input_buffer = self.input_buffer[:self.cursor_x] + chr(key) + self.input_buffer[self.cursor_x:]
self.cursor_x += 1

elif self.mode == "role":
if key == ord('u'):
self.current_role = "user"
self.mode = "normal"
elif key == ord('a'):
self.current_role = "assistant"
self.mode = "normal"
elif key == ord('s'):
self.current_role = "system"
self.mode = "normal"
elif key == 27: # ESC
self.mode = "normal"

def main(stdscr):
app = MessageApp(stdscr)
app.add_message("Welcome to the Message App!")
app.add_message("Press 'i' to enter input mode, 's' to enter select mode, 'r' to change role, and 'q' to quit.")
app.add_message("In select mode, use arrow keys to navigate, 'e' to edit, 'x' to expand/collapse, and 'd' to delete.")
app.run()

if __name__ == "__main__":
curses.wrapper(main)

0 comments on commit d3413ea

Please sign in to comment.