diff --git a/components/agents/AgentGenerationWorkspaceDarkV2.tsx b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
index 53e1e7878..4f6411bba 100644
--- a/components/agents/AgentGenerationWorkspaceDarkV2.tsx
+++ b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
@@ -2494,7 +2494,9 @@ export function AgentGenerationWorkspaceDarkV2({
     });
     try {
       sendMessage({ role: "user", parts });
-      setInput("");
+      if (selectedAgentPreset?.type !== "image") {
+        setInput("");
+      }
     } catch (error) {
       generationRequestSnapshotRef.current = {
         hiddenPrompt: null,