chat.py

import os
from dotenv import load_dotenv
import openai

load_dotenv()
openai.api_key = os.environ.get('OPENAI_API_KEY')
completion = openai.ChatCompletion()

start_chat_log = [
    {"role": "system", "content": "You are a helpful assistant."},
]


def askgpt(question, chat_log=None):
    if chat_log is None:
        chat_log = start_chat_log
    chat_log = chat_log + [{'role': 'user', 'content': question}]
    response = completion.create(model='gpt-3.5-turbo', messages=chat_log)
    answer = response.choices[0]['message']['content']
    chat_log = chat_log + [{'role': 'assistant', 'content': answer}]
    return answer, chat_log

# Define a function to ask questions
# def ask_question(question):
#     # Provide your input and prompt
#     prompt = f'Question: {question}\nAnswer:'

#     # Generate a completion using OpenAI's ChatGPT
#     response = openai.Completion.create(
#         # engine='text-davinci-003',test
#         engine='gpt-3.5-turbo',
#         prompt=prompt,
#         max_tokens=100,
#         temperature=0.7,
#         n=1,
#         stop=None,
#     )

#     # Extract the answer from the generated completion
#     answer = response.choices[0].text.strip()

#     return answer

# Ask a question and print the response
question = input("Ask a question: ")
answer = askgpt(question)
print(answer)