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

# ChatMessage Inline for ChatSession
class ChatMessageInline(admin.TabularInline):
    model = ChatMessage
    extra = 0
    readonly_fields = ("is_user", "message", "timestamp")

# ChatSession Admin Configuration
@admin.register(ChatSession)
class ChatSessionAdmin(admin.ModelAdmin):
    list_display = ('session_key', 'user', 'created_at')
    search_fields = ('session_key', 'user__username')
    list_filter = ('created_at',)
    inlines = [ChatMessageInline]  # Including ChatMessage Inline

# ChatMessage Admin Configuration
@admin.register(ChatMessage)
class ChatMessageAdmin(admin.ModelAdmin):
    list_display = ('session', 'is_user', 'message', 'timestamp')
    search_fields = ('message', 'session__session_key')
    list_filter = ('timestamp', 'is_user')