import warnings; warnings.filterwarnings("ignore")
import numpy as np
np.random.seed(42)
# This line disables unwanted warning messages.

import os
# os gives access to system settings, like environment variables.

API_key = os.environ["OPENAI_API_KEY"]
# Fetch the OpenAI API key from your environment settings.

from openai import OpenAI
# Import the main OpenAI client library.

client = OpenAI()
# Set up a client instance, ready for requests.

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

# You can control the conversation with roles.
messages = [
    {"role": "system", "content": "You are a helpful Python tutor."},
    {"role": "user", "content": "What is a variable?"}
]
response = client.chat.completions.create(
    model="gpt-4.1-mini",
    messages=messages
)
print(response.choices[0].message.content)

# You can send longer multi-turn conversations.
chat_history = [
    {"role": "system", "content": "You are a math assistant."},
    {"role": "user", "content": "What is 2 plus 2?"},
    {"role": "assistant", "content": "2 plus 2 is 4."},
    {"role": "user", "content": "What is 3 times 5?"}
]
resp = client.chat.completions.create(
    model="gpt-4.1-mini",
    messages=chat_history
)
print(resp.choices[0].message.content)

# Request text embeddings and inspect the result.
embed = client.embeddings.create(
    model="text-embedding-3-small",
    input=["Machine learning is fun"]
)
print(len(embed.data[0].embedding))

# Wrap your API call with error handling.
try:
    response = client.chat.completions.create(
        model="gpt-4.1-mini",
        messages=[{"role": "user", "content": "Will not crash!"}]
    )
    print(response.choices[0].message.content)
except Exception as e:
    print("Oops, something went wrong:", str(e))

# Simple user prompt with input()
user_query = input("What would you like to ask the AI? ")
result = client.chat.completions.create(
    model="gpt-4.1-mini",
    messages=[{"role": "user", "content": user_query}]
)
print(result.choices[0].message.content)

# Mini Project: Basic Chatbot Loop
while True:
    user_msg = input("You: ")
    if user_msg.lower().strip() in ["exit", "quit"]:
        print("Goodbye!")
        break
    reply = client.chat.completions.create(
        model="gpt-4.1-mini",
        messages=[{"role": "user", "content": user_msg}]
    )
    print("AI:", reply.choices[0].message.content)

# Troubleshooting: Check for common issues.
try:
    check = client.chat.completions.create(
        model="gpt-4.1-mini",
        messages=[{"role": "user", "content": "Test connection"}]
    )
    print("API working fine.")
except Exception as error:
    print("Check your API key or network. Error:", error)