新增模型补全空洞

client/core/net/ModelServerClient.cpp

@@ -87,6 +87,47 @@ QNetworkReply* ModelServerClient::segmentSamPromptAsync(
     return m_nam->post(req, body);
 }
 
+QNetworkReply* ModelServerClient::inpaintAsync(
+    const QByteArray& cropRgbPngBytes,
+    const QByteArray& maskPngBytes,
+    const QString& prompt,
+    const QString& negativePrompt,
+    double strength,
+    int maxSide,
+    QString* outImmediateError
+) {
+    if (outImmediateError) {
+        outImmediateError->clear();
+    }
+    if (!m_baseUrl.isValid() || m_baseUrl.isEmpty()) {
+        if (outImmediateError) *outImmediateError = QStringLiteral("后端地址无效。");
+        return nullptr;
+    }
+    if (cropRgbPngBytes.isEmpty()) {
+        if (outImmediateError) *outImmediateError = QStringLiteral("裁剪图像为空。");
+        return nullptr;
+    }
+    if (maskPngBytes.isEmpty()) {
+        if (outImmediateError) *outImmediateError = QStringLiteral("Mask 为空。");
+        return nullptr;
+    }
+
+    const QUrl url = m_baseUrl.resolved(QUrl(QStringLiteral("/inpaint")));
+    QNetworkRequest req(url);
+    req.setHeader(QNetworkRequest::ContentTypeHeader, QStringLiteral("application/json"));
+
+    QJsonObject payload;
+    payload.insert(QStringLiteral("image_b64"), QString::fromLatin1(cropRgbPngBytes.toBase64()));
+    payload.insert(QStringLiteral("mask_b64"), QString::fromLatin1(maskPngBytes.toBase64()));
+    payload.insert(QStringLiteral("prompt"), prompt);
+    payload.insert(QStringLiteral("negative_prompt"), negativePrompt);
+    payload.insert(QStringLiteral("strength"), strength);
+    payload.insert(QStringLiteral("max_side"), maxSide);
+
+    const QByteArray body = QJsonDocument(payload).toJson(QJsonDocument::Compact);
+    return m_nam->post(req, body);
+}
+
 bool ModelServerClient::computeDepthPng8(
     const QByteArray& imageBytes,
     QByteArray& outPngBytes,

client/core/net/ModelServerClient.h

@@ -37,6 +37,16 @@ public:
         const QJsonArray& boxXyxy,
         QString* outImmediateError = nullptr);
 
+    // POST /inpaint，JSON 响应由调用方解析（success / output_image_b64 / error / output_path）。
+    QNetworkReply* inpaintAsync(
+        const QByteArray& cropRgbPngBytes,
+        const QByteArray& maskPngBytes,
+        const QString& prompt,
+        const QString& negativePrompt,
+        double strength,
+        int maxSide,
+        QString* outImmediateError = nullptr);
+
 private:
     QNetworkAccessManager* m_nam = nullptr;
     QUrl m_baseUrl;

client/core/workspace/ProjectWorkspace.cpp

@@ -1869,6 +1869,69 @@ bool ProjectWorkspace::resolveBlackholeByCopyBackground(const QString& id, const
     return true;
 }
 
+bool ProjectWorkspace::resolveBlackholeByModelInpaint(const QString& id, const QImage& patchedBackground,
+                                                      bool hideBlackholeAfterFill) {
+    if (m_projectDir.isEmpty() || id.isEmpty()) {
+        return false;
+    }
+    const QString bgAbs = backgroundAbsolutePath();
+    if (bgAbs.isEmpty() || !QFileInfo::exists(bgAbs)) {
+        return false;
+    }
+    if (patchedBackground.isNull()) {
+        return false;
+    }
+
+    // 写回背景文件
+    {
+        QImage bg = patchedBackground;
+        if (bg.format() != QImage::Format_ARGB32_Premultiplied) {
+            bg = bg.convertToFormat(QImage::Format_ARGB32_Premultiplied);
+        }
+        QImageWriter writer(bgAbs);
+        writer.setFormat("png");
+        writer.setCompression(1);
+        if (!writer.write(bg)) {
+            return false;
+        }
+    }
+
+    // 更新实体黑洞状态 + 记录历史
+    const auto before = m_project.entities();
+    auto ents = before;
+    int hit = -1;
+    for (int i = 0; i < ents.size(); ++i) {
+        if (ents[i].id == id) {
+            hit = i;
+            break;
+        }
+    }
+    if (hit < 0) {
+        return false;
+    }
+
+    ents[hit].blackholeVisible = hideBlackholeAfterFill ? false : ents[hit].blackholeVisible;
+    if (ents[hit].blackholeId.isEmpty()) {
+        ents[hit].blackholeId = QStringLiteral("blackhole-%1").arg(ents[hit].id);
+    }
+    ents[hit].blackholeResolvedBy = QStringLiteral("model_inpaint");
+
+    m_project.setEntities(ents);
+    if (!saveSingleEntityPayload(ents[hit]) || !writeIndexJsonWithoutPayloadSync()) {
+        m_project.setEntities(before);
+        return false;
+    }
+
+    Operation op;
+    op.type = Operation::Type::SetEntities;
+    op.label = QStringLiteral("黑洞模型补全");
+    op.beforeEntities = before;
+    op.afterEntities = ents;
+    pushOperation(op);
+    m_redoStack.clear();
+    return true;
+}
+
 bool ProjectWorkspace::setEntityVisibilityKey(const QString& id, int frame, bool visible) {
     if (m_projectDir.isEmpty() || id.isEmpty() || frame < 0) {
         return false;

client/core/workspace/ProjectWorkspace.h

@@ -101,6 +101,9 @@ public:
     // 复制背景其他区域填充黑洞（sourceOffsetPx 以黑洞包围盒左上角为基准偏移）
     bool resolveBlackholeByCopyBackground(const QString& id, const QPoint& sourceOffsetPx,
                                           bool hideBlackholeAfterFill);
+    // 使用模型补全后的结果写回背景（patchedBackground 已包含补全贴合后的完整背景图像）
+    bool resolveBlackholeByModelInpaint(const QString& id, const QImage& patchedBackground,
+                                        bool hideBlackholeAfterFill);
     bool setEntityVisibilityKey(const QString& id, int frame, bool visible);
     bool removeEntityVisibilityKey(const QString& id, int frame);
     bool setEntityDisplayName(const QString& id, const QString& displayName);