Einen Chatbot erstellen

import openai
from dotenv import load_dotenv
import os

load_dotenv()
client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

def chat(user_message):
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": user_message}
        ],
        temperature=0.7,
        max_tokens=500
    )
    return response.choices[0].message.content

# Main loop
print("Chatbot ready! Type 'quit' to exit.")
while True:
    user_input = input("You: ")
    if user_input.lower() == 'quit':
        break
    response = chat(user_input)
    print(f"Bot: {response}")

class ChatBot:
    def __init__(self, system_prompt="You are a helpful assistant."):
        self.client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
        self.conversation_history = [
            {"role": "system", "content": system_prompt}
        ]
    
    def chat(self, user_message):
        self.conversation_history.append(
            {"role": "user", "content": user_message}
        )
        
        response = self.client.chat.completions.create(
            model="gpt-4",
            messages=self.conversation_history,
            temperature=0.7,
            max_tokens=500
        )
        
        assistant_message = response.choices[0].message.content
        self.conversation_history.append(
            {"role": "assistant", "content": assistant_message}
        )
        
        # Keep only last 20 messages to manage token usage
        if len(self.conversation_history) > 21:
            self.conversation_history = (
                [self.conversation_history[0]]  # Keep system prompt
                + self.conversation_history[-20:]
            )
        
        return assistant_message

# Customer support bot
support_bot = ChatBot(
    system_prompt="""You are a friendly customer support agent 
for a tech company. You are patient, empathetic, and always 
try to resolve issues step by step. If you don't know something, 
say so honestly and offer to escalate to a human agent."""
)

# Coding tutor bot
tutor_bot = ChatBot(
    system_prompt="""You are a Python programming tutor. 
Explain concepts clearly using simple language and examples. 
When the user makes a mistake, guide them to the answer 
instead of giving it directly. Use code examples frequently."""
)

import time

def chat_with_retry(self, user_message, max_retries=3):
    for attempt in range(max_retries):
        try:
            return self.chat(user_message)
        except openai.RateLimitError:
            wait_time = 2 ** attempt
            print(f"Rate limited. Waiting {wait_time}s...")
            time.sleep(wait_time)
        except openai.APIError as e:
            print(f"API error: {e}")
            if attempt == max_retries - 1:
                return "Sorry, I'm having trouble right now."
    return "Please try again later."