chatbot.py

import openai
import click
import os

# Set your API key from the environment variable
openai.api_key = os.getenv("OPENAI_API_KEY")

@click.command()
@click.option('--prompt', prompt="Enter your message", help="Your message to the AI Chatbot")
def chat(prompt):
    """
    A simple AI Chatbot that responds to your message using OpenAI's API.
    """
    try:
        response = openai.Completion.create(
            engine="text-davinci-003",
            prompt=prompt,
            max_tokens=150,
            n=1,
            stop=None,
            temperature=0.7,
        )
        message = response.choices[0].text.strip()
        click.echo(f"AI: {message}")
    except Exception as e:
        click.echo(f"Error: {e}")

if __name__ == "__main__":
    chat()