Demo Building a Simple Chatbot

from pydantic import BaseModel
from agents import Agent

class CalendarEvent(BaseModel):
    name: str
    date: str
    participants: list[str]

agent = Agent(
    name="Calendar extractor",
    instructions="Extract calendar events from text",
    output_type=CalendarEvent,
)

# chatbot_agent.py
from dotenv import load_dotenv
import os
import asyncio
from agents import Agent, Runner

# Load environment variables from .env
load_dotenv()

if not os.getenv("OPENAI_API_KEY"):
    raise RuntimeError("OPENAI_API_KEY not found in environment. Add it to your .env file.")

# Define the chatbot agent
agent = Agent(
    name="Police Sketch Artist",
    instructions=(
        "You are a police sketch artist. Collect specific details about the individual being sketched. "
        "Ask follow-up questions to clarify features (hair, eyes, nose, mouth, clothing, build, distinctive marks)."
    ),
)

async def main():
    chat_history: list[tuple[str, str]] = []

    print("Police Sketch Artist chatbot. Type 'history' to view conversation, 'exit' or 'quit' to end.\n")

    while True:
        user_input = input("Provide specific details about the individual: ").strip()
        if not user_input:
            continue

        # Exit commands
        if user_input.lower() in ("exit", "quit"):
            print("Goodbye!")
            break

        # Show chat history
        if user_input.lower() == "history":
            print("\n--- Chat History ---")
            if not chat_history:
                print("(no messages yet)")
            for i, (u, b) in enumerate(chat_history, start=1):
                print(f"{i}. You: {u}\n   ChatBot: {b}\n")
            print("---------------\n")
            continue

        # Run the agent and collect the response
        result = await Runner.run(agent, user_input)

        # The agent's response may be in result.final_output or result.output depending on SDK behavior
        response = getattr(result, "final_output", None)
        if response is None:
            response = getattr(result, "output", str(result))

        # Save to history and display
        chat_history.append((user_input, response))
        print("\nChatBot:", response)
        print("To end chat, type 'exit' or 'quit'. Type 'history' to view past conversations.\n")

if __name__ == "__main__":
    asyncio.run(main())