from django.contrib import admin
from .models import Bot, ChatMessage

@admin.register(Bot)
class BotAdmin(admin.ModelAdmin):
    list_display = ('id', 'name', 'description')
    search_fields = ('id', 'name', 'description')


@admin.register(ChatMessage)
class BotAdmin(admin.ModelAdmin):
    list_display = ('id', 'user', 'bot', 'message', 'timestamp', 'sender', 'message_type')
    search_fields = ('id', 'user', 'bot', 'message', 'timestamp', 'sender', 'message_type')