Skip to main content

How to Code AI with Python and OpenAI API: A Step-by-Step Guide

· 9 min read
Jesus Paz
Python Expert & AI Indie Hacker

🤖✨ Create Your Own AI Magic with Python and OpenAI API! ✨🤖

Have you ever dreamed of having your own smart assistant that can answer questions, tell stories, or help you with homework? With Python and OpenAI API, you can make that dream come true! This easy guide will show you how to use AI without needing a supercomputer. Whether you’re new to Python or AI, you’ll find this guide simple and fun. Let’s get started!

📚 What You’ll Learn

  • What AI is and why it needs a lot of computing power.
  • How to use OpenAI’s API with Python.
  • How to set up your OpenAI API key and write code to talk to AI.

🤔 What is AI and Why Does It Need So Much Power?

Artificial Intelligence (AI) helps machines learn from data and make smart decisions. Imagine teaching a robot to recognize your friends’ faces or to understand and write stories! The bigger and smarter the AI, the more computing power it needs to work well.

🧠 Large Language Models (LLMs)

Large Language Models (LLMs) like GPT-4 are super smart AIs that can understand and create human-like text. They’re like having a conversation with a knowledgeable friend! But, these models need a lot of data and powerful computers to run smoothly.

💻 Using AI Locally vs. API

Running AI models on your own computer can be tricky because they need a lot of power. Instead, you can use OpenAI’s API to ask questions and get answers from powerful AI models over the internet. This way, you don’t need a supercomputer—just your regular computer and an internet connection!

🌟 OpenAI Models Overview

OpenAI offers different AI models, each with its own strengths and costs. Here’s a simple overview:

ModelDescription
GPT-4oSuper smart model for complex tasks.
GPT-4o miniSmaller and cheaper model for quick tasks.
o1-preview/miniSpecial models trained for tougher problems.
GPT-4 TurboOlder but still very powerful version of GPT models.

For more details, visit OpenAI's Model Documentation.

🎯 Choosing the Right Model

For beginners, GPT-4o mini is a great choice because it’s affordable and perfect for learning. A single request with this model costs less than $0.001 (1/10th of a cent)! It’s a smart way to experiment without spending much money. Check out OpenAI's Quickstart Documentation for more info.

💰 Understanding OpenAI Pricing

Using OpenAI’s API costs money based on the number of tokens you use. Tokens are like pieces of words:

  • Input tokens: The words you send to the AI.
  • Output tokens: The words the AI sends back to you.

Here’s the current pricing for GPT-4o mini:

ModelPrice per 1M Input TokensPrice per 1M Output Tokens
GPT-4o mini$0.150$0.600
GPT-4o mini (Batch)$0.075$0.300

A simple request using around 1,000 tokens costs less than $0.001! For the latest prices, visit OpenAI’s pricing page.

note

Tip: Batch API pricing gives you a 50% discount if you send multiple requests together and wait up to 24 hours for the response.

🔑 Setting Up OpenAI API

To start coding with OpenAI’s API, you need an API key. Here’s how to get one:

  1. Go to the OpenAI API Key Page.
  2. Sign up or log in to your OpenAI account.
  3. Find the API Keys section in your dashboard.
  4. Click on "Create New Key" to generate your API key.

Once you have your key, you’re ready to start coding!

📝 Step-by-Step Guide: Interacting with OpenAI’s API Using Python

Let’s write some Python code to talk to OpenAI’s AI. Follow these easy steps!

🛠️ Step 1: Install the Required Libraries

First, you need to install the OpenAI library. Open your terminal or command prompt and type:

pip install openai

💻 Step 2: Writing the Python Script

Create a new Python file called openai_query.py and open it in your favorite editor. Let’s start coding!

📥 Importing the Libraries

We need to import the OpenAI library and set up our API key.

import openai
import os

# Replace 'your-api-key-here' with your actual OpenAI API key
openai.api_key = 'your-api-key-here'

Tip: For extra security, store your API key in an environment variable instead of writing it directly in the code.

import openai
import os

# Load your API key from an environment variable
openai.api_key = os.getenv("OPENAI_API_KEY")

Sending a Simple Query

Let’s ask the AI a simple question: "What is recursion in programming?"

def ask_openai(question):
response = openai.Completion.create(
model="gpt-4o-mini", # Using the affordable GPT-4o mini model
prompt=question,
max_tokens=100
)
return response.choices[0].text.strip()

question = "What is recursion in programming?"
answer = ask_openai(question)
print(f"Question: {question}")
print(f"Answer: {answer}")

Output:

Question: What is recursion in programming?
Answer: Recursion in programming is a technique where a function calls itself in order to solve smaller instances of the same problem. It helps in breaking down complex problems into simpler ones.

💬 Step 3: Handling Multiple Queries

Let’s make it more interactive! We can have a conversation with the AI by handling multiple questions.

def chat_with_openai(queries):
conversation = []
for query in queries:
response = openai.Completion.create(
model="gpt-4o-mini",
prompt=query,
max_tokens=100
)
ai_response = response.choices[0].text.strip()
conversation.append({"role": "user", "content": query})
conversation.append({"role": "assistant", "content": ai_response})
print(f"User: {query}")
print(f"AI: {ai_response}\n")

queries = [
"What is AI?",
"How does machine learning work?",
"Can AI solve complex problems?"
]
chat_with_openai(queries)

Output:

User: What is AI?
AI: Artificial Intelligence (AI) is the simulation of human intelligence in machines that are programmed to think and learn like humans. It includes capabilities like understanding language, recognizing patterns, and making decisions.

User: How does machine learning work?
AI: Machine learning works by training algorithms on large amounts of data. The algorithms learn patterns and make predictions or decisions based on new data they encounter, improving their accuracy over time.

User: Can AI solve complex problems?
AI: Yes, AI can solve complex problems by analyzing large datasets, identifying patterns, and making informed decisions. It is used in various fields like healthcare, finance, and technology to tackle challenges that are difficult for humans to solve alone.

🎨 Step 4: Customizing the AI’s Behavior

You can make the AI more creative or concise by adjusting settings like temperature and max_tokens.

def ask_openai_custom(question, temperature=0.7, max_tokens=150):
response = openai.Completion.create(
model="gpt-4o-mini",
prompt=question,
temperature=temperature, # Higher value makes responses more creative
max_tokens=max_tokens # Controls the response length
)
return response.choices[0].text.strip()

question = "Explain AI in simple terms."
answer = ask_openai_custom(question, temperature=0.5, max_tokens=100)
print(f"Answer: {answer}")

Output:

Answer: AI, or Artificial Intelligence, is like giving computers a brain. It helps them learn from information and make decisions, similar to how humans think and solve problems.

🛡️ Step 5: Protecting Your API Key

Keep your API key safe by storing it in environment variables. Here’s how:

  1. Set an Environment Variable:

    • On Windows:

    set OPENAI_API_KEY=your-api-key-here

    - On **macOS/Linux**:
    ```bash
    export OPENAI_API_KEY=your-api-key-here
  2. Update Your Python Script:

    import openai
    import os

    # Load your API key from an environment variable
    openai.api_key = os.getenv("OPENAI_API_KEY")

Never share your API key or include it directly in your code, especially if you’re sharing your project online.

🌟 Best Practices for Coding AI with OpenAI API

To make the most out of your AI projects, follow these simple tips:

1. Limit API Requests

Avoid high costs by limiting how often your app can call the OpenAI API. Set a budget and keep track of your usage.

2. Choose the Right Model

Use models like GPT-4o mini for a balance between cost and performance. They’re affordable and still very smart!

3. Protect Your API Key

Always keep your API key safe by using environment variables or a secrets manager. Never share it or put it directly in your code.

4. Monitor Token Usage

Keep an eye on how many tokens your app uses. This helps you stay within your budget and understand your usage better.

response = openai.Completion.create(
model="gpt-4o-mini",
prompt="Your prompt here",
max_tokens=100
)

print(f"Total Tokens Used: {response['usage']['total_tokens']}")

Frequently Asked Questions (FAQs)

Q1: What is AI?

A: AI stands for Artificial Intelligence. It helps machines learn from data and make smart decisions, like recognizing speech or playing games.

Q2: Why do AI models need so much power?

A: AI models, especially large ones like GPT-4, need a lot of computing power to process and understand huge amounts of data to make accurate predictions and responses.

Q3: Can I use OpenAI’s API for free?

A: OpenAI offers a free tier with limited usage. Check out their pricing page for more details on costs and plans.

Q4: How do I keep my API key safe?

A: Store your API key in environment variables or use a secrets manager. Never share it or include it directly in your code.

Q5: What can I build with OpenAI’s API?

A: You can create chatbots, virtual assistants, content generators, language translators, and much more! Explore your creativity and see what you can build.

🎉 Conclusion

Coding AI with Python and OpenAI API is easier than you think! By following this step-by-step guide, you’ve learned how to set up your environment, write simple scripts, and interact with powerful AI models. Whether you’re building a chatbot, generating stories, or answering questions, AI can make your projects smarter and more fun.

Ready to take your AI skills to the next level? Explore more tutorials like Integrating AI with Flask and How to Fine-Tune OpenAI Models in Python to keep learning and building amazing AI-powered applications!


Looking for more? Visit OpenAI's official Quickstart Guide to explore additional resources.

Happy Coding! 🚀🤖