From dbc3da4e1f938ca90f39bd2a2fbf2ea8f8b9000c Mon Sep 17 00:00:00 2001
From: Victor <victor@storyteller.ai>
Date: Mon, 8 Apr 2024 20:32:35 -0700
Subject: [PATCH] added feature flag checks for /style-video and meida pages

---
 .../fakeyou.com/src/v2/view/pages/media/MediaPage.tsx     | 4 ++--
 .../src/v2/view/pages/style-video/StyleVideo.tsx          | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/packages/fakeyou.com/src/v2/view/pages/media/MediaPage.tsx b/packages/fakeyou.com/src/v2/view/pages/media/MediaPage.tsx
index 243be34661..06e64b7ac4 100644
--- a/packages/fakeyou.com/src/v2/view/pages/media/MediaPage.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/media/MediaPage.tsx
@@ -41,7 +41,7 @@ import { EngineMediaPanel } from "./components/EngineMediaPanel/EngineMediaPanel
 import { GetMediaFileTitle } from "common/GetMediaFileTitle";
 
 export default function MediaPage() {
-  const { canEditMediaFile, user } = useSession();
+  const { canAccessStudio, canEditMediaFile, user } = useSession();
   const { token } = useParams<{ token: string }>();
   const bookmarks = useBookmarks();
   const ratings = useRatings();
@@ -540,7 +540,7 @@ export default function MediaPage() {
                     }}
                   />
                 ) : null}
-                {mediaFile?.media_type === MediaFileType.Video ? (
+                { canAccessStudio && mediaFile?.media_type === MediaFileType.Video ? (
                   <Button
                     {...{
                       icon: faArrowRightArrowLeft,
diff --git a/packages/fakeyou.com/src/v2/view/pages/style-video/StyleVideo.tsx b/packages/fakeyou.com/src/v2/view/pages/style-video/StyleVideo.tsx
index 45a4a51874..398d4283cb 100644
--- a/packages/fakeyou.com/src/v2/view/pages/style-video/StyleVideo.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/style-video/StyleVideo.tsx
@@ -1,7 +1,7 @@
 import React, { useState } from "react";
 import { FrontendInferenceJobType } from "@storyteller/components/src/jobs/InferenceJob";
 import { v4 as uuidv4 } from "uuid";
-import { useInferenceJobs } from "hooks";
+import { useInferenceJobs, useSession } from "hooks";
 import { Button, Container, Panel, TextArea, TempSelect as Select } from "components/common";
 import { EntityInput } from "components/entities";
 import {
@@ -14,7 +14,7 @@ import { STYLE_OPTIONS } from "common/StyleOptions";
 
 export default function StyleVideo() {
   const { mediaToken: pageMediaToken } = useParams<{ mediaToken: string }>();
-
+  const { studioAccessCheck } = useSession();
   const [style, styleSet] = useState("anime_2d_flat");
   const [mediaToken, mediaTokenSet] = useState(pageMediaToken || "");
   const [prompt, promptSet] = useState("");
@@ -51,7 +51,7 @@ export default function StyleVideo() {
     }
   });
 
-  return <Container {...{ className: "fy-style-video-page mt-5", type: "panel" }}>
+  return studioAccessCheck(<Container {...{ className: "fy-style-video-page mt-5", type: "panel" }}>
       <Panel {...{ padding: true }}>
         <header className="d-flex gap-3 flex-wrap">
           <h1 className="fw-semibold">
@@ -94,5 +94,5 @@ export default function StyleVideo() {
           }} />
         </div>
       </Panel>
-   </Container>;
+   </Container>);
 };
\ No newline at end of file