更新

frontend/src/features/home/ui/script-extraction/useScriptExtraction.ts

@@ -1,4 +1,4 @@
-import { useState, useEffect, useCallback } from "react";
+import { useState, useEffect, useCallback, useRef } from "react";
 import api from "@/shared/api/axios";
 import { ApiResponse, unwrap } from "@/shared/api/types";
 import { toast } from "sonner";
@@ -7,6 +7,7 @@ export type ExtractionStep = "config" | "processing" | "result";
 export type InputTab = "file" | "url";
 
 const VALID_FILE_TYPES = [".mp4", ".mov", ".avi", ".mp3", ".wav", ".m4a"];
+const CUSTOM_PROMPT_KEY = "vigent_rewriteCustomPrompt";
 
 interface UseScriptExtractionOptions {
     isOpen: boolean;
@@ -23,8 +24,19 @@ export const useScriptExtraction = ({ isOpen }: UseScriptExtractionOptions) => {
     const [selectedFile, setSelectedFile] = useState<File | null>(null);
     const [activeTab, setActiveTab] = useState<InputTab>("url");
     const [inputUrl, setInputUrl] = useState("");
+    const [customPrompt, setCustomPrompt] = useState(() => typeof window !== "undefined" ? localStorage.getItem(CUSTOM_PROMPT_KEY) || "" : "");
+    const [showCustomPrompt, setShowCustomPrompt] = useState(false);
 
-    // Reset state when modal opens
+    // Debounced save customPrompt to localStorage
+    const debounceRef = useRef<ReturnType<typeof setTimeout>>(undefined);
+    useEffect(() => {
+        debounceRef.current = setTimeout(() => {
+            localStorage.setItem(CUSTOM_PROMPT_KEY, customPrompt);
+        }, 300);
+        return () => clearTimeout(debounceRef.current);
+    }, [customPrompt]);
+
+    // Reset state when modal opens (customPrompt is persistent, not reset)
     useEffect(() => {
         if (isOpen) {
             setStep("config");
@@ -101,6 +113,9 @@ export const useScriptExtraction = ({ isOpen }: UseScriptExtractionOptions) => {
                 formData.append("url", inputUrl.trim());
             }
             formData.append("rewrite", doRewrite ? "true" : "false");
+            if (doRewrite && customPrompt.trim()) {
+                formData.append("custom_prompt", customPrompt.trim());
+            }
 
             const { data: res } = await api.post<
                 ApiResponse<{ original_script: string; rewritten_script?: string }>
@@ -126,7 +141,7 @@ export const useScriptExtraction = ({ isOpen }: UseScriptExtractionOptions) => {
         } finally {
             setIsLoading(false);
         }
-    }, [activeTab, selectedFile, inputUrl, doRewrite]);
+    }, [activeTab, selectedFile, inputUrl, doRewrite, customPrompt]);
 
     const copyToClipboard = useCallback((text: string) => {
         if (navigator.clipboard && window.isSecureContext) {
@@ -193,10 +208,14 @@ export const useScriptExtraction = ({ isOpen }: UseScriptExtractionOptions) => {
         selectedFile,
         activeTab,
         inputUrl,
+        customPrompt,
+        showCustomPrompt,
         // Setters
         setDoRewrite,
         setActiveTab,
         setInputUrl,
+        setCustomPrompt,
+        setShowCustomPrompt,
         // Handlers
         handleDrag,
         handleDrop,