feat(chat): new cli command for continuous chat (#11)

README.md

@@ -6,20 +6,31 @@
 
 A simple python CLI wrapper for the [ChatGPT API](https://platform.openai.com/docs/guides/chat/introduction). 
 
-Installation:
+## Installation
 ```
 $ pip install chatgpt-api-cli
 ```
 
-Setup:
+## Setup
 ```
 $ chatgpt init "your openai API key"
 ```
 
-Usage:
+## Usage
+
+### Single prompts:
 ```
 $ chatgpt prompt "Explain the current state of the economy as if you're Christopher Moltisanti from the Sopranos"
 
 
 "Listen up, Tony. The economy? Fuggedaboudit. It's in the shitter. The recession's got us all by the balls, and businesses are falling like dead flies. The GDP's down, unemployment's up - it's a real shitshow. People are struggling just to make ends meet, while the fat cats on Wall Street keep raking in the dough. And we all know who's responsible for this mess - the damn politicians and their crooked deals. I'm telling you, Tone, this whole system is screwed up, and it's getting worse every day. It's enough to make a guy wanna start his own damn business, just so he can make a buck without getting screwed over by the man."
 ```
+
+### Continuous chat:
+```
+$ chatgpt chat
+To exit, type `exit` or press CTRL+C
+
+You: why did the chicken cross the road?
+ChatGPT: As an AI language model, I don't have access to the context as to why the chicken might cross the road. However, the most common answer to that age-old question has been "to get to the other side."
+```

chatgpt_cli/main.py

@@ -2,7 +2,9 @@
 from pathlib import Path
 from chatgpt_cli.services.chatgpt import query_chatgpt
 from chatgpt_cli.services.chatgpt import ChatGPTResponse
+from rich.console import Console
 
+console = Console()
 
 app = typer.Typer()
 
@@ -26,14 +28,32 @@ def init(
     with config_path.open("w", encoding="utf-8") as f:
         f.write(f"OPENAI_API_KEY={open_api_key}")
 
-    print("Done!")
+    console.print("Done!", style="bold green")
 
 
 @app.command()
 def prompt(prompt: str = typer.Argument(..., help="Prompt for ChatGPT")):
     """Query ChatGPT with a prompt."""
     response: ChatGPTResponse = query_chatgpt(prompt=prompt)
-    print(response.choices[0].message.content)
+    message = response.choices[0].message.content.strip()
+    console.print(f"\n{message}", style="bold cyan")
+
+
+@app.command()
+def chat():
+    """Chat with ChatGPT."""
+    console.print("To exit, type `exit` or press CTRL+C", style="bold yellow")
+    while True:
+        prompt = typer.prompt("\nYou")
+        if prompt.lower() in ("exit", "quit"):
+            exit = typer.confirm("Are you sure you want to exit?")
+            if exit:
+                console.print("Goodbye!", style="bold green")
+                return
+        else:
+            response: ChatGPTResponse = query_chatgpt(prompt=prompt)
+            message = response.choices[0].message.content.strip()
+            console.print(f"ChatGPT: {message}", style="bold cyan")
 
 
 def main():

pyproject.toml

@@ -22,7 +22,8 @@ authors = [
 dependencies = [
     "openai>=0.27.0",
     "typer[all]>=0.7.0",
-    "pydantic[dotenv]>=1.10.5"
+    "pydantic[dotenv]>=1.10.5",
+    "rich>=12.6.0",
 ]
 requires-python = ">=3.11"
 license = {text = "MIT"}