新增二次编辑功能

README.md

@@ -75,7 +75,7 @@ pinned: false
 
 <br>
 
-## 前言
+# 前言
 
 很荣幸在这里介绍我的新项目ManimCat，它是~一只猫~
 
@@ -85,7 +85,7 @@ pinned: false
 
 用户只需输入自然语言描述，系统便会通过 AI 自动生成 Manim 代码并渲染出精美的数学可视化视频，支持 LaTeX 公式、模板化生成以及代码错误自动修复，让复杂概念的动态展示变得触手可及。
 
-## 样例
+# 样例
 
 ### 01
 
@@ -98,7 +98,7 @@ pinned: false
 </div>
 
 
-## 技术
+# 技术
 
 ### 技术栈
 
@@ -200,11 +200,11 @@ AI_TEMPERATURE=0.7
 CODE_RETRY_MAX_RETRIES=4
 ```
 
-## 部署
+# 部署
 
 请查看[部署文档](DEPLOYMENT.md)。
 
-## 贡献
+# 贡献
 
 我对原作品进行了比较大量的修改和重构，使其更符合我的设计想法：
 
@@ -295,7 +295,12 @@ CODE_RETRY_MAX_RETRIES=4
 
 </details>
 
-##  开源与版权声明 
+
+# 现状
+
+目前，该项目功能非常初级，最大的缺点是生成与渲染花费时间不小，非常慢。我正在试图解决这个问题！
+
+#  开源与版权声明 
 
 ### 1. 软件协议 
 本项目后端架构及前端部分实现参考/使用了 [manim-video-generator](https://github.com/rohitg00/manim-video-generator) 的核心思想。
@@ -319,9 +324,9 @@ CODE_RETRY_MAX_RETRIES=4
 
 > ManimCat 的诞生正是为了对标并挑战这些闭源商业软件。 我希望通过开源的方式，让每一位老师都能廉价地享受到 AI 带来的教学可视化便利————你只需要支付api的费用，幸运的是，对于优秀的中国LLM大模型来说，这些花费很廉价。为了保护这一愿景不被商业机构剽窃并反向收割用户，我坚决禁止任何对本项目核心提示词及索引数据的商业授权。
 
-## 维护说明
+# 维护说明
 
-由于作者精力有限（个人业余兴趣开发者，非专业背景），目前完全无法对外部代码进行有效的审查和长期维护。因此，本项目暂不支持团队协同开发，不接受 PR。感谢理解。
+由于作者精力有限（个人业余兴趣开发者，非专业背景），可能无法对外部代码进行有效的审查和长期维护。欢迎PR修复各类问题，不过过多的修改可能作者需要很长时间的审核。感谢理解。
 
 如果你有好的建议或发现了 Bug，欢迎提交 Issue 进行讨论，我会根据自己的节奏进行改进。如果你希望在本项目基础上进行大规模修改，欢迎 Fork 出属于你自己的版本。
 

frontend/src/App.tsx

@@ -1,197 +1,262 @@
-// 主应用组件
-
-import { useState } from 'react';
-import { useGeneration } from './hooks/useGeneration';
-import { InputForm } from './components/InputForm';
-import { LoadingSpinner } from './components/LoadingSpinner';
-import { ResultSection } from './components/ResultSection';
-import { ThemeToggle } from './components/ThemeToggle';
-import { SettingsModal } from './components/SettingsModal';
-import { PromptsManager } from './components/PromptsManager';
-import { DonationModal } from './components/DonationModal';
-import ManimCatLogo from './components/ManimCatLogo';
-import type { Quality } from './types/api';
-
-function App() {
-  const { status, result, error, jobId, stage, generate, reset, cancel } = useGeneration();
-  const [settingsOpen, setSettingsOpen] = useState(false);
-  const [promptsOpen, setPromptsOpen] = useState(false);
-  const [donationOpen, setDonationOpen] = useState(false);
-
-  const handleSubmit = (data: { concept: string; quality: Quality; forceRefresh: boolean }) => {
-    generate(data);
-  };
-
-  return (
-    <div className="min-h-screen bg-bg-primary transition-colors duration-300">
-      {/* 左上角图标 */}
-      <div className="fixed top-4 left-4 z-50 flex items-center gap-2">
-        <a
-          href="https://github.com/Wing900/ManimCat"
-          target="_blank"
-          rel="noopener noreferrer"
-          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
-          title="GitHub"
-        >
-          <svg className="w-5 h-5" fill="currentColor" viewBox="0 0 24 24">
-            <path fillRule="evenodd" d="M12 2C6.477 2 2 6.484 2 12.017c0 4.425 2.865 8.18 6.839 9.504.5.092.682-.217.682-.483 0-.237-.008-.868-.013-1.703-2.782.605-3.369-1.343-3.369-1.343-.454-1.158-1.11-1.466-1.11-1.466-.908-.62.069-.608.069-.608 1.003.07 1.531 1.032 1.531 1.032.892 1.53 2.341 1.088 2.91.832.092-.647.35-1.088.636-1.338-2.22-.253-4.555-1.113-4.555-4.951 0-1.093.39-1.988 1.029-2.688-.103-.253-.446-1.272.098-2.65 0 0 .84-.27 2.75 1.026A9.564 9.564 0 0112 6.844c.85.004 1.705.115 2.504.337 1.909-1.296 2.747-1.027 2.747-1.027.546 1.379.202 2.398.1 2.651.64.7 1.028 1.595 1.028 2.688 0 3.848-2.339 4.695-4.566 4.943.359.309.678.92.678 1.855 0 1.338-.012 2.419-.012 2.747 0 .268.18.58.688.482A10.019 10.019 0 0022 12.017C22 6.484 17.522 2 12 2z" clipRule="evenodd" />
-          </svg>
-        </a>
-        <button
-          onClick={() => setDonationOpen(true)}
-          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
-          title="请喝可乐"
-        >
-          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M17 8h1a4 4 0 010 8h-1m-3.413-8.866A6.501 6.501 0 0012 3c-1.93 0-3.694.84-4.9 2.176M4 20h16a1 1 0 001-1v-1a1 1 0 00-1-1H4a1 1 0 00-1 1v1a1 1 0 001 1zm1-9.5V12a3 3 0 003 3h8a3 3 0 003-3v-1.5M9 8h6" />
-          </svg>
-        </button>
-      </div>
-
-      {/* 右上角按钮 */}
-      <div className="fixed top-4 right-4 z-50 flex items-center gap-2">
-        <button
-          onClick={() => setPromptsOpen(true)}
-          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
-          title="提示词管理"
-        >
-          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M7 8h10M7 12h4m1 8l-4-4H5a2 2 0 01-2-2V6a2 2 0 012-2h14a2 2 0 012 2v8a2 2 0 01-2 2h-3l-4 4z" />
-          </svg>
-        </button>
-        <button
-          onClick={() => setSettingsOpen(true)}
-          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
-          title="API 设置"
-        >
-          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M10.325 4.317c.426-1.756 2.924-1.756 3.35 0a1.724 1.724 0 002.573 1.066c1.543-.94 3.31.826 2.37 2.37a1.724 1.724 0 001.065 2.572c1.756.426 1.756 2.924 0 3.35a1.724 1.724 0 00-1.066 2.573c.94 1.543-.826 3.31-2.37 2.37a1.724 1.724 0 00-2.572 1.065c-.426 1.756-2.924 1.756-3.35 0a1.724 1.724 0 00-2.573-1.066c-1.543.94-3.31-.826-2.37-2.37a1.724 1.724 0 00-1.065-2.572c-1.756-.426-1.756-2.924 0-3.35a1.724 1.724 0 001.066-2.573c-.94-1.543.826-3.31 2.37-2.37.996.608 2.296.07 2.572-1.065z" />
-            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15 12a3 3 0 11-6 0 3 3 0 016 0z" />
-          </svg>
-        </button>
-        <ThemeToggle />
-      </div>
-
-      {/* 主容器 - 使用黄金分割比例调整垂直位置 */}
-      <div className="max-w-4xl mx-auto px-4 min-h-screen flex flex-col justify-center" style={{ paddingTop: '20vh', paddingBottom: '32vh' }}>
-        {/* 标题 */}
-        <div className="text-center mb-12">
-          <div className="flex items-center justify-center gap-4 mb-3">
-            <ManimCatLogo className="w-16 h-16" />
-            <h1 className="text-5xl sm:text-6xl font-light tracking-tight text-text-primary">
-              ManimCat
-            </h1>
-          </div>
-          <p className="text-sm text-text-secondary/70 max-w-lg mx-auto">
-            用 AI 驱动 Manim 生成数学动画
-          </p>
-        </div>
-
-        {/* 状态显示区域 */}
-        <div className="mb-6">
-          {status === 'idle' && (
-            <InputForm
-              onSubmit={handleSubmit}
-              loading={false}
-            />
-          )}
-
-          {status === 'processing' && (
-            <div className="bg-bg-secondary/20 rounded-2xl p-8">
-              <LoadingSpinner stage={stage} jobId={jobId || undefined} onCancel={cancel} />
-            </div>
-          )}
-
-          {status === 'completed' && result && (
-            <div
-              className="space-y-6 animate-fade-in"
-              style={{
-                animation: 'fadeInUp 0.5s ease-out forwards',
-              }}
-            >
-              {/* 结果展示 */}
-              <ResultSection
-                code={result.code || ''}
-                videoUrl={result.video_url || ''}
-                usedAI={result.used_ai || false}
-                renderQuality={result.render_quality || ''}
-                generationType={result.generation_type || ''}
-              />
-
-              {/* 重新生成按钮 */}
-              <div className="text-center">
-                <button
-                  onClick={reset}
-                  className="px-8 py-2.5 text-sm text-text-secondary/80 hover:text-accent transition-colors bg-bg-secondary/30 hover:bg-bg-secondary/50 rounded-full"
-                >
-                  生成新的动画
-                </button>
-              </div>
-            </div>
-          )}
-
-          {status === 'error' && (
-            <div className="bg-red-50/80 dark:bg-red-900/20 rounded-2xl p-6">
-              <div className="flex items-start gap-3">
-                <div className="text-red-500 mt-0.5">
-                  <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                    <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 8v4m0 4h.01M21 12a9 9 0 11-18 0 9 9 0 0118 0z" />
-                  </svg>
-                </div>
-                <div className="flex-1">
-                  <p className="text-text-primary font-medium mb-1">出错了</p>
-                  <p className="text-text-secondary text-sm">{error || '生成失败，请重试'}</p>
-                </div>
-              </div>
-              <div className="mt-4 flex gap-3">
-                <button
-                  onClick={reset}
-                  className="px-4 py-2 text-sm text-accent hover:text-accent-hover transition-colors"
-                >
-                  重试
-                </button>
-              </div>
-            </div>
-          )}
-        </div>
-      </div>
-
-      {/* 添加淡入上浮动画 */}
-      <style>{`
-        @keyframes fadeInUp {
-          0% {
-            opacity: 0;
-            transform: translateY(30px);
-          }
-          100% {
-            opacity: 1;
-            transform: translateY(0);
-          }
-        }
-      `}</style>
-
-      {/* 设置模态框 */}
-      <SettingsModal
-        isOpen={settingsOpen}
-        onClose={() => setSettingsOpen(false)}
-        onSave={(config) => {
-          console.log('保存配置:', config);
-        }}
-      />
-
-      {/* 提示词管理 */}
-      <PromptsManager
-        isOpen={promptsOpen}
-        onClose={() => setPromptsOpen(false)}
-      />
-
-      {/* 捐赠模态框 */}
-      <DonationModal
-        isOpen={donationOpen}
-        onClose={() => setDonationOpen(false)}
-      />
-    </div>
-  );
-}
-
-export default App;
+// 主应用组件
+
+import { useState, useEffect } from 'react';
+import { useGeneration } from './hooks/useGeneration';
+import { InputForm } from './components/InputForm';
+import { LoadingSpinner } from './components/LoadingSpinner';
+import { ResultSection } from './components/ResultSection';
+import { AiModifyModal } from './components/AiModifyModal';
+import { ThemeToggle } from './components/ThemeToggle';
+import { SettingsModal } from './components/SettingsModal';
+import { PromptsManager } from './components/PromptsManager';
+import { DonationModal } from './components/DonationModal';
+import ManimCatLogo from './components/ManimCatLogo';
+import type { Quality } from './types/api';
+
+function App() {
+  const { status, result, error, jobId, stage, generate, renderWithCode, modifyWithAI, reset, cancel } = useGeneration();
+  const [settingsOpen, setSettingsOpen] = useState(false);
+  const [currentCode, setCurrentCode] = useState('');
+  const [lastRequest, setLastRequest] = useState<{ concept: string; quality: Quality; forceRefresh: boolean } | null>(null);
+  const [aiModifyOpen, setAiModifyOpen] = useState(false);
+  const [aiModifyInput, setAiModifyInput] = useState('');
+
+  const [promptsOpen, setPromptsOpen] = useState(false);
+  const [donationOpen, setDonationOpen] = useState(false);
+
+  useEffect(() => {
+    if (result?.code) {
+      setCurrentCode(result.code);
+    }
+  }, [result?.code]);
+
+  const handleSubmit = (data: { concept: string; quality: Quality; forceRefresh: boolean }) => {
+    setLastRequest(data);
+    generate(data);
+  };
+
+  const handleRerender = () => {
+    if (!lastRequest || !currentCode.trim()) {
+      return;
+    }
+    renderWithCode({ ...lastRequest, code: currentCode });
+  };
+
+  const handleAiModifySubmit = () => {
+    if (!lastRequest || !currentCode.trim()) {
+      return;
+    }
+    const instructions = aiModifyInput.trim();
+    if (!instructions) {
+      return;
+    }
+    setAiModifyOpen(false);
+    setAiModifyInput('');
+    modifyWithAI({
+      concept: lastRequest.concept,
+      quality: lastRequest.quality,
+      instructions,
+      code: currentCode
+    });
+  };
+
+  const isBusy = status === 'processing';
+
+  return (
+    <div className="min-h-screen bg-bg-primary transition-colors duration-300">
+      {/* 左上角图标 */}
+      <div className="fixed top-4 left-4 z-50 flex items-center gap-2">
+        <a
+          href="https://github.com/Wing900/ManimCat"
+          target="_blank"
+          rel="noopener noreferrer"
+          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
+          title="GitHub"
+        >
+          <svg className="w-5 h-5" fill="currentColor" viewBox="0 0 24 24">
+            <path fillRule="evenodd" d="M12 2C6.477 2 2 6.484 2 12.017c0 4.425 2.865 8.18 6.839 9.504.5.092.682-.217.682-.483 0-.237-.008-.868-.013-1.703-2.782.605-3.369-1.343-3.369-1.343-.454-1.158-1.11-1.466-1.11-1.466-.908-.62.069-.608.069-.608 1.003.07 1.531 1.032 1.531 1.032.892 1.53 2.341 1.088 2.91.832.092-.647.35-1.088.636-1.338-2.22-.253-4.555-1.113-4.555-4.951 0-1.093.39-1.988 1.029-2.688-.103-.253-.446-1.272.098-2.65 0 0 .84-.27 2.75 1.026A9.564 9.564 0 0112 6.844c.85.004 1.705.115 2.504.337 1.909-1.296 2.747-1.027 2.747-1.027.546 1.379.202 2.398.1 2.651.64.7 1.028 1.595 1.028 2.688 0 3.848-2.339 4.695-4.566 4.943.359.309.678.92.678 1.855 0 1.338-.012 2.419-.012 2.747 0 .268.18.58.688.482A10.019 10.019 0 0022 12.017C22 6.484 17.522 2 12 2z" clipRule="evenodd" />
+          </svg>
+        </a>
+        <button
+          onClick={() => setDonationOpen(true)}
+          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
+          title="请喝可乐"
+        >
+          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M17 8h1a4 4 0 010 8h-1m-3.413-8.866A6.501 6.501 0 0012 3c-1.93 0-3.694.84-4.9 2.176M4 20h16a1 1 0 001-1v-1a1 1 0 00-1-1H4a1 1 0 00-1 1v1a1 1 0 001 1zm1-9.5V12a3 3 0 003 3h8a3 3 0 003-3v-1.5M9 8h6" />
+          </svg>
+        </button>
+      </div>
+
+      {/* 右上角按钮 */}
+      <div className="fixed top-4 right-4 z-50 flex items-center gap-2">
+        <button
+          onClick={() => setPromptsOpen(true)}
+          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
+          title="提示词管理"
+        >
+          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M7 8h10M7 12h4m1 8l-4-4H5a2 2 0 01-2-2V6a2 2 0 012-2h14a2 2 0 012 2v8a2 2 0 01-2 2h-3l-4 4z" />
+          </svg>
+        </button>
+        <button
+          onClick={() => setSettingsOpen(true)}
+          className="p-2.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-secondary/50 rounded-full transition-all active:scale-90 active:duration-75"
+          title="API 设置"
+        >
+          <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M10.325 4.317c.426-1.756 2.924-1.756 3.35 0a1.724 1.724 0 002.573 1.066c1.543-.94 3.31.826 2.37 2.37a1.724 1.724 0 001.065 2.572c1.756.426 1.756 2.924 0 3.35a1.724 1.724 0 00-1.066 2.573c.94 1.543-.826 3.31-2.37 2.37a1.724 1.724 0 00-2.572 1.065c-.426 1.756-2.924 1.756-3.35 0a1.724 1.724 0 00-2.573-1.066c-1.543.94-3.31-.826-2.37-2.37a1.724 1.724 0 00-1.065-2.572c-1.756-.426-1.756-2.924 0-3.35a1.724 1.724 0 001.066-2.573c-.94-1.543.826-3.31 2.37-2.37.996.608 2.296.07 2.572-1.065z" />
+            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15 12a3 3 0 11-6 0 3 3 0 016 0z" />
+          </svg>
+        </button>
+        <ThemeToggle />
+      </div>
+
+      {/* 主容器 - 使用黄金分割比例调整垂直位置 */}
+      <div className="max-w-4xl mx-auto px-4 min-h-screen flex flex-col justify-center" style={{ paddingTop: '20vh', paddingBottom: '32vh' }}>
+        {/* 标题 */}
+        <div className="text-center mb-12">
+          <div className="flex items-center justify-center gap-4 mb-3">
+            <ManimCatLogo className="w-16 h-16" />
+            <h1 className="text-5xl sm:text-6xl font-light tracking-tight text-text-primary">
+              ManimCat
+            </h1>
+          </div>
+          <p className="text-sm text-text-secondary/70 max-w-lg mx-auto">
+            用 AI 驱动 Manim 生成数学动画
+          </p>
+        </div>
+
+        {/* 状态显示区域 */}
+        <div className="mb-6">
+          {status === 'idle' && (
+            <InputForm
+              onSubmit={handleSubmit}
+              loading={false}
+            />
+          )}
+
+          {status === 'processing' && (
+            <div className="bg-bg-secondary/20 rounded-2xl p-8">
+              <LoadingSpinner stage={stage} jobId={jobId || undefined} onCancel={cancel} />
+            </div>
+          )}
+
+          {status === 'completed' && result && (
+            <div
+              className="space-y-6 animate-fade-in"
+              style={{
+                animation: 'fadeInUp 0.5s ease-out forwards',
+              }}
+            >
+              {/* 结果展示 */}
+              <ResultSection
+                code={currentCode || result.code || ''}
+                videoUrl={result.video_url || ''}
+                usedAI={result.used_ai || false}
+                renderQuality={result.render_quality || ''}
+                generationType={result.generation_type || ''}
+                onCodeChange={setCurrentCode}
+                onRerender={handleRerender}
+                onAiModify={() => setAiModifyOpen(true)}
+                isBusy={isBusy}
+              />
+
+              {/* 重新生成按钮 */}
+              <div className="text-center">
+                <button
+                  onClick={() => {
+                    reset();
+                    setCurrentCode('');
+                    setLastRequest(null);
+                    setAiModifyInput('');
+                    setAiModifyOpen(false);
+                  }}
+                  className="px-8 py-2.5 text-sm text-text-secondary/80 hover:text-accent transition-colors bg-bg-secondary/30 hover:bg-bg-secondary/50 rounded-full"
+                >
+                  生成新的动画
+                </button>
+              </div>
+            </div>
+          )}
+
+          {status === 'error' && (
+            <div className="bg-red-50/80 dark:bg-red-900/20 rounded-2xl p-6">
+              <div className="flex items-start gap-3">
+                <div className="text-red-500 mt-0.5">
+                  <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                    <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 8v4m0 4h.01M21 12a9 9 0 11-18 0 9 9 0 0118 0z" />
+                  </svg>
+                </div>
+                <div className="flex-1">
+                  <p className="text-text-primary font-medium mb-1">出错了</p>
+                  <p className="text-text-secondary text-sm">{error || '生成失败，请重试'}</p>
+                </div>
+              </div>
+              <div className="mt-4 flex gap-3">
+                <button
+                  onClick={() => {
+                    reset();
+                    setCurrentCode('');
+                    setLastRequest(null);
+                    setAiModifyInput('');
+                    setAiModifyOpen(false);
+                  }}
+                  className="px-4 py-2 text-sm text-accent hover:text-accent-hover transition-colors"
+                >
+                  重试
+                </button>
+              </div>
+            </div>
+          )}
+        </div>
+      </div>
+
+      {/* 添加淡入上浮动画 */}
+      <style>{`
+        @keyframes fadeInUp {
+          0% {
+            opacity: 0;
+            transform: translateY(30px);
+          }
+          100% {
+            opacity: 1;
+            transform: translateY(0);
+          }
+        }
+      `}</style>
+
+      {/* 设置模态框 */}
+      <SettingsModal
+        isOpen={settingsOpen}
+        onClose={() => setSettingsOpen(false)}
+        onSave={(config) => {
+          console.log('保存配置:', config);
+        }}
+      />
+
+      {/* 提示词管理 */}
+      <PromptsManager
+        isOpen={promptsOpen}
+        onClose={() => setPromptsOpen(false)}
+      />
+
+      {/* 捐赠模态框 */}
+      <DonationModal
+        isOpen={donationOpen}
+        onClose={() => setDonationOpen(false)}
+      />
+
+      <AiModifyModal
+        isOpen={aiModifyOpen}
+        value={aiModifyInput}
+        loading={isBusy}
+        onChange={setAiModifyInput}
+        onClose={() => setAiModifyOpen(false)}
+        onSubmit={handleAiModifySubmit}
+      />
+    </div>
+  );
+}
+
+export default App;

frontend/src/components/AiModifyModal.tsx

@@ -0,0 +1,94 @@
+// AI 修改对话框
+
+interface AiModifyModalProps {
+  isOpen: boolean;
+  value: string;
+  loading?: boolean;
+  onChange: (value: string) => void;
+  onClose: () => void;
+  onSubmit: () => void;
+}
+
+export function AiModifyModal({ isOpen, value, loading = false, onChange, onClose, onSubmit }: AiModifyModalProps) {
+  if (!isOpen) return null;
+
+  return (
+    <div className="fixed inset-0 z-50 flex items-center justify-center p-4">
+      <div className="absolute inset-0 bg-black/50 backdrop-blur-sm" onClick={onClose} />
+
+      <div className="relative w-full max-w-lg bg-bg-secondary rounded-2xl p-6 shadow-xl animate-fade-in">
+        <div className="flex items-center justify-between mb-4">
+          <div className="flex items-center gap-2">
+            <svg className="w-6 h-6 text-accent" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+              <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 4.5v15m7.5-7.5h-15" />
+            </svg>
+            <h2 className="text-lg font-medium text-text-primary">AI修改</h2>
+          </div>
+          <button
+            onClick={onClose}
+            className="p-1.5 text-text-secondary/70 hover:text-text-secondary hover:bg-bg-primary/50 rounded-full transition-all"
+          >
+            <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+              <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M6 18L18 6M6 6l12 12" />
+            </svg>
+          </button>
+        </div>
+
+        <p className="text-text-secondary text-sm mb-4">
+          请描述你希望 AI 如何修改当前代码。
+        </p>
+
+        <div className="relative mb-6">
+          <label
+            htmlFor="aiModifyInput"
+            className="absolute left-4 -top-2.5 px-2 bg-bg-secondary text-xs font-medium text-text-secondary"
+          >
+            修改意见
+          </label>
+          <textarea
+            id="aiModifyInput"
+            rows={5}
+            value={value}
+            onChange={(e) => onChange(e.target.value)}
+            placeholder="例如：让标题更醒目，加入颜色渐变，缩短动画时长..."
+            className="w-full px-4 py-4 bg-bg-secondary/50 rounded-2xl text-sm text-text-primary placeholder-text-secondary/40 focus:outline-none focus:ring-2 focus:ring-accent/20 focus:bg-bg-secondary/70 transition-all resize-none"
+          />
+        </div>
+
+        <div className="flex gap-3">
+          <button
+            onClick={onClose}
+            disabled={loading}
+            className="flex-1 px-4 py-2.5 text-sm text-text-secondary hover:text-text-primary bg-bg-primary hover:bg-bg-tertiary rounded-xl transition-all active:scale-95 disabled:opacity-50 disabled:cursor-not-allowed"
+          >
+            取消
+          </button>
+          <button
+            onClick={onSubmit}
+            disabled={loading || value.trim().length === 0}
+            className="flex-1 px-4 py-2.5 text-sm text-white bg-accent hover:bg-accent-hover rounded-xl transition-all active:scale-95 disabled:opacity-50 disabled:cursor-not-allowed"
+          >
+            {loading ? '提交中...' : '提交修改'}
+          </button>
+        </div>
+      </div>
+
+      <style>{`
+        @keyframes fadeIn {
+          0% {
+            opacity: 0;
+            transform: scale(0.95);
+          }
+          100% {
+            opacity: 1;
+            transform: scale(1);
+          }
+        }
+        .animate-fade-in {
+          animation: fadeIn 0.2s ease-out forwards;
+        }
+      `}</style>
+    </div>
+  );
+}
+

frontend/src/components/CodeView.tsx

@@ -1,4 +1,4 @@
-// 代码预览组件
+// 代码预览组件
 
 import { Prism as SyntaxHighlighter } from 'react-syntax-highlighter';
 import { vscDarkPlus } from 'react-syntax-highlighter/dist/esm/styles/prism';
@@ -6,9 +6,12 @@ import { useState } from 'react';
 
 interface CodeViewProps {
   code: string;
+  editable?: boolean;
+  onChange?: (value: string) => void;
+  disabled?: boolean;
 }
 
-export function CodeView({ code }: CodeViewProps) {
+export function CodeView({ code, editable = false, onChange, disabled = false }: CodeViewProps) {
   const [copied, setCopied] = useState(false);
 
   const handleCopy = async () => {
@@ -17,6 +20,14 @@ export function CodeView({ code }: CodeViewProps) {
     setTimeout(() => setCopied(false), 2000);
   };
 
+  const textareaClassName = [
+    'w-full h-full resize-none bg-transparent p-4 text-[0.75rem] leading-relaxed',
+    'font-mono text-text-primary/90 focus:outline-none',
+    disabled ? 'opacity-60 cursor-not-allowed' : ''
+  ]
+    .filter(Boolean)
+    .join(' ');
+
   return (
     <div className="h-full flex flex-col bg-bg-secondary/30 rounded-2xl overflow-hidden">
       {/* 顶部工具栏 */}
@@ -46,22 +57,32 @@ export function CodeView({ code }: CodeViewProps) {
 
       {/* 代码区域 */}
       <div className="flex-1 overflow-auto">
-        <SyntaxHighlighter
-          language="python"
-          style={vscDarkPlus}
-          customStyle={{
-            margin: 0,
-            padding: '1rem',
-            fontSize: '0.75rem',
-            lineHeight: '1.6',
-            fontFamily: 'Monaco, Cascadia Code, Roboto Mono, monospace',
-            background: 'transparent',
-          }}
-          showLineNumbers
-        >
-          {code}
-        </SyntaxHighlighter>
+        {editable ? (
+          <textarea
+            value={code}
+            onChange={(event) => onChange?.(event.target.value)}
+            className={textareaClassName}
+            disabled={disabled}
+            spellCheck={false}
+          />
+        ) : (
+          <SyntaxHighlighter
+            language="python"
+            style={vscDarkPlus}
+            customStyle={{
+              margin: 0,
+              padding: '1rem',
+              fontSize: '0.75rem',
+              lineHeight: '1.6',
+              fontFamily: 'Monaco, Cascadia Code, Roboto Mono, monospace',
+              background: 'transparent'
+            }}
+            showLineNumbers
+          >
+            {code}
+          </SyntaxHighlighter>
+        )}
       </div>
     </div>
   );
-}
+}

frontend/src/components/PromptSidebar.tsx

@@ -1,189 +1,220 @@
-// 提示词管理侧边栏组件
-
-import type { ReactNode } from 'react';
-
-interface SidebarItemProps {
-  icon: ReactNode;
-  label: string;
-  active?: boolean;
-  onClick?: () => void;
-  children?: ReactNode;
-  expanded?: boolean;
-  onToggle?: () => void;
-  indent?: boolean;
-}
-
-function SidebarItem({
-  icon,
-  label,
-  active = false,
-  onClick,
-  children,
-  expanded = true,
-  onToggle,
-  indent = false
-}: SidebarItemProps) {
-  const hasChildren = !!children;
-
-  return (
-    <div className={`${indent ? 'ml-4' : ''}`}>
-      <button
-        onClick={() => hasChildren ? onToggle?.() : onClick?.()}
-        className={`w-full flex items-center gap-3 px-4 py-2.5 text-sm transition-colors ${
-          active
-            ? 'text-accent bg-bg-secondary/50 rounded-lg'
-            : 'text-text-secondary hover:text-text-primary hover:bg-bg-secondary/30 rounded-lg'
-        }`}
-      >
-        <span className="w-5 h-5 flex items-center justify-center">{icon}</span>
-        <span className="flex-1 text-left">{label}</span>
-        {hasChildren && (
-          <svg
-            className={`w-4 h-4 transition-transform ${expanded ? 'rotate-180' : ''}`}
-            fill="none"
-            stroke="currentColor"
-            viewBox="0 0 24 24"
-          >
-            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M19 9l-7 7-7-7" />
-          </svg>
-        )}
-      </button>
-      {hasChildren && expanded && (
-        <div className="mt-1 space-y-1">
-          {children}
-        </div>
-      )}
-    </div>
-  );
-}
-
-interface PromptSidebarProps {
-  activeSection: string;
-  activePrompt: string;
-  onSectionChange: (section: string, prompt: string) => void;
-}
-
-export function PromptSidebar({
-  activeSection,
-  activePrompt,
-  onSectionChange
-}: PromptSidebarProps) {
-  return (
-    <div className="w-64 bg-bg-secondary/30 border-r border-bg-secondary/50 overflow-y-auto">
-      <div className="p-4 space-y-6">
-        {/* 概念设计提示词 */}
-        <div>
-          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
-            概念设计
-          </h3>
-          <div className="space-y-1">
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M7 21a4 4 0 01-4-4V5a2 2 0 012-2h4a2 2 0 012 2v12a4 4 0 01-4 4zm0 0h12a2 2 0 002-2v-4a2 2 0 00-2-2h-2.343M11 7.343l1.657-1.657a2 2 0 012.828 0l2.829 2.829a2 2 0 010 2.828l-8.486 8.485M7 17h.01" />
-                </svg>
-              }
-              label="系统提示词"
-              active={activeSection === 'system' && activePrompt === 'conceptDesigner'}
-              onClick={() => onSectionChange('system', 'conceptDesigner')}
-              indent
-            />
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15.232 5.232l3.536 3.536m-2.036-5.036a2.5 2.5 0 113.536 3.536L6.5 21.036H3v-3.572L16.732 3.732z" />
-                </svg>
-              }
-              label="用户提示词"
-              active={activeSection === 'user' && activePrompt === 'conceptDesigner'}
-              onClick={() => onSectionChange('user', 'conceptDesigner')}
-              indent
-            />
-          </div>
-        </div>
-
-        {/* 代码生成提示词 */}
-        <div>
-          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
-            代码生成
-          </h3>
-          <div className="space-y-1">
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M8 9l4-4 4 4m0 6l-4 4-4-4" />
-                </svg>
-              }
-              label="系统提示词"
-              active={activeSection === 'system' && activePrompt === 'codeGeneration'}
-              onClick={() => onSectionChange('system', 'codeGeneration')}
-              indent
-            />
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M10 20l4-16m4 4l4 4-4 4M6 16l-4-4 4-4" />
-                </svg>
-              }
-              label="用户提示词"
-              active={activeSection === 'user' && activePrompt === 'codeGeneration'}
-              onClick={() => onSectionChange('user', 'codeGeneration')}
-              indent
-            />
-          </div>
-        </div>
-
-        {/* 代码修复提示词 */}
-        <div>
-          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
-            代码修复
-          </h3>
-          <div className="space-y-1">
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M11 5H6a2 2 0 00-2 2v11a2 2 0 002 2h11a2 2 0 002-2v-5m-1.414-9.414a2 2 0 112.828 2.828L11.828 15H9v-2.828l8.586-8.586z" />
-                </svg>
-              }
-              label="初始重试提示词"
-              active={activeSection === 'user' && activePrompt === 'codeRetryInitial'}
-              onClick={() => onSectionChange('user', 'codeRetryInitial')}
-              indent
-            />
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M13 10V3L4 14h7v7l9-11h-7z" />
-                </svg>
-              }
-              label="修复提示词"
-              active={activeSection === 'user' && activePrompt === 'codeRetryFix'}
-              onClick={() => onSectionChange('user', 'codeRetryFix')}
-              indent
-            />
-          </div>
-        </div>
-
-        {/* 系统重试提示词 */}
-        <div>
-          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
-            系统重试
-          </h3>
-          <div className="space-y-1">
-            <SidebarItem
-              icon={
-                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M4 4v5h.582m15.356 2A8.001 8.001 0 004.582 9m0 0H9m11 11v-5h-.581m0 0a8.003 8.003 0 01-15.357-2m15.357 2H15" />
-                </svg>
-              }
-              label="重试提示词"
-              active={activeSection === 'system' && activePrompt === 'codeRetry'}
-              onClick={() => onSectionChange('system', 'codeRetry')}
-              indent
-            />
-          </div>
-        </div>
-      </div>
-    </div>
-  );
-}
+// 提示词管理侧边栏组件
+
+import type { ReactNode } from 'react';
+
+interface SidebarItemProps {
+  icon: ReactNode;
+  label: string;
+  active?: boolean;
+  onClick?: () => void;
+  children?: ReactNode;
+  expanded?: boolean;
+  onToggle?: () => void;
+  indent?: boolean;
+}
+
+function SidebarItem({
+  icon,
+  label,
+  active = false,
+  onClick,
+  children,
+  expanded = true,
+  onToggle,
+  indent = false
+}: SidebarItemProps) {
+  const hasChildren = !!children;
+
+  return (
+    <div className={`${indent ? 'ml-4' : ''}`}>
+      <button
+        onClick={() => hasChildren ? onToggle?.() : onClick?.()}
+        className={`w-full flex items-center gap-3 px-4 py-2.5 text-sm transition-colors ${
+          active
+            ? 'text-accent bg-bg-secondary/50 rounded-lg'
+            : 'text-text-secondary hover:text-text-primary hover:bg-bg-secondary/30 rounded-lg'
+        }`}
+      >
+        <span className="w-5 h-5 flex items-center justify-center">{icon}</span>
+        <span className="flex-1 text-left">{label}</span>
+        {hasChildren && (
+          <svg
+            className={`w-4 h-4 transition-transform ${expanded ? 'rotate-180' : ''}`}
+            fill="none"
+            stroke="currentColor"
+            viewBox="0 0 24 24"
+          >
+            <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M19 9l-7 7-7-7" />
+          </svg>
+        )}
+      </button>
+      {hasChildren && expanded && (
+        <div className="mt-1 space-y-1">
+          {children}
+        </div>
+      )}
+    </div>
+  );
+}
+
+interface PromptSidebarProps {
+  activeSection: string;
+  activePrompt: string;
+  onSectionChange: (section: string, prompt: string) => void;
+}
+
+export function PromptSidebar({
+  activeSection,
+  activePrompt,
+  onSectionChange
+}: PromptSidebarProps) {
+  return (
+    <div className="w-64 bg-bg-secondary/30 border-r border-bg-secondary/50 overflow-y-auto">
+      <div className="p-4 space-y-6">
+        {/* 概念设计提示词 */}
+        <div>
+          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
+            概念设计
+          </h3>
+          <div className="space-y-1">
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M7 21a4 4 0 01-4-4V5a2 2 0 012-2h4a2 2 0 012 2v12a4 4 0 01-4 4zm0 0h12a2 2 0 002-2v-4a2 2 0 00-2-2h-2.343M11 7.343l1.657-1.657a2 2 0 012.828 0l2.829 2.829a2 2 0 010 2.828l-8.486 8.485M7 17h.01" />
+                </svg>
+              }
+              label="系统���示词"
+              active={activeSection === 'system' && activePrompt === 'conceptDesigner'}
+              onClick={() => onSectionChange('system', 'conceptDesigner')}
+              indent
+            />
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15.232 5.232l3.536 3.536m-2.036-5.036a2.5 2.5 0 113.536 3.536L6.5 21.036H3v-3.572L16.732 3.732z" />
+                </svg>
+              }
+              label="用户提示词"
+              active={activeSection === 'user' && activePrompt === 'conceptDesigner'}
+              onClick={() => onSectionChange('user', 'conceptDesigner')}
+              indent
+            />
+          </div>
+        </div>
+
+        {/* 代码生成提示词 */}
+        <div>
+          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
+            代码生成
+          </h3>
+          <div className="space-y-1">
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M8 9l4-4 4 4m0 6l-4 4-4-4" />
+                </svg>
+              }
+              label="系统提示词"
+              active={activeSection === 'system' && activePrompt === 'codeGeneration'}
+              onClick={() => onSectionChange('system', 'codeGeneration')}
+              indent
+            />
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M10 20l4-16m4 4l4 4-4 4M6 16l-4-4 4-4" />
+                </svg>
+              }
+              label="用户提示词"
+              active={activeSection === 'user' && activePrompt === 'codeGeneration'}
+              onClick={() => onSectionChange('user', 'codeGeneration')}
+              indent
+            />
+          </div>
+        </div>
+
+        {/* 代码修复提示词 */}
+        <div>
+          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
+            代码修复
+          </h3>
+          <div className="space-y-1">
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M11 5H6a2 2 0 00-2 2v11a2 2 0 002 2h11a2 2 0 002-2v-5m-1.414-9.414a2 2 0 112.828 2.828L11.828 15H9v-2.828l8.586-8.586z" />
+                </svg>
+              }
+              label="初始重试提示词"
+              active={activeSection === 'user' && activePrompt === 'codeRetryInitial'}
+              onClick={() => onSectionChange('user', 'codeRetryInitial')}
+              indent
+            />
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M13 10V3L4 14h7v7l9-11h-7z" />
+                </svg>
+              }
+              label="修复提示词"
+              active={activeSection === 'user' && activePrompt === 'codeRetryFix'}
+              onClick={() => onSectionChange('user', 'codeRetryFix')}
+              indent
+            />
+          </div>
+        </div>
+
+        {/* AI修改提示词 */}
+        <div>
+          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
+            AI修改
+          </h3>
+          <div className="space-y-1">
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 4.354a4 4 0 110 15.292M15 21H9m6 0a3 3 0 01-6 0m6 0H9" />
+                </svg>
+              }
+              label="系统提示词"
+              active={activeSection === 'system' && activePrompt === 'codeEdit'}
+              onClick={() => onSectionChange('system', 'codeEdit')}
+              indent
+            />
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M12 4v16m8-8H4" />
+                </svg>
+              }
+              label="用户提示词"
+              active={activeSection === 'user' && activePrompt === 'codeEdit'}
+              onClick={() => onSectionChange('user', 'codeEdit')}
+              indent
+            />
+          </div>
+        </div>
+
+        {/* 系统重试提示词 */}
+        <div>
+          <h3 className="px-4 text-xs font-medium text-text-secondary/60 uppercase tracking-wider mb-2">
+            系统重试
+          </h3>
+          <div className="space-y-1">
+            <SidebarItem
+              icon={
+                <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                  <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M4 4v5h.582m15.356 2A8.001 8.001 0 004.582 9m0 0H9m11 11v-5h-.581m0 0a8.003 8.003 0 01-15.357-2m15.357 2H15" />
+                </svg>
+              }
+              label="重试提示词"
+              active={activeSection === 'system' && activePrompt === 'codeRetry'}
+              onClick={() => onSectionChange('system', 'codeRetry')}
+              indent
+            />
+          </div>
+        </div>
+      </div>
+    </div>
+  );
+}

frontend/src/components/PromptsManager.tsx

@@ -1,214 +1,224 @@
-// 提示词管理主页面组件
-// 包含侧边栏导航和主编辑区的完整布局
-
-import { useEffect, useState } from 'react';
-import { PromptSidebar } from './PromptSidebar';
-import { PromptInput } from './PromptInput';
-import { usePrompts } from '../hooks/usePrompts';
-
-interface PromptsManagerProps {
-  isOpen: boolean;
-  onClose: () => void;
-}
-
-// 提示词类型配置
-const PROMPT_CONFIG = {
-  system: {
-    conceptDesigner: {
-      label: '概念设计系统提示词',
-      placeholder: '输入概念设计阶段的系统提示词...',
-      description: '用于指导 AI 理解数学概念并设计动画场景'
-    },
-    codeGeneration: {
-      label: '代码生成系统提示词',
-      placeholder: '输入代码生成阶段的系统提示词...',
-      description: '用于指导 AI 生成符合规范的 Manim 代码'
-    },
-    codeRetry: {
-      label: '系统重试提示词',
-      placeholder: '输入系统重试阶段的提示词...',
-      description: '用于指导 AI 在代码失败时进行重试和优化'
-    }
-  },
-  user: {
-    conceptDesigner: {
-      label: '概念设计用户提示词',
-      placeholder: '输入概念设计阶段的用户提示词...',
-      description: '补充说明概念设计的具体需求和风格'
-    },
-    codeGeneration: {
-      label: '代码生成用户提示词',
-      placeholder: '输入代码生成阶段的用户提示词...',
-      description: '补充说明代码生成的具体要求和规范'
-    },
-    codeRetryInitial: {
-      label: '代码修复初始重试提示词',
-      placeholder: '输入代码修复初始重试阶段的提示词...',
-      description: '代码第一次失败时的修复指导'
-    },
-    codeRetryFix: {
-      label: '代码修复提示词',
-      placeholder: '输入代码修复阶段的提示词...',
-      description: '代码第二次失败时的详细修复指导'
-    }
-  }
-};
-
-export function PromptsManager({ isOpen, onClose }: PromptsManagerProps) {
-  const {
-    isLoading,
-    activeSection,
-    activePrompt,
-    setActiveSection,
-    setActivePrompt,
-    getCurrentPrompt,
-    setCurrentPrompt,
-    restoreDefault
-  } = usePrompts();
-
-    const [saveStatus, setSaveStatus] = useState<'idle' | 'saving' | 'success' | 'error'>('idle');
-
-    const TRANSITION_MS = 400;
-
-    const [shouldRender, setShouldRender] = useState(isOpen);
-
-    const [isVisible, setIsVisible] = useState(isOpen);
-
-  useEffect(() => {
-    if (isOpen) {
-      setShouldRender(true);
-      // 延迟一帧再显示，确保初始 opacity-0 生效
-      setTimeout(() => setIsVisible(true), 50);
-    } else {
-      setIsVisible(false);
-      const timeout = window.setTimeout(() => setShouldRender(false), 400);
-      return () => window.clearTimeout(timeout);
-    }
-  }, [isOpen]);
-
-  // 处理侧边栏导航
-  const handleSectionChange = (section: string, prompt: string) => {
-    setActiveSection(section);
-    setActivePrompt(prompt);
-    setSaveStatus('idle');
-  };
-
-  // 处理保存
-  const handleSave = () => {
-    setSaveStatus('saving');
-    setTimeout(() => {
-      setSaveStatus('success');
-      setTimeout(() => setSaveStatus('idle'), 2000);
-    }, 500);
-  };
-
-  // 处理恢复默认
-  const handleRestoreDefault = () => {
-    restoreDefault();
-    setSaveStatus('success');
-    setTimeout(() => setSaveStatus('idle'), 2000);
-  };
-
-  // 获取当前配置
-  const currentConfig = activeSection === 'system'
-    ? PROMPT_CONFIG.system[activePrompt as keyof typeof PROMPT_CONFIG.system]
-    : PROMPT_CONFIG.user[activePrompt as keyof typeof PROMPT_CONFIG.user];
-
-  if (!shouldRender) return null;
-
-  return (
-    <div
-      className={`fixed inset-0 z-50 flex flex-col bg-bg-primary transition-all duration-[400ms] ease-out ${isVisible ? 'opacity-100 translate-y-0 scale-100' : 'opacity-0 translate-y-8 scale-95 pointer-events-none'}`}
-    >
-      {/* 顶部导航栏 */}
-      <div className="h-16 bg-bg-secondary border-b border-bg-secondary/50 flex items-center justify-between px-6">
-        <div className="flex items-center gap-4">
-          <button
-            onClick={onClose}
-            className="p-2 text-text-secondary hover:text-text-primary hover:bg-bg-secondary/50 rounded-lg transition-colors"
-            title="返回主界面"
-          >
-            <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-              <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15 19l-7-7 7-7" />
-            </svg>
-          </button>
-          <h1 className="text-lg font-medium text-text-primary">提示词管理</h1>
-        </div>
-        <div className="flex items-center gap-3">
-          {saveStatus === 'success' && (
-            <div className="flex items-center gap-2 text-sm text-green-600 dark:text-green-400">
-              <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
-                <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M5 13l4 4L19 7" />
-              </svg>
-              <span>保存成功</span>
-            </div>
-          )}
-          <button
-            onClick={handleSave}
-            disabled={saveStatus === 'saving'}
-            className="px-4 py-2 bg-accent text-white text-sm font-medium rounded-lg hover:bg-accent-hover disabled:opacity-50 disabled:cursor-not-allowed transition-all"
-          >
-            {saveStatus === 'saving' ? (
-              <>
-                <svg className="animate-spin w-4 h-4 inline mr-2" fill="none" viewBox="0 0 24 24">
-                  <circle className="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" strokeWidth="4" />
-                  <path className="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z" />
-                </svg>
-                保存中...
-              </>
-            ) : (
-              '保存'
-            )}
-          </button>
-        </div>
-      </div>
-
-      {/* 主内容区 */}
-      <div className="flex-1 flex overflow-hidden">
-        {/* 侧边栏 */}
-        <PromptSidebar
-          activeSection={activeSection}
-          activePrompt={activePrompt}
-          onSectionChange={handleSectionChange}
-        />
-
-        {/* 主编辑区 */}
-        <div className="flex-1 overflow-y-auto p-8">
-          <div className="max-w-4xl mx-auto space-y-8">
-            {/* 当前提示词信息 */}
-            <div className="bg-bg-secondary/30 rounded-xl p-6">
-              <h2 className="text-xl font-medium text-text-primary mb-2">
-                {currentConfig?.label || '提示词编辑'}
-              </h2>
-              <p className="text-sm text-text-secondary/70">
-                {currentConfig?.description || '在此处编辑提示词，它将用于指导 AI 生成和优化动画'}
-              </p>
-            </div>
-
-            {/* 输入组件 */}
-            <div className="space-y-6">
-              <PromptInput
-                value={getCurrentPrompt()}
-                onChange={setCurrentPrompt}
-                label={currentConfig?.label || '提示词'}
-                placeholder={currentConfig?.placeholder}
-                showWordCount
-                disabled={isLoading}
-                onSave={handleSave}
-                onRestoreDefault={handleRestoreDefault}
-              />
-
-            </div>
-          </div>
-        </div>
-      </div>
-
-      {/* 底部状态栏 */}
-      <div className="h-10 bg-bg-secondary border-t border-bg-secondary/50 flex items-center justify-between px-6 text-xs text-text-secondary">
-        <span>提示词管理</span>
-        <span>字符限制：20000</span>
-      </div>
-    </div>
-  );
-}
-
+// 提示词管理主页面组件
+// 包含侧边栏导航和主编辑区的完整布局
+
+import { useEffect, useState } from 'react';
+import { PromptSidebar } from './PromptSidebar';
+import { PromptInput } from './PromptInput';
+import { usePrompts } from '../hooks/usePrompts';
+
+interface PromptsManagerProps {
+  isOpen: boolean;
+  onClose: () => void;
+}
+
+// 提示词类型配置
+const PROMPT_CONFIG = {
+  system: {
+    conceptDesigner: {
+      label: '概念设计系统提示词',
+      placeholder: '输入概念设计阶段的系统提示词...',
+      description: '用于指导 AI 理解数学概念并设计动画场景'
+    },
+    codeGeneration: {
+      label: '代码生成系统提示词',
+      placeholder: '输入代码生成阶段的系统提示词...',
+      description: '用于指导 AI 生成符合规范的 Manim 代码'
+    },
+    codeRetry: {
+      label: '系统重试提示词',
+      placeholder: '输入系统重试阶段的提示词...',
+      description: '用于指导 AI 在代码失败时进行重试和优化'
+    },
+    codeEdit: {
+      label: 'AI修改系统提示词',
+      placeholder: '用于 AI 修改阶段的系统提示词...',
+      description: '用于指导 AI 基于现有代码进行修改'
+    }
+  },
+  user: {
+    conceptDesigner: {
+      label: '概念设计用户提示词',
+      placeholder: '输入概念设计阶段的用户提示词...',
+      description: '补充说明概念设计的具体需求和风格'
+    },
+    codeGeneration: {
+      label: '代码生成用户提示词',
+      placeholder: '输入代码生成阶段的用户提示词...',
+      description: '补充说明代码生成的具体要求和规范'
+    },
+    codeRetryInitial: {
+      label: '代码修复初始重试提示词',
+      placeholder: '输入代码修复初始重试阶段的提示词...',
+      description: '代码第一次失败时的修复指导'
+    },
+    codeRetryFix: {
+      label: '代码修复提示词',
+      placeholder: '输入代码修复阶段的提示词...',
+      description: '代码第二次失败时的详细修复指导'
+    },
+    codeEdit: {
+      label: 'AI修改用户提示词',
+      placeholder: '用于 AI 修改阶段的用户提示词...',
+      description: '用于描述修改目标、约束与期望效果'
+    }
+  }
+};
+
+export function PromptsManager({ isOpen, onClose }: PromptsManagerProps) {
+  const {
+    isLoading,
+    activeSection,
+    activePrompt,
+    setActiveSection,
+    setActivePrompt,
+    getCurrentPrompt,
+    setCurrentPrompt,
+    restoreDefault
+  } = usePrompts();
+
+    const [saveStatus, setSaveStatus] = useState<'idle' | 'saving' | 'success' | 'error'>('idle');
+
+    const TRANSITION_MS = 400;
+
+    const [shouldRender, setShouldRender] = useState(isOpen);
+
+    const [isVisible, setIsVisible] = useState(isOpen);
+
+  useEffect(() => {
+    if (isOpen) {
+      setShouldRender(true);
+      // 延迟一帧再显示，确保初始 opacity-0 生效
+      setTimeout(() => setIsVisible(true), 50);
+    } else {
+      setIsVisible(false);
+      const timeout = window.setTimeout(() => setShouldRender(false), 400);
+      return () => window.clearTimeout(timeout);
+    }
+  }, [isOpen]);
+
+  // 处理侧边栏导航
+  const handleSectionChange = (section: string, prompt: string) => {
+    setActiveSection(section);
+    setActivePrompt(prompt);
+    setSaveStatus('idle');
+  };
+
+  // 处理保存
+  const handleSave = () => {
+    setSaveStatus('saving');
+    setTimeout(() => {
+      setSaveStatus('success');
+      setTimeout(() => setSaveStatus('idle'), 2000);
+    }, 500);
+  };
+
+  // 处理恢复默认
+  const handleRestoreDefault = () => {
+    restoreDefault();
+    setSaveStatus('success');
+    setTimeout(() => setSaveStatus('idle'), 2000);
+  };
+
+  // 获取当前配置
+  const currentConfig = activeSection === 'system'
+    ? PROMPT_CONFIG.system[activePrompt as keyof typeof PROMPT_CONFIG.system]
+    : PROMPT_CONFIG.user[activePrompt as keyof typeof PROMPT_CONFIG.user];
+
+  if (!shouldRender) return null;
+
+  return (
+    <div
+      className={`fixed inset-0 z-50 flex flex-col bg-bg-primary transition-all duration-[400ms] ease-out ${isVisible ? 'opacity-100 translate-y-0 scale-100' : 'opacity-0 translate-y-8 scale-95 pointer-events-none'}`}
+    >
+      {/* 顶部导航栏 */}
+      <div className="h-16 bg-bg-secondary border-b border-bg-secondary/50 flex items-center justify-between px-6">
+        <div className="flex items-center gap-4">
+          <button
+            onClick={onClose}
+            className="p-2 text-text-secondary hover:text-text-primary hover:bg-bg-secondary/50 rounded-lg transition-colors"
+            title="返回主界面"
+          >
+            <svg className="w-5 h-5" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+              <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M15 19l-7-7 7-7" />
+            </svg>
+          </button>
+          <h1 className="text-lg font-medium text-text-primary">提示词管理</h1>
+        </div>
+        <div className="flex items-center gap-3">
+          {saveStatus === 'success' && (
+            <div className="flex items-center gap-2 text-sm text-green-600 dark:text-green-400">
+              <svg className="w-4 h-4" fill="none" stroke="currentColor" viewBox="0 0 24 24">
+                <path strokeLinecap="round" strokeLinejoin="round" strokeWidth={2} d="M5 13l4 4L19 7" />
+              </svg>
+              <span>保存成功</span>
+            </div>
+          )}
+          <button
+            onClick={handleSave}
+            disabled={saveStatus === 'saving'}
+            className="px-4 py-2 bg-accent text-white text-sm font-medium rounded-lg hover:bg-accent-hover disabled:opacity-50 disabled:cursor-not-allowed transition-all"
+          >
+            {saveStatus === 'saving' ? (
+              <>
+                <svg className="animate-spin w-4 h-4 inline mr-2" fill="none" viewBox="0 0 24 24">
+                  <circle className="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" strokeWidth="4" />
+                  <path className="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z" />
+                </svg>
+                保存中...
+              </>
+            ) : (
+              '保存'
+            )}
+          </button>
+        </div>
+      </div>
+
+      {/* 主内容区 */}
+      <div className="flex-1 flex overflow-hidden">
+        {/* 侧边栏 */}
+        <PromptSidebar
+          activeSection={activeSection}
+          activePrompt={activePrompt}
+          onSectionChange={handleSectionChange}
+        />
+
+        {/* 主编辑区 */}
+        <div className="flex-1 overflow-y-auto p-8">
+          <div className="max-w-4xl mx-auto space-y-8">
+            {/* 当前提示词信息 */}
+            <div className="bg-bg-secondary/30 rounded-xl p-6">
+              <h2 className="text-xl font-medium text-text-primary mb-2">
+                {currentConfig?.label || '提示词编辑'}
+              </h2>
+              <p className="text-sm text-text-secondary/70">
+                {currentConfig?.description || '在此处编辑提示词，它将用于指导 AI 生成和优化动画'}
+              </p>
+            </div>
+
+            {/* 输入组件 */}
+            <div className="space-y-6">
+              <PromptInput
+                value={getCurrentPrompt()}
+                onChange={setCurrentPrompt}
+                label={currentConfig?.label || '提示词'}
+                placeholder={currentConfig?.placeholder}
+                showWordCount
+                disabled={isLoading}
+                onSave={handleSave}
+                onRestoreDefault={handleRestoreDefault}
+              />
+
+            </div>
+          </div>
+        </div>
+      </div>
+
+      {/* 底部状态栏 */}
+      <div className="h-10 bg-bg-secondary border-t border-bg-secondary/50 flex items-center justify-between px-6 text-xs text-text-secondary">
+        <span>提示词管理</span>
+        <span>字符限制：20000</span>
+      </div>
+    </div>
+  );
+}
+

frontend/src/components/ResultSection.tsx

@@ -1,4 +1,4 @@
-// 结果展示组件
+// 结果展示区域
 
 import { CodeView } from './CodeView';
 import { VideoPreview } from './VideoPreview';
@@ -9,27 +9,66 @@ interface ResultSectionProps {
   usedAI: boolean;
   renderQuality: string;
   generationType: string;
+  onCodeChange?: (code: string) => void;
+  onRerender?: () => void;
+  onAiModify?: () => void;
+  isBusy?: boolean;
 }
 
-export function ResultSection({ code, videoUrl, usedAI, renderQuality, generationType }: ResultSectionProps) {
+export function ResultSection({
+  code,
+  videoUrl,
+  usedAI,
+  renderQuality,
+  generationType,
+  onCodeChange,
+  onRerender,
+  onAiModify,
+  isBusy = false
+}: ResultSectionProps) {
+  const hasActions = onRerender || onAiModify;
+
   return (
     <div className="w-full max-w-4xl mx-auto space-y-5">
-      {/* 代码和视频预览 */}
+      {/* 代码与视频预览 */}
       <div className="grid grid-cols-1 lg:grid-cols-2 gap-4">
         <div className="h-[360px]">
-          <CodeView code={code} />
+          <CodeView code={code} editable={Boolean(onCodeChange)} onChange={onCodeChange} disabled={isBusy} />
         </div>
         <div className="h-[360px]">
           <VideoPreview videoUrl={videoUrl} />
         </div>
       </div>
 
-      {/* 生成信息 */}
+      {/* 结果信息 */}
       <div className="bg-bg-secondary/30 rounded-xl px-4 py-2.5">
         <p className="text-xs text-text-secondary/70">
           {generationType}{usedAI ? ' (AI)' : ''} · {renderQuality}
         </p>
       </div>
+
+      {hasActions && (
+        <div className="flex flex-wrap items-center justify-center gap-3">
+          {onRerender && (
+            <button
+              onClick={onRerender}
+              disabled={isBusy}
+              className="px-5 py-2 text-xs font-medium text-text-secondary/80 hover:text-text-primary bg-bg-secondary/30 hover:bg-bg-secondary/50 rounded-full transition-all disabled:opacity-50 disabled:cursor-not-allowed"
+            >
+              重新渲染
+            </button>
+          )}
+          {onAiModify && (
+            <button
+              onClick={onAiModify}
+              disabled={isBusy}
+              className="px-5 py-2 text-xs font-medium text-white bg-accent hover:bg-accent-hover rounded-full transition-all disabled:opacity-50 disabled:cursor-not-allowed shadow-sm shadow-accent/20"
+            >
+              AI修改
+            </button>
+          )}
+        </div>
+      )}
     </div>
   );
-}
+}

frontend/src/hooks/useGeneration.ts

@@ -1,10 +1,10 @@
 // 生成请求 Hook
 
 import { useState, useCallback, useRef, useEffect } from 'react';
-import { generateAnimation, getJobStatus, cancelJob } from '../lib/api';
+import { generateAnimation, getJobStatus, cancelJob, modifyAnimation } from '../lib/api';
 import { loadCustomConfig, generateWithCustomApi } from '../lib/custom-ai';
 import { loadPrompts } from './usePrompts';
-import type { GenerateRequest, JobResult, ProcessingStage, VideoConfig } from '../types/api';
+import type { GenerateRequest, JobResult, ProcessingStage, VideoConfig, ModifyRequest } from '../types/api';
 
 interface UseGenerationReturn {
   status: 'idle' | 'processing' | 'completed' | 'error';
@@ -13,6 +13,8 @@ interface UseGenerationReturn {
   jobId: string | null;
   stage: ProcessingStage;
   generate: (request: GenerateRequest) => Promise<void>;
+  renderWithCode: (request: GenerateRequest & { code: string }) => Promise<void>;
+  modifyWithAI: (request: ModifyRequest) => Promise<void>;
   reset: () => void;
   cancel: () => void;
 }
@@ -165,6 +167,56 @@ export function useGeneration(): UseGenerationReturn {
     }, POLL_INTERVAL);
   }, [requestCancel, updateStage]);
 
+  // 使用现有代码重新渲染
+  const renderWithCode = useCallback(async (request: GenerateRequest & { code: string }) => {
+    setStatus('processing');
+    setError(null);
+    setResult(null);
+    setStage('rendering');
+    pollCountRef.current = 0;
+    abortControllerRef.current = new AbortController();
+
+    try {
+      const promptOverrides = loadPrompts();
+      const response = await generateAnimation(
+        { ...request, promptOverrides },
+        abortControllerRef.current.signal
+      );
+      startPolling(response.jobId);
+    } catch (err) {
+      if (err instanceof Error && err.name === 'AbortError') {
+        return;
+      }
+      setStatus('error');
+      setError(err instanceof Error ? err.message : '重新渲染失败');
+    }
+  }, [startPolling]);
+
+  // AI 修改后渲染
+  const modifyWithAI = useCallback(async (request: ModifyRequest) => {
+    setStatus('processing');
+    setError(null);
+    setResult(null);
+    setStage('generating');
+    pollCountRef.current = 0;
+    abortControllerRef.current = new AbortController();
+
+    try {
+      const promptOverrides = loadPrompts();
+      const response = await modifyAnimation(
+        { ...request, promptOverrides },
+        abortControllerRef.current.signal
+      );
+      startPolling(response.jobId);
+    } catch (err) {
+      if (err instanceof Error && err.name === 'AbortError') {
+        return;
+      }
+      setStatus('error');
+      setError(err instanceof Error ? err.message : 'AI 修改失败');
+    }
+  }, [startPolling]);
+
   // 生成动画
   const generate = useCallback(async (request: GenerateRequest) => {
     setStatus('processing');
@@ -247,6 +299,8 @@ export function useGeneration(): UseGenerationReturn {
     jobId,
     stage,
     generate,
+    renderWithCode,
+    modifyWithAI,
     reset,
     cancel,
   };

frontend/src/hooks/usePrompts.ts

@@ -1,173 +1,175 @@
-// 提示词管理 Hook
-// 负责提示词的状态管理、存储和加载
-
-import { useState, useEffect, useCallback } from 'react';
-import { getPromptDefaults } from '../lib/api';
-import type { PromptOverrides } from '../types/api';
-
-/** 存储在 localStorage 中的键名 */
-const PROMPTS_STORAGE_KEY = 'manimcat_prompt_overrides';
-
-/** 默认提示词配置 */
-const DEFAULT_PROMPTS: PromptOverrides = {
-  system: {
-    conceptDesigner: '',
-    codeGeneration: '',
-    codeRetry: '',
-  },
-  user: {
-    conceptDesigner: '',
-    codeGeneration: '',
-    codeRetryInitial: '',
-    codeRetryFix: '',
-  },
-};
-
-/** 从 localStorage 加载提示词配置 */
-/** Load prompt overrides from localStorage if available. */
-export function loadStoredPrompts(): PromptOverrides | null {
-  try {
-    const saved = localStorage.getItem(PROMPTS_STORAGE_KEY);
-    if (saved) {
-      const parsed = JSON.parse(saved);
-      return {
-        system: { ...DEFAULT_PROMPTS.system, ...parsed.system },
-        user: { ...DEFAULT_PROMPTS.user, ...parsed.user },
-      };
-    }
-  } catch (error) {
-    console.error('Failed to load prompts:', error);
-  }
-  return null;
-}
-
-export function loadPrompts(): PromptOverrides {
-  return loadStoredPrompts() || DEFAULT_PROMPTS;
-}
-
-/** 保存提示词配置到 localStorage */
-export function savePrompts(prompts: PromptOverrides): void {
-  try {
-    localStorage.setItem(PROMPTS_STORAGE_KEY, JSON.stringify(prompts));
-  } catch (error) {
-    console.error('Failed to save prompts:', error);
-  }
-}
-
-/** 提示词管理 Hook */
-export function usePrompts() {
-  const [prompts, setPrompts] = useState<PromptOverrides>(DEFAULT_PROMPTS);
-  const [defaultPrompts, setDefaultPrompts] = useState<PromptOverrides>(DEFAULT_PROMPTS);
-  const [isLoading, setIsLoading] = useState<boolean>(true);
-  const [activeSection, setActiveSection] = useState<string>('system');
-  const [activePrompt, setActivePrompt] = useState<string>('conceptDesigner');
-
-  // 从 localStorage 加载配置
-  useEffect(() => {
-    const savedPrompts = loadStoredPrompts();
-    if (savedPrompts) {
-      setPrompts(savedPrompts);
-    }
-
-    let isActive = true;
-    const loadDefaults = async () => {
-      setIsLoading(true);
-      try {
-        const defaults = await getPromptDefaults();
-        if (isActive) {
-          setDefaultPrompts(defaults);
-        }
-      } catch (error) {
-        console.error('Failed to load prompt defaults:', error);
-      } finally {
-        if (isActive) {
-          setIsLoading(false);
-        }
-      }
-    };
-
-    void loadDefaults();
-    return () => {
-      isActive = false;
-    };
-  }, []);
-
-  // 保存到 localStorage
-  useEffect(() => {
-    savePrompts(prompts);
-  }, [prompts]);
-
-  // 更新提示词
-  const updatePrompt = useCallback((section: string, type: string, value: string) => {
-    setPrompts(prev => {
-      const newPrompts = { ...prev };
-
-      if (section === 'system') {
-        if (!newPrompts.system) {
-          newPrompts.system = {};
-        }
-        newPrompts.system[type as keyof PromptOverrides['system']] = value;
-      } else if (section === 'user') {
-        if (!newPrompts.user) {
-          newPrompts.user = {};
-        }
-        newPrompts.user[type as keyof PromptOverrides['user']] = value;
-      }
-
-      return newPrompts;
-    });
-  }, []);
-
-  // 恢复默认值
-  const restoreDefault = useCallback(() => {
-    setPrompts(DEFAULT_PROMPTS);
-  }, []);
-
-  // 获取当前编辑的提示词
-  const getDefaultPrompt = useCallback((section: string, type: string) => {
-    if (section === 'system') {
-      return defaultPrompts.system?.[type as keyof PromptOverrides['system']] || '';
-    }
-    if (section === 'user') {
-      return defaultPrompts.user?.[type as keyof PromptOverrides['user']] || '';
-    }
-    return '';
-  }, [defaultPrompts]);
-
-  // 
-  const getCurrentPrompt = useCallback(() => {
-    let current = '';
-    if (activeSection === 'system') {
-      current = prompts.system?.[activePrompt as keyof PromptOverrides['system']] || '';
-    } else if (activeSection === 'user') {
-      current = prompts.user?.[activePrompt as keyof PromptOverrides['user']] || '';
-    }
-
-    if (current && current.trim().length > 0) {
-      return current;
-    }
-
-    return getDefaultPrompt(activeSection, activePrompt);
-  }, [prompts, activeSection, activePrompt, getDefaultPrompt]);
-
-  // 
-  const setCurrentPrompt = useCallback((value: string) => {
-    const defaultValue = getDefaultPrompt(activeSection, activePrompt);
-    const nextValue = value === defaultValue ? '' : value;
-    updatePrompt(activeSection, activePrompt, nextValue);
-  }, [activeSection, activePrompt, getDefaultPrompt, updatePrompt]);
-
-  return {
-    prompts,
-    defaultPrompts,
-    isLoading,
-    activeSection,
-    activePrompt,
-    setActiveSection,
-    setActivePrompt,
-    updatePrompt,
-    restoreDefault,
-    getCurrentPrompt,
-    setCurrentPrompt,
-  };
-}
+// 提示词管理 Hook
+// 负责提示词的状态管理、存储和加载
+
+import { useState, useEffect, useCallback } from 'react';
+import { getPromptDefaults } from '../lib/api';
+import type { PromptOverrides } from '../types/api';
+
+/** 存储在 localStorage 中的键名 */
+const PROMPTS_STORAGE_KEY = 'manimcat_prompt_overrides';
+
+/** 默认提示词配置 */
+const DEFAULT_PROMPTS: PromptOverrides = {
+  system: {
+    conceptDesigner: '',
+    codeGeneration: '',
+    codeRetry: '',
+    codeEdit: '',
+  },
+  user: {
+    conceptDesigner: '',
+    codeGeneration: '',
+    codeRetryInitial: '',
+    codeRetryFix: '',
+    codeEdit: '',
+  },
+};
+
+/** 从 localStorage 加载提示词配置 */
+/** Load prompt overrides from localStorage if available. */
+export function loadStoredPrompts(): PromptOverrides | null {
+  try {
+    const saved = localStorage.getItem(PROMPTS_STORAGE_KEY);
+    if (saved) {
+      const parsed = JSON.parse(saved);
+      return {
+        system: { ...DEFAULT_PROMPTS.system, ...parsed.system },
+        user: { ...DEFAULT_PROMPTS.user, ...parsed.user },
+      };
+    }
+  } catch (error) {
+    console.error('Failed to load prompts:', error);
+  }
+  return null;
+}
+
+export function loadPrompts(): PromptOverrides {
+  return loadStoredPrompts() || DEFAULT_PROMPTS;
+}
+
+/** 保存提示词配置到 localStorage */
+export function savePrompts(prompts: PromptOverrides): void {
+  try {
+    localStorage.setItem(PROMPTS_STORAGE_KEY, JSON.stringify(prompts));
+  } catch (error) {
+    console.error('Failed to save prompts:', error);
+  }
+}
+
+/** 提示词管理 Hook */
+export function usePrompts() {
+  const [prompts, setPrompts] = useState<PromptOverrides>(DEFAULT_PROMPTS);
+  const [defaultPrompts, setDefaultPrompts] = useState<PromptOverrides>(DEFAULT_PROMPTS);
+  const [isLoading, setIsLoading] = useState<boolean>(true);
+  const [activeSection, setActiveSection] = useState<string>('system');
+  const [activePrompt, setActivePrompt] = useState<string>('conceptDesigner');
+
+  // 从 localStorage 加载配置
+  useEffect(() => {
+    const savedPrompts = loadStoredPrompts();
+    if (savedPrompts) {
+      setPrompts(savedPrompts);
+    }
+
+    let isActive = true;
+    const loadDefaults = async () => {
+      setIsLoading(true);
+      try {
+        const defaults = await getPromptDefaults();
+        if (isActive) {
+          setDefaultPrompts(defaults);
+        }
+      } catch (error) {
+        console.error('Failed to load prompt defaults:', error);
+      } finally {
+        if (isActive) {
+          setIsLoading(false);
+        }
+      }
+    };
+
+    void loadDefaults();
+    return () => {
+      isActive = false;
+    };
+  }, []);
+
+  // 保存到 localStorage
+  useEffect(() => {
+    savePrompts(prompts);
+  }, [prompts]);
+
+  // 更新提示词
+  const updatePrompt = useCallback((section: string, type: string, value: string) => {
+    setPrompts(prev => {
+      const newPrompts = { ...prev };
+
+      if (section === 'system') {
+        if (!newPrompts.system) {
+          newPrompts.system = {};
+        }
+        newPrompts.system[type as keyof PromptOverrides['system']] = value;
+      } else if (section === 'user') {
+        if (!newPrompts.user) {
+          newPrompts.user = {};
+        }
+        newPrompts.user[type as keyof PromptOverrides['user']] = value;
+      }
+
+      return newPrompts;
+    });
+  }, []);
+
+  // 恢复默认值
+  const restoreDefault = useCallback(() => {
+    setPrompts(DEFAULT_PROMPTS);
+  }, []);
+
+  // 获取当前编辑的提示词
+  const getDefaultPrompt = useCallback((section: string, type: string) => {
+    if (section === 'system') {
+      return defaultPrompts.system?.[type as keyof PromptOverrides['system']] || '';
+    }
+    if (section === 'user') {
+      return defaultPrompts.user?.[type as keyof PromptOverrides['user']] || '';
+    }
+    return '';
+  }, [defaultPrompts]);
+
+  // 
+  const getCurrentPrompt = useCallback(() => {
+    let current = '';
+    if (activeSection === 'system') {
+      current = prompts.system?.[activePrompt as keyof PromptOverrides['system']] || '';
+    } else if (activeSection === 'user') {
+      current = prompts.user?.[activePrompt as keyof PromptOverrides['user']] || '';
+    }
+
+    if (current && current.trim().length > 0) {
+      return current;
+    }
+
+    return getDefaultPrompt(activeSection, activePrompt);
+  }, [prompts, activeSection, activePrompt, getDefaultPrompt]);
+
+  // 
+  const setCurrentPrompt = useCallback((value: string) => {
+    const defaultValue = getDefaultPrompt(activeSection, activePrompt);
+    const nextValue = value === defaultValue ? '' : value;
+    updatePrompt(activeSection, activePrompt, nextValue);
+  }, [activeSection, activePrompt, getDefaultPrompt, updatePrompt]);
+
+  return {
+    prompts,
+    defaultPrompts,
+    isLoading,
+    activeSection,
+    activePrompt,
+    setActiveSection,
+    setActivePrompt,
+    updatePrompt,
+    restoreDefault,
+    getCurrentPrompt,
+    setCurrentPrompt,
+  };
+}

frontend/src/lib/api.ts

@@ -1,6 +1,6 @@
 // API 请求函数
 
-import type { GenerateRequest, GenerateResponse, JobResult, ApiError, VideoConfig, PromptOverrides } from '../types/api';
+import type { GenerateRequest, GenerateResponse, JobResult, ApiError, VideoConfig, PromptOverrides, ModifyRequest } from '../types/api';
 
 const API_BASE = '/api';
 
@@ -40,6 +40,25 @@ function getAuthHeaders(): HeadersInit {
 /**
  * 提交动画生成请求
  */
+export async function modifyAnimation(request: ModifyRequest, signal?: AbortSignal): Promise<GenerateResponse> {
+  const videoConfig = request.videoConfig || loadVideoConfig();
+
+  const payload = { ...request, videoConfig };
+  const response = await fetch(`${API_BASE}/modify`, {
+    method: 'POST',
+    headers: getAuthHeaders(),
+    body: JSON.stringify(payload),
+    signal,
+  });
+
+  if (!response.ok) {
+    const error: ApiError = await response.json();
+    throw new Error(error.error || 'AI 修改失败');
+  }
+
+  return response.json();
+}
+
 export async function generateAnimation(request: GenerateRequest, signal?: AbortSignal): Promise<GenerateResponse> {
   // 如果请求中没有 videoConfig，则从设置中加载默认值
   const videoConfig = request.videoConfig || loadVideoConfig();

frontend/src/types/api.ts

@@ -1,93 +1,105 @@
-// API 类型定义
-
-/** 视频质量选项 */
-export type Quality = 'low' | 'medium' | 'high';
-
-/** API 配置 */
-export interface ApiConfig {
-  apiUrl: string;
-  apiKey: string;
-  model: string;
-  manimcatApiKey: string;
-}
-
-export interface PromptOverrides {
-  system?: {
-    conceptDesigner?: string;
-    codeGeneration?: string;
-    codeRetry?: string;
-  };
-  user?: {
-    conceptDesigner?: string;
-    codeGeneration?: string;
-    codeRetryInitial?: string;
-    codeRetryFix?: string;
-  };
-}
-
-/** 视频配置 */
-export interface VideoConfig {
-  /** 默认质量 */
-  quality: Quality;
-  /** 帧率 */
-  frameRate: number;
-  /** 超时时间（秒），默认 600 秒（10 分钟） */
-  timeout?: number;
-}
-
-/** 设置配置 */
-export interface SettingsConfig {
-  api: ApiConfig;
-  video: VideoConfig;
-}
-
-/** 任务状态 */
-export type JobStatus = 'processing' | 'completed' | 'failed';
-
-/** 处理阶段 */
-export type ProcessingStage = 'analyzing' | 'generating' | 'refining' | 'rendering' | 'still-rendering';
-
-/** 生成请求 */
-export interface GenerateRequest {
-  concept: string;
-  quality?: Quality;
-  forceRefresh?: boolean;
-  /** 预生成的代码（使用自定义 AI 时） */
-  code?: string;
-  /** 视频配置 */
-  videoConfig?: VideoConfig;
-  /** Prompt overrides */
-  promptOverrides?: PromptOverrides;
-}
-
-/** 生成响应 */
-export interface GenerateResponse {
-  success: boolean;
-  jobId: string;
-  message: string;
-  status: JobStatus;
-}
-
-/** 任务结果 */
-export interface JobResult {
-  jobId: string;
-  status: JobStatus;
-  stage?: ProcessingStage;
-  message?: string;
-  success?: boolean;
-  video_url?: string;
-  code?: string;
-  used_ai?: boolean;
-  render_quality?: string;
-  generation_type?: string;
-  render_peak_memory_mb?: number;
-
-  error?: string;
-  details?: string;
-  cancel_reason?: string;
-}
-
-/** API 错误 */
-export interface ApiError {
-  error: string;
-}
+// API 类型定义
+
+/** 视频质量选项 */
+export type Quality = 'low' | 'medium' | 'high';
+
+/** API 配置 */
+export interface ApiConfig {
+  apiUrl: string;
+  apiKey: string;
+  model: string;
+  manimcatApiKey: string;
+}
+
+export interface PromptOverrides {
+  system?: {
+    conceptDesigner?: string;
+    codeGeneration?: string;
+    codeRetry?: string;
+    codeEdit?: string;
+  };
+  user?: {
+    conceptDesigner?: string;
+    codeGeneration?: string;
+    codeRetryInitial?: string;
+    codeRetryFix?: string;
+    codeEdit?: string;
+  };
+}
+
+/** 视频配置 */
+export interface VideoConfig {
+  /** 默认质量 */
+  quality: Quality;
+  /** 帧率 */
+  frameRate: number;
+  /** 超时时间（秒），默认 600 秒（10 分钟） */
+  timeout?: number;
+}
+
+/** 设置配置 */
+export interface SettingsConfig {
+  api: ApiConfig;
+  video: VideoConfig;
+}
+
+/** 任务状态 */
+export type JobStatus = 'processing' | 'completed' | 'failed';
+
+/** 处理阶段 */
+export type ProcessingStage = 'analyzing' | 'generating' | 'refining' | 'rendering' | 'still-rendering';
+
+/** 生成请求 */
+export interface GenerateRequest {
+  concept: string;
+  quality?: Quality;
+  forceRefresh?: boolean;
+  /** 预生成的代码（使用自定义 AI 时） */
+  code?: string;
+  /** 视频配置 */
+  videoConfig?: VideoConfig;
+  /** Prompt overrides */
+  promptOverrides?: PromptOverrides;
+}
+
+/** AI 修改请求 */
+export interface ModifyRequest {
+  concept: string;
+  quality?: Quality;
+  instructions: string;
+  code: string;
+  videoConfig?: VideoConfig;
+  promptOverrides?: PromptOverrides;
+}
+
+/** 生成响应 */
+export interface GenerateResponse {
+  success: boolean;
+  jobId: string;
+  message: string;
+  status: JobStatus;
+}
+
+/** 任务结果 */
+export interface JobResult {
+  jobId: string;
+  status: JobStatus;
+  stage?: ProcessingStage;
+  message?: string;
+  success?: boolean;
+  video_url?: string;
+  code?: string;
+  used_ai?: boolean;
+  render_quality?: string;
+  generation_type?: string;
+  render_peak_memory_mb?: number;
+
+  error?: string;
+  details?: string;
+  cancel_reason?: string;
+}
+
+/** API 错误 */
+export interface ApiError {
+  error: string;
+}

src/config/bull.ts

@@ -1,24 +1,13 @@
-/**
- * Bull Queue Configuration
- * 任务队列配置和初始化
- */
-
-import { default as Bull, type Queue, type QueueOptions } from 'bull'
+import { default as Bull, type Queue, type QueueOptions } from 'bull'
 import Redis from 'ioredis'
 import { redisClient, REDIS_KEYS } from './redis'
 import { createLogger } from '../utils/logger'
 
 const logger = createLogger('BullQueue')
 
-/**
- * Bull 队列配置选项
- */
 const queueOptions: QueueOptions = {
   prefix: REDIS_KEYS.QUEUE_PREFIX,
-  // ... (保持不变)
-
-  // 为 Bull 创建专用的 Redis 客户端
-  createClient: (type) => {
+  createClient: (_type) => {
     const REDIS_HOST = process.env.REDIS_HOST || 'localhost'
     const REDIS_PORT = parseInt(process.env.REDIS_PORT || '6379', 10)
     const REDIS_PASSWORD = process.env.REDIS_PASSWORD || undefined
@@ -29,40 +18,31 @@ const queueOptions: QueueOptions = {
       port: REDIS_PORT,
       password: REDIS_PASSWORD,
       db: REDIS_DB,
-      maxRetriesPerRequest: null,  // Bull 需要
-      enableReadyCheck: false      // Bull 需要
+      maxRetriesPerRequest: null,
+      enableReadyCheck: false
     })
 
     return client
   },
-
-  // 默认任务配置
   defaultJobOptions: {
-    attempts: 3,              // 失败后重试 3 次
+    attempts: 3,
     backoff: {
-      type: 'exponential',    // 指数退避
-      delay: 2000             // 初始延迟 2 秒
+      type: 'exponential',
+      delay: 2000
     },
-    removeOnComplete: true,   // 完成后自动清理
-    removeOnFail: true,       // 失败后自动清理
-    timeout: 600000           // 任务超时 10 分钟
+    removeOnComplete: true,
+    removeOnFail: true,
+    timeout: 600000
   },
-
   settings: {
-    lockDuration: 600000,     // 锁定时长 10 分钟
-    stalledInterval: 30000,   // 检查停滞任务间隔 30 秒
-    maxStalledCount: 1        // 最多标记为停滞 1 次
+    lockDuration: 600000,
+    stalledInterval: 30000,
+    maxStalledCount: 1
   }
 }
 
-/**
- * 视频生成队列
- */
 export const videoQueue: Queue = new Bull('video-generation', queueOptions)
 
-/**
- * 启动时清理旧任务（防止后端重启后僵尸任务残留）
- */
 async function cleanupStaleJobs() {
   try {
     await videoQueue.isReady()
@@ -73,9 +53,6 @@ async function cleanupStaleJobs() {
       return
     }
 
-    // 使用 clean() 方法移除旧任务
-    // age=0 表示移除所有任务，不限制时间
-    // 注意：Bull 的 clean 方法状态参数是 'wait', 'active', 'completed', 'failed', 'delayed'
     const [removedWait, removedActive, removedFailed, removedCompleted, removedDelayed] = await Promise.all([
       videoQueue.clean(0, 'wait'),
       videoQueue.clean(0, 'active'),
@@ -84,40 +61,34 @@ async function cleanupStaleJobs() {
       videoQueue.clean(0, 'delayed')
     ])
 
-    const totalRemoved = (removedWait || 0) + (removedActive || 0) + (removedFailed || 0) + (removedCompleted || 0) + (removedDelayed || 0)
+    const totalRemoved = removedWait.length + removedActive.length + removedFailed.length + removedCompleted.length + removedDelayed.length
 
     if (totalRemoved > 0) {
-      logger.info(`启动清理: 移除了 ${totalRemoved} 个残留队列任务`, {
-        wait: removedWait,
-        active: removedActive,
-        failed: removedFailed,
-        completed: removedCompleted,
-        delayed: removedDelayed
+      logger.info(`Startup cleanup: removed ${totalRemoved} queue jobs`, {
+        wait: removedWait.length,
+        active: removedActive.length,
+        failed: removedFailed.length,
+        completed: removedCompleted.length,
+        delayed: removedDelayed.length
       })
     }
 
-    // 额外的彻底清理：删除 JobStore 存储的任务结果
-    // 扫描所有以 'job:result:' 开头的 Key
     const resultKeys = await redisClient.keys(`${REDIS_KEYS.JOB_RESULT}*`)
     if (resultKeys.length > 0) {
       await redisClient.del(...resultKeys)
-      logger.info(`启动清理: 删除了 ${resultKeys.length} 个历史任务结果缓存`)
+      logger.info(`Startup cleanup: removed ${resultKeys.length} job result entries`)
     }
 
     if (totalRemoved === 0 && resultKeys.length === 0) {
-      logger.info('启动检查: 无需清理残留数据')
+      logger.info('Startup cleanup: no stale data found')
     }
   } catch (error: any) {
-    logger.warn('启动清理任务时遇到警告 (非致命)', { error: error?.message || String(error) })
+    logger.warn('Startup cleanup warning', { error: error?.message || String(error) })
   }
 }
 
-// 启动时执行清理
 cleanupStaleJobs()
 
-/**
- * 队列事件监听
- */
 videoQueue.on('error', (error) => {
   logger.error('Queue error', { message: error.message })
 })
@@ -130,7 +101,7 @@ videoQueue.on('active', (job) => {
   logger.info(`Job ${job.id} started processing`)
 })
 
-videoQueue.on('completed', (job, result) => {
+videoQueue.on('completed', (job) => {
   logger.info(`Job ${job.id} completed`)
 })
 
@@ -146,18 +117,12 @@ videoQueue.on('stalled', (job) => {
   logger.warn(`Job ${job.id} stalled`)
 })
 
-/**
- * 清理队列（开发调试用）
- */
 export async function cleanQueue(): Promise<void> {
   await videoQueue.clean(0, 'completed')
   await videoQueue.clean(0, 'failed')
   logger.info('Queue cleaned')
 }
 
-/**
- * 获取队列统计信息
- */
 export async function getQueueStats() {
   const [waiting, active, completed, failed, delayed] = await Promise.all([
     videoQueue.getWaitingCount(),
@@ -177,24 +142,18 @@ export async function getQueueStats() {
   }
 }
 
-/**
- * 优雅关闭队列
- */
-export async function closeQueue(): Promise<void> {
-  await videoQueue.close()
-  logger.info('Queue closed gracefully')
-}
-
-/**
- * 检查队列健康状态
- */
 export async function checkQueueHealth(): Promise<boolean> {
   try {
-    const stats = await getQueueStats()
-    // 如果有太多失败的任务，认为不健康
-    return stats.failed < 100
+    await videoQueue.isReady()
+    await videoQueue.getJobCounts()
+    return true
   } catch (error) {
-    logger.error('Queue health check failed', { error })
+    logger.warn('Queue health check failed', { error: String(error) })
     return false
   }
 }
+
+export async function closeQueue(): Promise<void> {
+  await videoQueue.close()
+  logger.info('Queue connection closed')
+}

src/prompts/index.ts

@@ -61,7 +61,14 @@ export const SYSTEM_PROMPTS = {
 
 - **动态更新**：对于涉及数值变化的过程，优先使用 \`ValueTracker\` 和 \`always_redraw\`。
 - **公式操作规范**：禁止使用硬编码索引，必须通过 \`substrings_to_isolate\` 配合 \`get_part_by_tex\` 来操作公式的特定部分。
-- **坐标系一致性**：所有图形必须通过 \`axes.c2p\` 映射到坐标轴上，严禁脱离坐标系的自由定位。`
+- **坐标系一致性**：所有图形必须通过 \`axes.c2p\` 映射到坐标轴上，严禁脱离坐标系的自由定位。`,
+
+  /** AI 修改时的 system prompt */
+  codeEdit: `你是一位 Manim 动画专家，擅长在既有代码基础上进行精准修改。
+严格遵循提示词规范，确保输出符合 Manim Community Edition (v0.19.2) 的可执行代码。
+
+- **保持可运行**：修改后代码必须完整可运行，结构保持为 \`MainScene\`。
+- **只输出代码**：禁止任何解释或 Markdown 包裹。`
 }
 
 export const SYSTEM_PROMPT_BASE = SYSTEM_PROMPTS.codeGeneration
@@ -336,6 +343,42 @@ ${brokenCode}
 请修复上述代码，只输出修复后的完整 Python 代码，不要任何解释。`
 }
 
+/**
+ * 生成 AI 修改时的用户 prompt
+ */
+export function generateCodeEditPrompt(
+  concept: string,
+  instructions: string,
+  code: string
+): string {
+  return `## 目标
+
+### 修改信息
+
+- **概念**：${concept}
+- **修改意见**：${instructions}
+
+### 输出要求
+
+- **仅输出代码**：禁止解释或 Markdown 包裹
+- **锚点协议**：使用 ### START ### 开始，### END ### 结束，仅输出锚点之间的代码
+- **结构规范**：场景类固定为 \`MainScene\`，统一使用 \`from manim import *\`
+
+## 知识库
+
+### API 索引表
+
+${API_INDEX}
+
+## 原始代码
+
+\`\`\`python
+${code}
+\`\`\`
+
+请根据修改意见输出完整的 Manim Python 代码。`;
+}
+
 // =====================
 // Prompt Builder Utilities
 // =====================

src/queues/processors/steps/render-step.ts

@@ -1,24 +1,17 @@
-/**
- * 视频渲染步骤
- * 执行 Manim 渲染，使用新的重试机制
- */
-
-import * as fs from 'fs'
-import * as path from 'path'
-import * as os from 'os'
-import { executeManimCommand, ManimExecuteOptions } from '../../../utils/manim-executor'
-import { findVideoFile } from '../../../utils/file-utils'
-import { storeJobStage } from '../../../services/job-store'
+import fs from 'fs'
+import os from 'os'
+import path from 'path'
 import { createLogger } from '../../../utils/logger'
 import { cleanManimCode } from '../../../utils/manim-code-cleaner'
-import { createRetryContext, executeCodeRetry } from '../../../services/code-retry'
-import type { PromptOverrides, VideoJobData, VideoConfig } from '../../../types'
+import { executeManimCommand, type ManimExecuteOptions } from '../../../utils/manim-executor'
+import { findVideoFile } from '../../../utils/file-utils'
+import { createRetryContext, executeCodeRetry } from '../../../services/code-retry/manager'
+import { storeJobStage } from '../../../services/job-store'
+import type { GenerationResult } from './analysis-step'
+import type { PromptOverrides, VideoConfig, VideoJobData } from '../../../types'
 
 const logger = createLogger('RenderStep')
 
-/**
- * 渲染结果类型
- */
 export interface RenderResult {
   jobId: string
   concept: string
@@ -27,37 +20,10 @@ export interface RenderResult {
   generationType: string
   quality: string
   videoUrl: string
-  renderPeakMemoryMB: number
+  renderPeakMemoryMB?: number
 }
-
 /**
- * 代码生成结果类型
- */
-export interface GenerationResult {
-  manimCode: string
-  usedAI: boolean
-  generationType: string
-  sceneDesign?: string // 场景设计方案，用于重试
-}
-
-// 质量对应的分辨率配置
-const QUALITY_RESOLUTION: Record<string, { width: number; height: number }> = {
-  low: { width: 854, height: 480 },
-  medium: { width: 1280, height: 720 },
-  high: { width: 1920, height: 1080 }
-}
-
-/**
- * 任务结果
- */
-export interface TaskResult {
-  success: boolean
-  source: string
-  timings: Record<string, number>
-}
-
-/**
- * 渲染视频
+ * 娓叉煋瑙嗛
  */
 export async function renderVideo(
   jobId: string,
@@ -72,12 +38,12 @@ export async function renderVideo(
 ): Promise<RenderResult> {
   const { manimCode, usedAI, generationType, sceneDesign } = codeResult
 
-  // 应用视频配置
+  // 搴旂敤瑙嗛閰嶇疆
   const frameRate = videoConfig?.frameRate || 30
 
   logger.info('Rendering video', { jobId, quality, usedAI, frameRate })
 
-  // 创建临时目录
+  // 鍒涘缓涓存椂鐩綍
   const tempDir = path.join(os.tmpdir(), `manim-${jobId}`)
   const mediaDir = path.join(tempDir, 'media')
   const codeFile = path.join(tempDir, 'scene.py')
@@ -91,7 +57,7 @@ export async function renderVideo(
     let lastRenderedCode = manimCode
     let lastRenderPeakMemoryMB = 0
 
-    // 渲染函数 - 供重试机制使用
+    // 娓叉煋鍑芥暟 - 渚涢噸璇曟満鍒朵娇鐢?
     const renderCode = async (code: string): Promise<{
       success: boolean
       stderr: string
@@ -102,7 +68,7 @@ export async function renderVideo(
       lastRenderedCode = cleaned.code
 
       if (cleaned.changes.length > 0) {
-        logger.info('Manim 代码已清洗', {
+        logger.info('Manim code cleaned', {
           jobId,
           changes: cleaned.changes,
           originalLength: code.length,
@@ -110,16 +76,16 @@ export async function renderVideo(
         })
       }
 
-      logger.info('开始渲染代码', {
+      logger.info('Starting render', {
         jobId,
         codeLength: cleaned.code.length
       })
 
-      // 写入代码文件
+      // 鍐欏叆浠ｇ爜鏂囦欢
       fs.writeFileSync(codeFile, cleaned.code, 'utf-8')
-      logger.info('代码已写入文件', { jobId, codeFile })
+      logger.info('Code written to file', { jobId, codeFile })
 
-      // 执行 manim
+      // 鎵ц manim
       const options: ManimExecuteOptions = {
         jobId,
         quality,
@@ -131,7 +97,7 @@ export async function renderVideo(
       const result = await executeManimCommand(codeFile, options)
       lastRenderPeakMemoryMB = result.peakMemoryMB
 
-      logger.info('Manim 渲染完成', {
+      logger.info('Manim 娓叉煋瀹屾垚', {
         jobId,
         success: result.success,
         stdoutLength: result.stdout.length,
@@ -139,41 +105,41 @@ export async function renderVideo(
         peakMemoryMB: result.peakMemoryMB
       })
 
-      // 记录完整输出
+      // 璁板綍瀹屾暣杈撳嚭
       if (result.stdout) {
-        logger.info('Manim stdout 输出', { jobId, stdout: result.stdout })
+        logger.info('Manim stdout 杈撳嚭', { jobId, stdout: result.stdout })
       }
 
       if (result.stderr) {
         if (result.success) {
-          logger.info('Manim stderr 输出（非错误）', { jobId, stderr: result.stderr })
+          logger.info('Manim stderr output (non-error)', { jobId, stderr: result.stderr })
         } else {
-          logger.error('Manim stderr 输出（错误）', { jobId, stderr: result.stderr })
+          logger.error('Manim stderr 杈撳嚭锛堥敊璇級', { jobId, stderr: result.stderr })
         }
       }
 
       return result
     }
 
-    // 决定使用哪种渲染策略
+    // 鍐冲畾浣跨敤鍝娓叉煋绛栫暐
     let finalCode = manimCode
     let renderResult: { success: boolean; stderr: string; stdout: string; peakMemoryMB: number }
 
-    // 检查是否有场景设计方案（来自两阶段 AI 生成）
+    // 妫€鏌ユ槸鍚︽湁鍦烘櫙璁捐鏂规锛堟潵鑷袱闃舵 AI 鐢熸垚锛?
     const hasSceneDesign = usedAI && !!sceneDesign
 
     if (hasSceneDesign) {
-      // 使用新的重试机制：维护完整对话历史
+      // 浣跨敤鏂扮殑閲嶈瘯鏈哄埗锛氱淮鎶ゅ畬鏁村璇濆巻鍙?
       logger.info('Using new code retry mechanism', { jobId, hasSceneDesign })
 
       await storeJobStage(jobId, 'generating')
 
       const retryStart = Date.now()
 
-      // 创建重试上下文
+      // 鍒涘缓閲嶈瘯涓婁笅鏂?
       const retryContext = createRetryContext(concept, sceneDesign, promptOverrides)
 
-      // 执行重试管理器
+      // 鎵ц閲嶈瘯绠＄悊鍣?
       const retryManagerResult = await executeCodeRetry(
         retryContext,
         renderCode,
@@ -191,18 +157,18 @@ export async function renderVideo(
           peakMemoryMB: lastRenderPeakMemoryMB
         }
 
-        logger.info('代码重试成功', {
+        logger.info('浠ｇ爜閲嶈瘯鎴愬姛', {
           jobId,
           attempts: retryManagerResult.attempts
         })
       } else {
-        // 所有重试均失败
+        // 鎵€鏈夐噸璇曞潎澶辫触
         throw new Error(
           `Code retry failed after ${retryManagerResult.attempts} attempts: ${retryManagerResult.lastError}`
         )
       }
     } else {
-      // 非AI生成或无场景设计方案：单次渲染
+      // 闈濧I鐢熸垚鎴栨棤鍦烘櫙璁捐鏂规锛氬崟娆℃覆鏌?
       logger.info('Using single render attempt', {
         jobId,
         reason: usedAI ? 'no_scene_design' : 'not_ai_generated'
@@ -212,7 +178,7 @@ export async function renderVideo(
       renderResult = await renderCode(manimCode)
 
       if (!renderResult.success) {
-        logger.error('Manim 渲染失败', {
+        logger.error('Manim 娓叉煋澶辫触', {
           jobId,
           stderrLength: renderResult.stderr.length,
           stdoutLength: renderResult.stdout.length,
@@ -225,13 +191,13 @@ export async function renderVideo(
       finalCode = lastRenderedCode
     }
 
-    // 查找生成的视频文件
+    // 鏌ユ壘鐢熸垚鐨勮棰戞枃浠?
     const videoPath = findVideoFile(mediaDir, quality, frameRate)
     if (!videoPath) {
       throw new Error('Video file not found after render')
     }
 
-    // 复制到输出目录
+    // 澶嶅埗鍒拌緭鍑虹洰褰?
     const outputFilename = `${jobId}.mp4`
     const outputPath = path.join(outputDir, outputFilename)
     fs.copyFileSync(videoPath, outputPath)
@@ -249,7 +215,7 @@ export async function renderVideo(
       renderPeakMemoryMB: renderResult.peakMemoryMB
     }
   } finally {
-    // 清理临时目录
+    // 娓呯悊涓存椂鐩綍
     try {
       fs.rmSync(tempDir, { recursive: true, force: true })
       logger.info('Cleaned up temp dir', { jobId })
@@ -260,7 +226,7 @@ export async function renderVideo(
 }
 
 /**
- * 处理预生成代码
+ * 澶勭悊棰勭敓鎴愪唬鐮?
  */
 export async function handlePreGeneratedCode(
   jobId: string,
@@ -269,14 +235,14 @@ export async function handlePreGeneratedCode(
   preGeneratedCode: string,
   timings: Record<string, number>,
   jobData: VideoJobData
-): Promise<TaskResult> {
+): Promise<RenderResult> {
   logger.info('Using pre-generated code from frontend', {
     jobId,
     codeLength: preGeneratedCode.length,
     hasCustomApi: !!jobData.customApiConfig
   })
 
-  // 直接进入渲染阶段
+  // 鐩存帴杩涘叆娓叉煋闃舵
   const renderStart = Date.now()
   const renderResult = await renderVideo(jobId, concept, quality, {
     manimCode: preGeneratedCode,
@@ -285,10 +251,6 @@ export async function handlePreGeneratedCode(
   }, timings, jobData.customApiConfig, jobData.videoConfig, jobData.promptOverrides)
   timings.render = Date.now() - renderStart
 
-  // 存储结果
-  const storeStart = Date.now()
-  timings.store = Date.now() - storeStart
-
   logger.info('Job completed (pre-generated code)', { jobId, timings })
-  return { success: true, source: 'custom-api', timings }
+  return renderResult
 }

src/queues/processors/steps/storage-step.ts

@@ -16,9 +16,11 @@ const logger = createLogger('StorageStep')
  */
 export async function storeResult(
   renderResult: RenderResult,
-  _timings: Record<string, number>
+  _timings: Record<string, number>,
+  options: { skipCache?: boolean } = {}
 ): Promise<void> {
   const { jobId, concept, manimCode, usedAI, generationType, quality, videoUrl, renderPeakMemoryMB } = renderResult
+  const { skipCache = false } = options
 
   // 存储到 Redis（用于 API 查询）
   await storeJobResult(jobId, {

src/queues/processors/video.processor.ts

@@ -7,6 +7,7 @@
 
 import { videoQueue } from '../../config/bull'
 import { storeJobResult } from '../../services/job-store'
+import { generateEditedManimCode } from '../../services/code-edit'
 import { ensureJobNotCancelled } from '../../services/job-cancel'
 import { clearJobCancelled } from '../../services/job-cancel-store'
 import { JobCancelledError } from '../../utils/errors'
@@ -26,9 +27,9 @@ const logger = createLogger('VideoProcessor')
  */
 videoQueue.process(async (job) => {
   const data = job.data as VideoJobData
-  const { jobId, concept, quality, forceRefresh = false, preGeneratedCode, promptOverrides } = data
+  const { jobId, concept, quality, forceRefresh = false, preGeneratedCode, editCode, editInstructions, promptOverrides } = data
 
-  logger.info('Processing video job', { jobId, concept, quality, hasPreGeneratedCode: !!preGeneratedCode })
+  logger.info('Processing video job', { jobId, concept, quality, hasPreGeneratedCode: !!preGeneratedCode, hasEditRequest: !!editInstructions })
 
   // 阶段时长追踪
   const timings: Record<string, number> = {}
@@ -38,7 +39,55 @@ videoQueue.process(async (job) => {
     // 如果有预生成代码，跳过缓存和 AI 生成阶段，直接渲染
     if (preGeneratedCode) {
       await ensureJobNotCancelled(jobId, job)
-      return await handlePreGeneratedCode(jobId, concept, quality, preGeneratedCode, timings, data)
+      const renderResult = await handlePreGeneratedCode(jobId, concept, quality, preGeneratedCode, timings, data)
+
+      const storeStart = Date.now()
+      await storeResult(renderResult, timings, { skipCache: true })
+      timings.store = Date.now() - storeStart
+      timings.total = timings.render + timings.store
+
+      logger.info('Job completed (pre-generated code)', { jobId, timings })
+      return { success: true, source: 'pre-generated', timings }
+    }
+
+    // AI 修改流程
+    if (editCode && editInstructions) {
+      await ensureJobNotCancelled(jobId, job)
+      await storeJobStage(jobId, 'generating')
+      const editStart = Date.now()
+      const editedCode = await generateEditedManimCode(concept, editInstructions, editCode, data.customApiConfig, promptOverrides)
+      timings.edit = Date.now() - editStart
+
+      if (!editedCode) {
+        throw new Error('AI 修改未返回有效代码')
+      }
+
+      await ensureJobNotCancelled(jobId, job)
+      const renderStart = Date.now()
+      const renderResult = await renderVideo(
+        jobId,
+        concept,
+        quality,
+        {
+          manimCode: editedCode,
+          usedAI: true,
+          generationType: 'ai-edit'
+        },
+        timings,
+        data.customApiConfig,
+        data.videoConfig,
+        promptOverrides,
+        () => storeJobStage(jobId, 'rendering')
+      )
+      timings.render = Date.now() - renderStart
+
+      const storeStart = Date.now()
+      await storeResult(renderResult, timings, { skipCache: true })
+      timings.store = Date.now() - storeStart
+      timings.total = timings.edit + timings.render + timings.store
+
+      logger.info('Job completed', { jobId, source: 'ai-edit', timings })
+      return { success: true, source: 'ai-edit', timings }
     }
 
     // Step 1: 检查缓存

src/routes/generate.route.ts

@@ -73,13 +73,15 @@ const bodySchema = z.object({
     system: z.object({
       conceptDesigner: z.string().max(20000).optional(),
       codeGeneration: z.string().max(20000).optional(),
-      codeRetry: z.string().max(20000).optional()
+      codeRetry: z.string().max(20000).optional(),
+      codeEdit: z.string().max(20000).optional()
     }).optional(),
     user: z.object({
       conceptDesigner: z.string().max(20000).optional(),
       codeGeneration: z.string().max(20000).optional(),
       codeRetryInitial: z.string().max(20000).optional(),
-      codeRetryFix: z.string().max(20000).optional()
+      codeRetryFix: z.string().max(20000).optional(),
+      codeEdit: z.string().max(20000).optional()
     }).optional()
   }).optional(),
   /** 视频配置 */

src/routes/index.ts

@@ -7,6 +7,7 @@
 
 import express from 'express'
 import generateRouter from './generate.route'
+import modifyRouter from './modify.route'
 import jobStatusRouter from './job-status.route'
 import jobCancelRouter from './job-cancel.route'
 import promptsRouter from './prompts.route'
@@ -20,6 +21,7 @@ router.use(healthRouter)
 
 // 挂载 API 路由（使用 /api 前缀）
 router.use('/api', generateRouter)
+router.use('/api', modifyRouter)
 router.use('/api', jobStatusRouter)
 router.use('/api', jobCancelRouter)
 router.use('/api', promptsRouter)

src/routes/modify.route.ts

@@ -0,0 +1,142 @@
+/**
+ * AI 修改路由
+ * POST /api/modify - 基于现有代码进行 AI 修改并渲染
+ */
+
+import express from 'express'
+import { z } from 'zod'
+import { v4 as uuidv4 } from 'uuid'
+import { videoQueue } from '../config/bull'
+import { storeJobStage } from '../services/job-store'
+import { createLogger } from '../utils/logger'
+import { AuthenticationError, ValidationError } from '../utils/errors'
+import { asyncHandler } from '../middlewares/error-handler'
+import { authMiddleware } from '../middlewares/auth.middleware'
+import type { GenerateResponse } from '../types'
+
+const router = express.Router()
+const logger = createLogger('ModifyRoute')
+
+function extractToken(authHeader: string | string[] | undefined): string {
+  if (!authHeader) return ''
+
+  if (typeof authHeader === 'string') {
+    return authHeader.replace(/^Bearer\s+/i, '')
+  }
+  if (Array.isArray(authHeader)) {
+    return authHeader[0]?.replace(/^Bearer\s+/i, '') || ''
+  }
+  return ''
+}
+
+function hasPromptOverrides(promptOverrides: any): boolean {
+  if (!promptOverrides) return false
+  const system = promptOverrides.system || {}
+  const user = promptOverrides.user || {}
+  return Object.values(system).some((value) => typeof value === 'string' && value.trim().length > 0) ||
+    Object.values(user).some((value) => typeof value === 'string' && value.trim().length > 0)
+}
+
+function requirePromptOverrideAuth(req: express.Request): void {
+  const manimcatApiKey = process.env.MANIMCAT_API_KEY
+  if (!manimcatApiKey) {
+    throw new AuthenticationError('Prompt overrides require MANIMCAT_API_KEY to be set.')
+  }
+
+  const token = extractToken(req.headers?.authorization)
+  if (!token || token !== manimcatApiKey) {
+    throw new AuthenticationError('Prompt overrides require a valid MANIMCAT_API_KEY token.')
+  }
+}
+
+const bodySchema = z.object({
+  concept: z.string().min(1, '概念不能为空'),
+  quality: z.enum(['low', 'medium', 'high']).optional().default('low'),
+  instructions: z.string().min(1, '修改意见不能为空'),
+  code: z.string().min(1, '原始代码不能为空'),
+  customApiConfig: z.object({
+    apiUrl: z.string(),
+    apiKey: z.string(),
+    model: z.string()
+  }).optional(),
+  promptOverrides: z.object({
+    system: z.object({
+      conceptDesigner: z.string().max(20000).optional(),
+      codeGeneration: z.string().max(20000).optional(),
+      codeRetry: z.string().max(20000).optional(),
+      codeEdit: z.string().max(20000).optional()
+    }).optional(),
+    user: z.object({
+      conceptDesigner: z.string().max(20000).optional(),
+      codeGeneration: z.string().max(20000).optional(),
+      codeRetryInitial: z.string().max(20000).optional(),
+      codeRetryFix: z.string().max(20000).optional(),
+      codeEdit: z.string().max(20000).optional()
+    }).optional()
+  }).optional(),
+  videoConfig: z.object({
+    quality: z.enum(['low', 'medium', 'high']).optional(),
+    frameRate: z.number().int().min(1).max(120).optional(),
+    timeout: z.number().optional()
+  }).optional()
+})
+
+async function handleModifyRequest(req: express.Request, res: express.Response) {
+  const parsed = bodySchema.parse(req.body)
+  const { concept, quality, instructions, code, customApiConfig, promptOverrides, videoConfig } = parsed
+
+  if (hasPromptOverrides(promptOverrides)) {
+    requirePromptOverrideAuth(req)
+  }
+
+  const sanitizedConcept = concept.trim().replace(/\s+/g, ' ')
+  if (sanitizedConcept.length === 0) {
+    throw new ValidationError('提供的概念为空', { concept })
+  }
+
+  const sanitizedInstructions = instructions.trim()
+  if (!sanitizedInstructions) {
+    throw new ValidationError('修改意见不能为空', { instructions })
+  }
+
+  const jobId = uuidv4()
+
+  logger.info('收到 AI 修改请求', {
+    jobId,
+    concept: sanitizedConcept,
+    quality,
+    hasCode: !!code,
+    videoConfig
+  })
+
+  await storeJobStage(jobId, 'generating')
+
+  await videoQueue.add(
+    {
+      jobId,
+      concept: sanitizedConcept,
+      quality,
+      editCode: code,
+      editInstructions: sanitizedInstructions,
+      customApiConfig,
+      promptOverrides,
+      videoConfig,
+      timestamp: new Date().toISOString()
+    },
+    { jobId }
+  )
+
+  const response: GenerateResponse = {
+    success: true,
+    jobId,
+    message: 'AI 修改已开始',
+    status: 'processing'
+  }
+
+  res.status(202).json(response)
+}
+
+router.post('/modify', authMiddleware, asyncHandler(handleModifyRequest))
+
+export default router
+

src/routes/prompts.route.ts

@@ -1,6 +1,11 @@
-import express from 'express'
+import express from 'express'
 
-import { SYSTEM_PROMPTS, generateConceptDesignerPrompt, generateCodeGenerationPrompt } from '../prompts'
+import {
+  SYSTEM_PROMPTS,
+  generateConceptDesignerPrompt,
+  generateCodeGenerationPrompt,
+  generateCodeEditPrompt
+} from '../prompts'
 import { CODE_RETRY_SYSTEM_PROMPT, buildInitialCodePrompt } from '../services/code-retry/prompts'
 import { buildRetryFixPrompt } from '../services/code-retry/manager'
 import type { PromptOverrides } from '../types'
@@ -12,7 +17,9 @@ const PLACEHOLDERS = {
   seed: '{{seed}}',
   sceneDesign: '{{sceneDesign}}',
   errorMessage: '{{errorMessage}}',
-  attempt: '{{attempt}}'
+  attempt: '{{attempt}}',
+  instructions: '{{instructions}}',
+  code: '{{code}}'
 }
 
 function buildDefaultPromptTemplates(): PromptOverrides {
@@ -20,13 +27,15 @@ function buildDefaultPromptTemplates(): PromptOverrides {
     system: {
       conceptDesigner: SYSTEM_PROMPTS.conceptDesigner,
       codeGeneration: SYSTEM_PROMPTS.codeGeneration,
-      codeRetry: CODE_RETRY_SYSTEM_PROMPT
+      codeRetry: CODE_RETRY_SYSTEM_PROMPT,
+      codeEdit: SYSTEM_PROMPTS.codeEdit
     },
     user: {
       conceptDesigner: generateConceptDesignerPrompt(PLACEHOLDERS.concept, PLACEHOLDERS.seed),
       codeGeneration: generateCodeGenerationPrompt(PLACEHOLDERS.concept, PLACEHOLDERS.seed, PLACEHOLDERS.sceneDesign),
       codeRetryInitial: buildInitialCodePrompt(PLACEHOLDERS.concept, PLACEHOLDERS.seed, PLACEHOLDERS.sceneDesign),
-      codeRetryFix: buildRetryFixPrompt(PLACEHOLDERS.concept, PLACEHOLDERS.errorMessage, PLACEHOLDERS.attempt)
+      codeRetryFix: buildRetryFixPrompt(PLACEHOLDERS.concept, PLACEHOLDERS.errorMessage, PLACEHOLDERS.attempt),
+      codeEdit: generateCodeEditPrompt(PLACEHOLDERS.concept, PLACEHOLDERS.instructions, PLACEHOLDERS.code)
     }
   }
 }

src/services/code-edit.ts

@@ -0,0 +1,134 @@
+import OpenAI from 'openai'
+import { createLogger } from '../utils/logger'
+import { SYSTEM_PROMPTS, generateCodeEditPrompt } from '../prompts'
+import type { CustomApiConfig, PromptOverrides } from '../types'
+
+const logger = createLogger('CodeEditService')
+
+const OPENAI_MODEL = process.env.OPENAI_MODEL || 'glm-4-flash'
+const CODER_TEMPERATURE = parseFloat(process.env.AI_TEMPERATURE || '0.7')
+const MAX_TOKENS = parseInt(process.env.AI_MAX_TOKENS || '1200', 10)
+const OPENAI_TIMEOUT = parseInt(process.env.OPENAI_TIMEOUT || '600000', 10)
+
+const CUSTOM_API_URL = process.env.CUSTOM_API_URL?.trim()
+
+let openaiClient: OpenAI | null = null
+
+try {
+  const baseConfig = {
+    timeout: OPENAI_TIMEOUT,
+    defaultHeaders: {
+      'User-Agent': 'ManimCat/1.0'
+    }
+  }
+
+  if (CUSTOM_API_URL) {
+    openaiClient = new OpenAI({
+      ...baseConfig,
+      baseURL: CUSTOM_API_URL,
+      apiKey: process.env.OPENAI_API_KEY
+    })
+  } else {
+    openaiClient = new OpenAI(baseConfig)
+  }
+} catch (error) {
+  logger.warn('OpenAI 客户端初始化失败', { error })
+}
+
+function createCustomClient(config: CustomApiConfig): OpenAI {
+  return new OpenAI({
+    baseURL: config.apiUrl.trim().replace(/\/+$/, ''),
+    apiKey: config.apiKey,
+    timeout: OPENAI_TIMEOUT,
+    defaultHeaders: {
+      'User-Agent': 'ManimCat/1.0'
+    }
+  })
+}
+
+function applyPromptTemplate(template: string, values: Record<string, string>): string {
+  let output = template
+  for (const [key, value] of Object.entries(values)) {
+    output = output.replace(new RegExp(`{{\\s*${key}\\s*}}`, 'g'), value)
+  }
+  return output
+}
+
+function extractCodeFromResponse(text: string): string {
+  if (!text) return ''
+  const sanitized = text.replace(/<think>[\s\S]*?<\/think>/gi, '')
+  const anchorMatch = sanitized.match(/### START ###([\s\S]*?)### END ###/)
+  if (anchorMatch) {
+    return anchorMatch[1].trim()
+  }
+  const codeMatch = sanitized.match(/```(?:python)?\n([\s\S]*?)```/i)
+  if (codeMatch) {
+    return codeMatch[1].trim()
+  }
+  return sanitized.trim()
+}
+
+export async function generateEditedManimCode(
+  concept: string,
+  instructions: string,
+  code: string,
+  customApiConfig?: CustomApiConfig,
+  promptOverrides?: PromptOverrides
+): Promise<string> {
+  const client = customApiConfig ? createCustomClient(customApiConfig) : openaiClient
+
+  if (!client) {
+    logger.warn('OpenAI 客户端不可用')
+    return ''
+  }
+
+  try {
+    const systemPrompt = promptOverrides?.system?.codeEdit || SYSTEM_PROMPTS.codeEdit
+    const userPromptOverride = promptOverrides?.user?.codeEdit
+    const userPrompt = userPromptOverride
+      ? applyPromptTemplate(userPromptOverride, { concept, instructions, code })
+      : generateCodeEditPrompt(concept, instructions, code)
+
+    logger.info('开始 AI 修改代码', { concept })
+
+    const response = await client.chat.completions.create({
+      model: OPENAI_MODEL,
+      messages: [
+        { role: 'system', content: systemPrompt },
+        { role: 'user', content: userPrompt }
+      ],
+      temperature: CODER_TEMPERATURE,
+      max_tokens: MAX_TOKENS
+    })
+
+    const content = response.choices[0]?.message?.content || ''
+    if (!content) {
+      logger.warn('AI 修改返回空内容')
+      return ''
+    }
+
+    const extracted = extractCodeFromResponse(content)
+    logger.info('AI 修改完成', { concept, length: extracted.length })
+    return extracted
+  } catch (error) {
+    if (error instanceof OpenAI.APIError) {
+      logger.error('AI 修改 API 错误', {
+        concept,
+        status: error.status,
+        code: error.code,
+        type: error.type,
+        message: error.message
+      })
+    } else if (error instanceof Error) {
+      logger.error('AI 修改失败', { concept, errorName: error.name, errorMessage: error.message })
+    } else {
+      logger.error('AI 修改失败，未知错误', { concept, error: String(error) })
+    }
+    return ''
+  }
+}
+
+export function isCodeEditAvailable(): boolean {
+  return openaiClient !== null
+}
+

src/services/code-retry/types.ts

@@ -1,11 +1,11 @@
-/**
- * Code Retry Service - 类型定义
+/**
+ * Code Retry Service - 绫诲瀷瀹氫箟
  */
 
 import type { CustomApiConfig, PromptOverrides } from '../../types'
 
 /**
- * 对话消息类型
+ * 瀵硅瘽娑堟伅绫诲瀷
  */
 export interface ChatMessage {
   role: 'system' | 'user' | 'assistant'
@@ -13,18 +13,19 @@ export interface ChatMessage {
 }
 
 /**
- * 代码重试上下文
- * 维护完整的对话历史
+ * 浠ｇ爜閲嶈瘯涓婁笅鏂?
+ * 缁存姢瀹屾暣鐨勫璇濆巻鍙?
  */
 export interface CodeRetryContext {
   concept: string
   sceneDesign: string
-  originalPrompt: string // 原始写代码的提示词
-  messages: ChatMessage[] // 完整对话历史
+  originalPrompt: string
+  messages: ChatMessage[]
+  promptOverrides?: PromptOverrides
 }
 
 /**
- * 代码重试选项
+ * 浠ｇ爜閲嶈瘯閫夐」
  */
 export interface CodeRetryOptions {
   context: CodeRetryContext
@@ -32,7 +33,7 @@ export interface CodeRetryOptions {
 }
 
 /**
- * 代码重试结果
+ * 浠ｇ爜閲嶈瘯缁撴灉
  */
 export interface CodeRetryResult {
   success: boolean
@@ -42,7 +43,7 @@ export interface CodeRetryResult {
 }
 
 /**
- * 渲染结果
+ * 娓叉煋缁撴灉
  */
 export interface RenderResult {
   success: boolean
@@ -52,7 +53,7 @@ export interface RenderResult {
 }
 
 /**
- * 重试管理器结果
+ * 閲嶈瘯绠＄悊鍣ㄧ粨鏋?
  */
 export interface RetryManagerResult {
   code: string

src/types/index.ts

@@ -1,40 +1,40 @@
-/**
+/**
  * Type Definitions
- * 全局类型定义
+ * 鍏ㄥ眬绫诲瀷瀹氫箟
  */
 
 /**
- * 视频质量选项
+ * 瑙嗛璐ㄩ噺閫夐」
  */
 export type VideoQuality = 'low' | 'medium' | 'high'
 
 /**
- * 视频配置
+ * 瑙嗛閰嶇疆
  */
 export interface VideoConfig {
-  /** 默认质量 */
+  /** 榛樿璐ㄩ噺 */
   quality: VideoQuality
-  /** 帧率 */
+  /** 甯х巼 */
   frameRate: number
 }
 
 /**
- * 任务状态
+ * 浠诲姟鐘舵€?
  */
 export type JobStatus = 'queued' | 'processing' | 'completed' | 'failed'
 
 /**
- * 处理阶段
+ * 澶勭悊闃舵
  */
 export type ProcessingStage = 'analyzing' | 'generating' | 'refining' | 'rendering' | 'still-rendering'
 
 /**
- * 生成类型
+ * 鐢熸垚绫诲瀷
  */
 export type GenerationType = 'template' | 'ai' | 'cached'
 
 /**
- * 自定义 API 配置
+ * 鑷畾涔?API 閰嶇疆
  */
 export interface CustomApiConfig {
   apiUrl: string
@@ -50,17 +50,19 @@ export interface PromptOverrides {
     conceptDesigner?: string
     codeGeneration?: string
     codeRetry?: string
+    codeEdit?: string
   }
   user?: {
     conceptDesigner?: string
     codeGeneration?: string
     codeRetryInitial?: string
     codeRetryFix?: string
+    codeEdit?: string
   }
 }
 
 /**
- * 视频生成任务数据
+ * 瑙嗛鐢熸垚浠诲姟鏁版嵁
  */
 export interface VideoJobData {
   jobId: string
@@ -68,16 +70,22 @@ export interface VideoJobData {
   quality: VideoQuality
   forceRefresh?: boolean
   timestamp: string
-  /** 预生成的代码（使用自定义 AI 时） */
+  /** 棰勭敓鎴愮殑浠ｇ爜锛堜娇鐢ㄨ嚜瀹氫箟 AI 鏃讹級 */
   preGeneratedCode?: string
-  /** 自定义 API 配置（用于代码修复） */
+  /** AI 淇敼鏃剁殑鍘熷浠ｇ爜 */
+  editCode?: string
+  /** AI 淇敼鏃剁殑鐢ㄦ埛鎸囦护 */
+  editInstructions?: string
+  /** 鑷畾涔?API 閰嶇疆锛堢敤浜庝唬鐮佷慨澶嶏級 */
   customApiConfig?: CustomApiConfig
-  /** 视频配置 */
+  /** 瑙嗛閰嶇疆 */
   videoConfig?: VideoConfig
+  /** Prompt 覆盖 */
+  promptOverrides?: PromptOverrides
 }
 
 /**
- * 任务结果 - 完成状态
+ * 浠诲姟缁撴灉 - 瀹屾垚鐘舵€?
  */
 export interface CompletedJobResult {
   status: 'completed'
@@ -93,7 +101,7 @@ export interface CompletedJobResult {
 }
 
 /**
- * 任务结果 - 失败状态
+ * 浠诲姟缁撴灉 - 澶辫触鐘舵€?
  */
 export interface FailedJobResult {
   status: 'failed'
@@ -106,12 +114,12 @@ export interface FailedJobResult {
 }
 
 /**
- * 任务结果联合类型
+ * 浠诲姟缁撴灉鑱斿悎绫诲瀷
  */
 export type JobResult = CompletedJobResult | FailedJobResult
 
 /**
- * 概念缓存数据
+ * 姒傚康缂撳瓨鏁版嵁
  */
 export interface ConceptCacheData {
   jobId: string
@@ -126,7 +134,7 @@ export interface ConceptCacheData {
 }
 
 /**
- * API 请求 - 生成视频
+ * API 璇锋眰 - 鐢熸垚瑙嗛
  */
 export interface GenerateRequest {
   concept: string
@@ -136,7 +144,20 @@ export interface GenerateRequest {
 }
 
 /**
- * API 响应 - 生成视频
+ * API 璇锋眰 - AI 淇敼
+ */
+export interface ModifyRequest {
+  concept: string
+  quality?: VideoQuality
+  instructions: string
+  code: string
+  promptOverrides?: PromptOverrides
+  videoConfig?: VideoConfig
+  customApiConfig?: CustomApiConfig
+}
+
+/**
+ * API 鍝嶅簲 - 鐢熸垚瑙嗛
  */
 export interface GenerateResponse {
   success: boolean
@@ -146,7 +167,7 @@ export interface GenerateResponse {
 }
 
 /**
- * API 响应 - 任务状态（处理中）
+ * API 鍝嶅簲 - 浠诲姟鐘舵€侊紙澶勭悊涓級
  */
 export interface JobStatusProcessingResponse {
   status: 'processing' | 'queued'
@@ -156,8 +177,8 @@ export interface JobStatusProcessingResponse {
 }
 
 /**
- * API 响应 - 任务状态（完成）
- * 与前端 api.ts JobResult 类型兼容
+ * API 鍝嶅簲 - 浠诲姟鐘舵€侊紙瀹屾垚锛?
+ * 涓庡墠绔?api.ts JobResult 绫诲瀷鍏煎
  */
 export interface JobStatusCompletedResponse {
   status: 'completed'
@@ -172,8 +193,8 @@ export interface JobStatusCompletedResponse {
 }
 
 /**
- * API 响应 - 任务状态（失败）
- * 与前端 api.ts JobResult 类型兼容
+ * API 鍝嶅簲 - 浠诲姟鐘舵€侊紙澶辫触锛?
+ * 涓庡墠绔?api.ts JobResult 绫诲瀷鍏煎
  */
 export interface JobStatusFailedResponse {
   status: 'failed'
@@ -185,7 +206,7 @@ export interface JobStatusFailedResponse {
 }
 
 /**
- * API 响应 - 任务状态联合类型
+ * API 鍝嶅簲 - 浠诲姟鐘舵€佽仈鍚堢被鍨?
  */
 export type JobStatusResponse =
   | JobStatusProcessingResponse
@@ -193,7 +214,7 @@ export type JobStatusResponse =
   | JobStatusFailedResponse
 
 /**
- * API 响应 - 健康检查
+ * API 鍝嶅簲 - 鍋ュ悍妫€鏌?
  */
 export interface HealthCheckResponse {
   status: 'ok' | 'degraded' | 'down'
@@ -214,7 +235,7 @@ export interface HealthCheckResponse {
 }
 
 /**
- * API 错误响应
+ * API 閿欒鍝嶅簲
  */
 export interface ErrorResponse {
   error: string
@@ -223,7 +244,7 @@ export interface ErrorResponse {
 }
 
 /**
- * Bull 任务进度数据
+ * Bull 浠诲姟杩涘害鏁版嵁
  */
 export interface JobProgress {
   step: string
@@ -232,7 +253,7 @@ export interface JobProgress {
 }
 
 /**
- * Manim 渲染选项
+ * Manim 娓叉煋閫夐」
  */
 export interface ManimRenderOptions {
   quality: VideoQuality
@@ -242,9 +263,10 @@ export interface ManimRenderOptions {
 }
 
 /**
- * 缓存查询结果
+ * 缂撳瓨鏌ヨ缁撴灉
  */
 export interface CacheCheckResult {
   hit: boolean
   data?: ConceptCacheData
 }
+