@if (count($messages) > 0)
    @foreach ($messages as $message)
        @if ($message->user_type == 'user' && $message->user_id == \Illuminate\Support\Facades\Auth::user()->id)
            <div class="pb-4 chat-message-right" id="message-text-{{ $message->id }}">

                <div class="mt-2 text-muted small text-nowrap">
                    {{ \Carbon\Carbon::createFromFormat('Y-m-d H:i:s', $message->created_at) }}</div>

                <div class="px-3 py-2 mr-3 rounded flex-shrink-1 bg-light">
                    <div class="mb-1 font-weight-bold">You</div>
                    <span> {{ $message->message }}</span>
                    @if ($message->file_url)
                        <br>
                        @if (in_array(pathinfo($message->file_url, PATHINFO_EXTENSION), ['jpg', 'jpeg', 'png', 'gif']))
                            <!-- Display image if it's an image file -->
                            <img src="{{ asset('storage/' . $message->file_url) }}" alt="attachment" class="img-fluid mt-2 chat-image" onclick="openImage(this)">
                        @else
                            <!-- Otherwise display as download link -->
                            <a href="{{ asset('storage/' . $message->file_url) }}" target="_blank">Download Attachment</a>
                        @endif
                    @endif
                </div>

                <!-- Edit and Delete options for current user -->
                @if ($message->user_id == Auth::id())
                    <div class="text-right">
                        <button class="btn btn-sm btn-outline-secondary" data-edit-message="{{ $message->id }}">Edit</button>
                        <button class="btn btn-sm btn-outline-danger" data-delete-message="{{ $message->id }}">Delete</button>
                    </div>
                @endif
            </div>
        @else
            <div class="pb-4 chat-message-left" id="message-text-{{ $message->id }}">

                <div class="mt-2 text-muted small text-nowrap">
                    {{ \Carbon\Carbon::createFromFormat('Y-m-d H:i:s', $message->created_at) }}</div>

                <div class="px-3 py-2 ml-3 rounded flex-shrink-1 bg-light">
                    <div class="mb-1 font-weight-bold">
                        {{ $message->user_type === 'teacher' ? $message->teacher->name : $message->student->name }}
                    </div>
                    <span> {{ $message->message }}</span>
                    @if ($message->file_url)
                        <br>
                        @if (in_array(pathinfo($message->file_url, PATHINFO_EXTENSION), ['jpg', 'jpeg', 'png', 'gif']))
                            <!-- Display image if it's an image file -->
                            <img src="{{ asset('storage/' . $message->file_url) }}" alt="attachment" class="img-fluid mt-2 chat-image" onclick="openImage(this)">
                        @else
                            <!-- Otherwise display as download link -->
                            <a href="{{ asset('storage/' . $message->file_url) }}" target="_blank">Download Attachment</a>
                        @endif
                    @endif
                </div>
            </div>
        @endif
    @endforeach
@endif