diff --git a/components/agents/AgentGenerationWorkspaceDarkV2.tsx b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
index ecc805559..4682a6db7 100644
--- a/components/agents/AgentGenerationWorkspaceDarkV2.tsx
+++ b/components/agents/AgentGenerationWorkspaceDarkV2.tsx
@@ -1660,6 +1660,8 @@ export function AgentGenerationWorkspaceDarkV2({
   const [activeToolStep, setActiveToolStep] = useState<ToolStepKey>("preset");
   const hasBrandModelStep = Boolean(brandModelPanel) && modelSlotEnabled;
   const hasWorkflowStyleStep = Boolean(workflowStylePanel) || hasBrandModelStep;
+  const shouldShowDirectUploadInStyleStep =
+    isDirectReferenceUploadEnabled && hasWorkflowStyleStep;
   const visibleWorkflowSectionKeys = useMemo(
     () => [
       ...(isPresetStepEnabled ? ["mode"] : []),
@@ -4589,6 +4591,7 @@ export function AgentGenerationWorkspaceDarkV2({
                   </div>
 
                   <div className="space-y-4">
+                    {shouldShowDirectUploadInStyleStep ? attachmentSlotPanel : null}
                     {workflowStylePanel}
                     {hasBrandModelStep ? brandModelPanel : null}
                   </div>
@@ -4647,7 +4650,7 @@ export function AgentGenerationWorkspaceDarkV2({
                     </div>
                   ) : null}
 
-                  {isDirectReferenceUploadEnabled ||
+                  {(!shouldShowDirectUploadInStyleStep && isDirectReferenceUploadEnabled) ||
                   (hasAgentUploadSlots && isAttachmentSlotExpanded)
                     ? attachmentSlotPanel
                     : null}