import warnings; warnings.filterwarnings("ignore")
import numpy as np
np.random.seed(42)
# We turn off Python warnings for a cleaner learning experience

# If you have not installed the openai package, use this command in your terminal:
print('pip install openai')

# You will not run this in Python, but in your command line or Anaconda prompt

import os
from openai import OpenAI

# Now you have the tools and modules needed for OpenAI access

# Securely load your API key from the system environment
API_key = os.environ["OPENAI_API_KEY"]

# This way, your private key is not shown directly in your code.

# Initialize the OpenAI client for API calls
client = OpenAI(api_key=API_key)

# This client is your gateway to talk with OpenAIs models

# Let us try a basic chat completion using the OpenAI API
response = client.chat.completions.create(
    model='gpt-4.1-mini',
    messages=[{"role": "user", "content": "Hello!"}]
)
print(response.choices[0].message.content)

# Let us send a custom question to the chat model
user_message = input("Type your question for the AI: ")

chat_response = client.chat.completions.create(
    model='gpt-4.1-mini',
    messages=[{"role": "user", "content": user_message}]
)
print(chat_response.choices[0].message.content)

# Using the Responses endpoint for direct completions
resp = client.responses.create(
    model='gpt-4.1-mini',
    input='Explain embeddings in one sentence.'
)
print(resp.output[0].content[0].text)

# Let us see how embeddings work with the OpenAI API
embed = client.embeddings.create(
    model='text-embedding-3-small',
    input=['Machine learning is fun']
)
print(len(embed.data[0].embedding))

# Simple error handling: What if something goes wrong?
try:
    bad_client = OpenAI(api_key="wrong_key")
    bad_client.chat.completions.create(
        model='gpt-4.1-mini',
        messages=[{"role": "user", "content": "Hello"}]
    )
except Exception as e:
    print("An error happened:", e)

# Mini Project: Make the AI a helpful chatbot
def simple_chat():
    print("Type 'bye' to stop chatting.")
    while True:
        user = input("You: ")
        if user.lower() == "bye":
            print("Chat ended.")
            break
        reply = client.chat.completions.create(
            model='gpt-4.1-mini',
            messages=[{"role": "user", "content": user}]
        )
        print("AI:", reply.choices[0].message.content)

simple_chat()

# Troubleshooting: Common issues with the OpenAI API
try:
    response = client.chat.completions.create(
        model='gpt-4.1-mini',
        messages=[{"role": "user", "content": "Test"}]
    )
    print("API is working!")
except Exception as error:
    print("If you see an error, check these:")
    print("Is your API key correct?")
    print("Are you connected to the internet?")
    print("Are you using the right model name?")
    print('Error details:', error)

# Challenge Exercise 1: Try sending your own creative prompt
prompt = input("Give the AI a creative task to do: ")
result = client.chat.completions.create(
    model='gpt-4.1-mini',
    messages=[{"role": "user", "content": prompt}]
)
print(result.choices[0].message.content)

# Challenge Exercise 2: Experiment with another endpoint
sentence = input("Give me a sentence to turn into an embedding: ")
em = client.embeddings.create(
    model='text-embedding-3-small',
    input=[sentence]
)
print("Vector length:", len(em.data[0].embedding))