diff --git a/apps/drive-integration/src/hooks/useWorkflowAgent.ts b/apps/drive-integration/src/hooks/useWorkflowAgent.ts
index 1c5c51d4fc..04ce098dab 100644
--- a/apps/drive-integration/src/hooks/useWorkflowAgent.ts
+++ b/apps/drive-integration/src/hooks/useWorkflowAgent.ts
@@ -127,6 +127,14 @@ const getBackendWorkflowFailureReason = (runData: AgentRunData): WorkflowFailure
     return WorkflowFailureReason.GOOGLE_DOCS_NOT_FOUND;
   }
 
+  if (workflowFailure.code === WorkflowFailureReason.AI_SERVICE_UNAVAILABLE) {
+    return WorkflowFailureReason.AI_SERVICE_UNAVAILABLE;
+  }
+
+  if (workflowFailure.code === WorkflowFailureReason.APP_NOT_INSTALLED) {
+    return WorkflowFailureReason.APP_NOT_INSTALLED;
+  }
+
   if (workflowFailure.code === WorkflowFailureReason.GENERIC) {
     return WorkflowFailureReason.GENERIC;
   }
@@ -146,6 +154,14 @@ const getWorkflowFailureMessage = (
     return ERROR_MESSAGES.GOOGLE_DOCS_NOT_FOUND;
   }
 
+  if (failureReason === WorkflowFailureReason.AI_SERVICE_UNAVAILABLE) {
+    return ERROR_MESSAGES.AI_SERVICE_UNAVAILABLE;
+  }
+
+  if (failureReason === WorkflowFailureReason.APP_NOT_INSTALLED) {
+    return ERROR_MESSAGES.APP_NOT_INSTALLED;
+  }
+
   return getRunErrorMessage(runData);
 };
 
diff --git a/apps/drive-integration/src/locations/Page/components/mainpage/ModalOrchestrator.tsx b/apps/drive-integration/src/locations/Page/components/mainpage/ModalOrchestrator.tsx
index ecfc233ba3..31e6431bee 100644
--- a/apps/drive-integration/src/locations/Page/components/mainpage/ModalOrchestrator.tsx
+++ b/apps/drive-integration/src/locations/Page/components/mainpage/ModalOrchestrator.tsx
@@ -175,6 +175,30 @@ export const ModalOrchestrator = forwardRef<ModalOrchestratorHandle, ModalOrches
         return;
       }
 
+      if (
+        error instanceof WorkflowRunError &&
+        error.reason === WorkflowFailureReason.AI_SERVICE_UNAVAILABLE
+      ) {
+        setPreviewErrorState({
+          reason: WorkflowFailureReason.AI_SERVICE_UNAVAILABLE,
+          title: 'AI service temporarily unavailable',
+          message: ERROR_MESSAGES.AI_SERVICE_UNAVAILABLE,
+        });
+        return;
+      }
+
+      if (
+        error instanceof WorkflowRunError &&
+        error.reason === WorkflowFailureReason.APP_NOT_INSTALLED
+      ) {
+        setPreviewErrorState({
+          reason: WorkflowFailureReason.APP_NOT_INSTALLED,
+          title: 'App not installed in this environment',
+          message: ERROR_MESSAGES.APP_NOT_INSTALLED,
+        });
+        return;
+      }
+
       setPreviewErrorState({
         reason: WorkflowFailureReason.GENERIC,
         title: 'Unable to generate preview',
diff --git a/apps/drive-integration/src/types/workflow.ts b/apps/drive-integration/src/types/workflow.ts
index 87f05ff82f..58601281c5 100644
--- a/apps/drive-integration/src/types/workflow.ts
+++ b/apps/drive-integration/src/types/workflow.ts
@@ -14,6 +14,8 @@ export enum WorkflowFailureReason {
   GENERIC = 'generic',
   GOOGLE_DRIVE_AUTH_EXPIRED = 'google-drive-auth-expired',
   GOOGLE_DOCS_NOT_FOUND = 'google-docs-not-found',
+  AI_SERVICE_UNAVAILABLE = 'ai-service-unavailable',
+  APP_NOT_INSTALLED = 'app-not-installed',
 }
 
 export interface WorkflowFailure {
diff --git a/apps/drive-integration/src/utils/constants/messages.ts b/apps/drive-integration/src/utils/constants/messages.ts
index 4841a6f157..73b02a57fe 100644
--- a/apps/drive-integration/src/utils/constants/messages.ts
+++ b/apps/drive-integration/src/utils/constants/messages.ts
@@ -10,6 +10,10 @@ export const ERROR_MESSAGES = {
     'Your Drive connection has expired or is no longer valid. Reconnect your account to continue generating a preview.',
   GOOGLE_DOCS_NOT_FOUND:
     'Google Doc not found. Make sure the document exists and your Google account has access to it.',
+  AI_SERVICE_UNAVAILABLE:
+    'The AI service is temporarily unavailable. Please try again in a few minutes.',
+  APP_NOT_INSTALLED:
+    'The Drive Integration app is not installed in this environment. Install it via Apps > Manage apps and try again.',
 } as const;
 
 export const SUCCESS_MESSAGES = {