import_openai

import openai
from transformers import GPT2LMHeadModel, GPT2Tokenizer

openai.api_key = "***********"

model_name = "gpt3.5-turbo"
model = GPT2LMHeadModel.from_pretrained(model_name)
tokenizer = GPT2Tokenizer.from_pretrained(model_name)

def generate_response(user_input):
    input_ids = tokenizer.encode(user_input, return_tensors='pt')
    output = model.generate(input_ids, max_length=100)
    response = tokenizer.decode(output[0], skip_special_tokens=True)
    return response