{% extends 'base.html' %}

{% block title %}我的模型{% endblock %}

{% block content %}
<div class="container mt-5">
    <div class="row justify-content-center">
        <div class="col-md-10">
            <div class="card shadow-sm">
                <div class="card-header bg-success text-white">
                    <h2 class="mb-0 text-center">我的模型</h2>
                </div>
                <div class="card-body">
                    {% if models %}
                        <table class="table table-hover">
                            <thead class="table-dark">
                                <tr>
                                    <th>模型名称</th>
                                    <th>上传时间</th>
                                    <th>操作</th>
                                </tr>
                            </thead>
                            <tbody>
                                {% for model in models %}
                                <tr>
                                    <td>{{ model.file.name }}</td>
                                    <td>{{ model.uploaded_at }}</td>
                                    <td>
                                        <a href="{% url 'edit_model' model.id %}" class="btn btn-warning btn-sm">
                                            <i class="fas fa-edit"></i> 编辑
                                        </a>
                                        <button class="btn btn-danger btn-sm delete-model" data-id="{{ model.id }}">
                                            <i class="fas fa-trash"></i> 删除
                                        </button>
                                    </td>
                                </tr>
                                {% endfor %}
                            </tbody>
                        </table>
                    {% else %}
                        <p class="text-muted text-center">你还没有上传任何模型。</p>
                    {% endif %}
                </div>
            </div>

            <div class="text-center mt-4">
                <a href="{% url 'upload_model' %}" class="btn btn-primary">
                    <i class="fas fa-upload"></i> 上传新模型
                </a>
            </div>
        </div>
    </div>
</div>

<!-- 删除模型的 JavaScript 代码 -->
<script>
document.addEventListener("DOMContentLoaded", function () {
    const deleteButtons = document.querySelectorAll(".delete-model");

    deleteButtons.forEach(button => {
        button.addEventListener("click", function () {
            const modelId = this.getAttribute("data-id");
            if (confirm("确定要删除这个模型吗？")) {
                fetch(`/delete_model/${modelId}/`, {
                    method: "POST",
                    headers: {
                        "X-CSRFToken": "{{ csrf_token }}",
                        "Content-Type": "application/json"
                    },
                    body: JSON.stringify({})
                })
                .then(response => response.json())
                .then(data => {
                    if (data.status === "success") {
                        alert("模型已删除！");
                        location.reload();
                    } else {
                        alert("删除失败：" + data.message);
                    }
                })
                .catch(error => console.error("删除出错:", error));
            }
        });
    });
});
</script>

{% endblock %}