#!/usr/bin/env python3

import argparse
from chat import chat_with_qwen

def main():
    parser = argparse.ArgumentParser(description="Interact with the Qwen model.")
    parser.add_argument("-p", "--prompt", type=str, help="The prompt to send to the Qwen model.", required=True)
    parser.add_argument("-t", "--temperature", type=float, default=0.7, help="The temperature to use for sampling.")
    parser.add_argument("-m", "--max_tokens", type=int, default=256, help="The maximum number of tokens to generate.")

    args = parser.parse_args()

    response = chat_with_qwen(args.prompt, temperature=args.temperature, max_tokens=args.max_tokens)
    print(response)

if __name__ == "__main__":
    main()