ai-tutor-chatbot / scripts /basic_tutor.py
Louis-François Bouchard
basic ai tutor QA script
592ba10
raw
history blame
1.85 kB
import sys
import os
from openai import OpenAI
# Retrieve your OpenAI API key from the environment variables and activate the OpenAI client
openai_api_key = os.environ.get('OPENAI_API_KEY')
client = OpenAI(api_key=openai_api_key)
def ask_ai_tutor(question):
# Check if OpenAI key has been correctly added
if not openai_api_key:
return "OpenAI API key not found in environment variables."
try:
# Formulating the system prompt
system_prompt = (
"You are an AI tutor specialized in answering artificial intelligence-related questions. "
"Only answer AI-related question, else say that you cannot answer this question."
)
# Combining the system prompt with the user's question
prompt = f"Please provide an informative and accurate answer to the following question.\nQuestion: {question}\nAnswer:"
# Call the OpenAI API
response = client.chat.completions.create(
model='gpt-3.5-turbo-16k',
messages=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": prompt}
]
)
# Return the AI's response
return response.choices[0].message.content.strip()
except Exception as e:
return f"An error occurred: {e}"
def main():
# Check if a question was provided as a command-line argument
if len(sys.argv) != 2:
print("Usage: python script_name.py 'Your AI-related question'")
sys.exit(1)
# The user's question is the first command-line argument
user_question = sys.argv[1]
# Get the AI's response
ai_response = ask_ai_tutor(user_question)
# Print the AI's response
print(f"AI Tutor says: {ai_response}")
if __name__ == "__main__":
main()