From 55aa101f0d3882b7fae319d95f3c04c4b342f00a Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Wed, 7 Feb 2024 03:50:11 -0500
Subject: [PATCH] disable weights views for non-Studio

---
 packages/fakeyou.com/src/v2/view/PageContainer.tsx  |  1 +
 .../src/v2/view/pages/weight/WeightPage.tsx         | 13 +++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/packages/fakeyou.com/src/v2/view/PageContainer.tsx b/packages/fakeyou.com/src/v2/view/PageContainer.tsx
index f7631fd27a..e940538c04 100644
--- a/packages/fakeyou.com/src/v2/view/PageContainer.tsx
+++ b/packages/fakeyou.com/src/v2/view/PageContainer.tsx
@@ -373,6 +373,7 @@ class PageContainer extends React.Component<
                   render={props => (
                     <WeightPage
                       key={props.match.params.weight_token}
+                      sessionWrapper={this.props.sessionWrapper}
                       sessionSubscriptionsWrapper={
                         this.props.sessionSubscriptionsWrapper
                       }
diff --git a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
index 3ba6c1ed19..cb357b14c0 100644
--- a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
@@ -35,8 +35,11 @@ import WeightCoverImage from "components/common/WeightCoverImage";
 import { BucketConfig } from "@storyteller/components/src/api/BucketConfig";
 import SdInferencePanel from "./inference_panels/SdInferencePanel";
 import SdCoverImagePanel from "./cover_image_panels/SdCoverImagePanel";
+import { StudioNotAvailable } from "v2/view/_common/StudioNotAvailable";
+import { SessionWrapper } from "@storyteller/components/src/session/SessionWrapper";
 
 interface WeightProps {
+  sessionWrapper: SessionWrapper;
   sessionSubscriptionsWrapper: SessionSubscriptionsWrapper;
   inferenceJobs: Array<InferenceJob>;
   ttsInferenceJobs: Array<TtsInferenceJob>;
@@ -49,6 +52,7 @@ interface WeightProps {
 }
 
 export default function WeightPage({
+  sessionWrapper,
   sessionSubscriptionsWrapper,
   inferenceJobs,
   ttsInferenceJobs,
@@ -100,6 +104,15 @@ export default function WeightPage({
 
   const deleteWeight = () => remove(!!user?.can_ban_users);
 
+  switch (weight?.weight_type) {
+    case WeightType.SD_15:
+    case WeightType.SDXL:
+    case WeightType.LORA:
+      if (!sessionWrapper.canAccessStudio()) {
+        return <StudioNotAvailable />
+      }
+  }
+
   function renderWeightComponent(weight: Weight) {
     switch (weight.weight_category) {
       case WeightCategory.TTS: