1

server-rs/crates/spacetime-module/src/puzzle.rs

@@ -685,7 +685,7 @@ fn save_puzzle_generated_images_tx(
     if candidates.is_empty() {
         return Err("拼图候选图不能为空".to_string());
     }
-    append_generated_candidates(&mut draft, candidates);
+    replace_generated_candidate(&mut draft, candidates);
     draft.generation_status = "ready".to_string();
     if let Some(selected) = draft
         .candidates
@@ -724,7 +724,7 @@ fn save_puzzle_generated_images_tx(
             stage: next_stage,
             anchor_pack_json: row.anchor_pack_json.clone(),
             draft_json: Some(serialize_json(&draft)),
-            last_assistant_reply: Some("候选图已经生成，请选择正式拼图图片。".to_string()),
+            last_assistant_reply: Some("拼图图片已经生成，并已替换当前正式图。".to_string()),
             published_profile_id: row.published_profile_id.clone(),
             created_at: row.created_at,
             updated_at: saved_at,
@@ -1510,21 +1510,19 @@ fn increment_puzzle_profile_play_count(
     );
 }
 
-fn append_generated_candidates(
+fn replace_generated_candidate(
     draft: &mut PuzzleResultDraft,
     candidates: Vec<PuzzleGeneratedImageCandidate>,
 ) {
-    let has_selected_candidate = draft.candidates.iter().any(|entry| entry.selected);
-    // 再次生成图片是扩充候选池，不覆盖创作者已经看到或已经选择的候选图。
-    // 若已有正式选择，新追加候选图保持未选中，避免同一草稿出现多个 selected。
-    draft
-        .candidates
-        .extend(candidates.into_iter().map(|mut candidate| {
-            if has_selected_candidate {
-                candidate.selected = false;
-            }
+    // 结果页生图采用单图替换：每次只保留最新图片，并立即作为正式图。
+    draft.candidates = candidates
+        .into_iter()
+        .take(1)
+        .map(|mut candidate| {
+            candidate.selected = true;
             candidate
-        }));
+        })
+        .collect();
 }
 
 fn list_published_puzzle_profiles(ctx: &TxContext) -> Result<Vec<PuzzleWorkProfile>, String> {
@@ -1634,7 +1632,7 @@ mod tests {
     }
 
     #[test]
-    fn puzzle_generated_images_are_appended_without_clearing_existing_candidates() {
+    fn puzzle_generated_images_replace_existing_candidate() {
         let anchor_pack = infer_anchor_pack("蒸汽城市雨夜猫咪", Some("蒸汽城市雨夜猫咪"));
         let mut draft = compile_result_draft(&anchor_pack, &[]);
         draft.candidates = vec![PuzzleGeneratedImageCandidate {
@@ -1647,7 +1645,7 @@ mod tests {
             selected: true,
         }];
 
-        append_generated_candidates(
+        replace_generated_candidate(
             &mut draft,
             vec![PuzzleGeneratedImageCandidate {
                 candidate_id: "session-1-candidate-2".to_string(),
@@ -1660,11 +1658,9 @@ mod tests {
             }],
         );
 
-        assert_eq!(draft.candidates.len(), 2);
-        assert_eq!(draft.candidates[0].candidate_id, "session-1-candidate-1");
+        assert_eq!(draft.candidates.len(), 1);
+        assert_eq!(draft.candidates[0].candidate_id, "session-1-candidate-2");
         assert!(draft.candidates[0].selected);
-        assert_eq!(draft.candidates[1].candidate_id, "session-1-candidate-2");
-        assert!(!draft.candidates[1].selected);
     }
 
     #[test]