remove more experimentinfo directly

Author: aliasaria

src/renderer/components/Experiment/Diffusion/Diffusion.tsx

@@ -41,10 +41,7 @@ import History from './History';
 import Inpainting from './Inpainting';
 import HistoryImageSelector from './HistoryImageSelector';
 import ControlNetModal from './ControlNetModal';
-
-type DiffusionProps = {
-  experimentInfo: any;
-};
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 
 // Helper component for labels with tooltips
 const LabelWithTooltip = ({
@@ -98,7 +95,8 @@ const samplePrompts = [
   'a whimsical and highly detailed illustration of a miniature world inside a glass terrarium, tiny people tending to giant flowers, a small waterfall, magical realism, storybook style.',
 ];
 
-export default function Diffusion({ experimentInfo }: DiffusionProps) {
+export default function Diffusion() {
+  const { experimentInfo } = useExperimentInfo();
   const analytics = useAnalytics();
 
   const initialModel = experimentInfo?.config?.foundation || '';

src/renderer/components/Experiment/Export/Export.tsx

@@ -15,22 +15,20 @@ import {
   Table,
   Typography,
 } from '@mui/joy';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 
 // fetcher used by SWR
 const fetcher = (url: string) => fetch(url).then((res) => res.json());
 
-interface ExportProps {
-  experimentInfo: any;
-}
-
 interface Plugin {
   uniqueId: string;
   name: string;
   description: string;
   model_architectures: string[];
 }
 
-export default function Export({ experimentInfo }: ExportProps) {
+export default function Export() {
+  const { experimentInfo } = useExperimentInfo();
   const [runningPlugin, setRunningPlugin] = useState<string | null>(null);
   const [exportDetailsJobId, setExportDetailsJobId] = useState<number>(-1);
   const [selectedPlugin, setSelectedPlugin] = useState<Plugin | null>(null);

src/renderer/components/Experiment/Rag/index.tsx

@@ -11,20 +11,21 @@ import {
   Typography,
   Option,
   Chip,
-  FormLabel,
 } from '@mui/joy';
-import Documents from '../../Shared/Documents';
-import Query from './Query';
 import useSWR from 'swr';
 
 import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
 import { CogIcon, XCircleIcon } from 'lucide-react';
-import ConfigurePlugin from './ConfigurePlugin';
 import { useState } from 'react';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
+import Documents from '../../Shared/Documents';
+import Query from './Query';
+import ConfigurePlugin from './ConfigurePlugin';
 
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
-export default function DocumentSearch({ experimentInfo, setRagEngine }) {
+export default function DocumentSearch({ setRagEngine }) {
+  const { experimentInfo } = useExperimentInfo();
   const [openConfigureModal, setOpenConfigureModal] = useState(false);
   const {
     data: plugins,
@@ -117,7 +118,7 @@ export default function DocumentSearch({ experimentInfo, setRagEngine }) {
             flexDirection: 'column',
           }}
         >
-          <Documents experimentInfo={experimentInfo} fixedFolder="rag" />
+          <Documents fixedFolder="rag" />
         </Box>
         <Box sx={{ flex: 1, display: 'flex', flexDirection: 'column' }}>
           <Query experimentInfo={experimentInfo} />

src/renderer/components/Experiment/Settings.tsx

@@ -5,14 +5,12 @@ import { Button, Chip, Divider, Switch, Typography } from '@mui/joy';
 
 import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
 import { useState } from 'react';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
-export default function ExperimentSettings({
-  experimentInfo,
-  setExperimentId,
-  experimentInfoMutate,
-}) {
+export default function ExperimentSettings() {
+  const { experimentInfo, setExperimentId } = useExperimentInfo();
   const [showJSON, setShowJSON] = useState(false);
 
   if (!experimentInfo) {

src/renderer/components/MainAppPanel.tsx

@@ -375,66 +375,22 @@ export default function MainAppPanel({ setLogsDrawerOpen = null }) {
         <Route path="/experiment/documents" element={<Documents />} />
         <Route
           path="/experiment/rag"
-          element={
-            <Rag experimentInfo={experimentInfo} setRagEngine={setRagEngine} />
-          }
-        />
-        <Route
-          path="/experiment/export"
-          element={<Export experimentInfo={experimentInfo} />}
-        />
-        <Route
-          path="/experiment/generate"
-          element={<Generate experimentInfo={experimentInfo} />}
-        />
-        <Route
-          path="/experiment/diffusion"
-          element={<Diffusion experimentInfo={experimentInfo} />}
+          element={<Rag setRagEngine={setRagEngine} />}
         />
+        <Route path="/experiment/export" element={<Export />} />
+        <Route path="/experiment/diffusion" element={<Diffusion />} />
         <Route
           path="/plugins"
-          element={
-            <Plugins
-              experimentInfo={experimentInfo}
-              setLogsDrawerOpen={setLogsDrawerOpen}
-            />
-          }
-        />
-        <Route
-          path="/plugins/:pluginName"
-          element={<PluginDetails experimentInfo={experimentInfo} />}
+          element={<Plugins setLogsDrawerOpen={setLogsDrawerOpen} />}
         />
+        <Route path="/plugins/:pluginName" element={<PluginDetails />} />
         <Route path="/api" element={<Api />} />
-        <Route
-          path="/experiment/settings"
-          element={
-            <Settings
-              experimentInfo={experimentInfo}
-              setExperimentId={setExperimentId}
-              experimentInfoMutate={experimentInfoMutate}
-            />
-          }
-        />
-        <Route
-          path="/zoo"
-          element={<ModelZoo experimentInfo={experimentInfo} tab="groups" />}
-        />
-        <Route
-          path="/zoo/local"
-          element={<ModelZoo experimentInfo={experimentInfo} tab="local" />}
-        />
-        <Route
-          path="/zoo/generated"
-          element={<ModelZoo experimentInfo={experimentInfo} tab="generated" />}
-        />
-        <Route
-          path="/zoo/store"
-          element={<ModelZoo experimentInfo={experimentInfo} tab="store" />}
-        />
-        <Route
-          path="/zoo/groups"
-          element={<ModelZoo experimentInfo={experimentInfo} tab="groups" />}
-        />
+        <Route path="/experiment/settings" element={<Settings />} />
+        <Route path="/zoo" element={<ModelZoo tab="groups" />} />
+        <Route path="/zoo/local" element={<ModelZoo tab="local" />} />
+        <Route path="/zoo/generated" element={<ModelZoo tab="generated" />} />
+        <Route path="/zoo/store" element={<ModelZoo tab="store" />} />
+        <Route path="/zoo/groups" element={<ModelZoo tab="groups" />} />
         <Route path="/data" element={<Data />} />
         <Route path="/computer" element={<Computer />} />
         <Route path="/settings" element={<TransformerLabSettings />} />

src/renderer/components/ModelZoo/ModelZoo.tsx

@@ -1,16 +1,15 @@
 /* eslint-disable jsx-a11y/anchor-is-valid */
-import { useState, useCallback } from 'react';
 import Sheet from '@mui/joy/Sheet';
 import { StoreIcon } from 'lucide-react';
 import { Tab, TabList, TabPanel, Tabs } from '@mui/joy';
-import ModelStore from './ModelStore';
-import LocalModels from './LocalModels';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 import { useNavigate } from 'react-router-dom';
+import LocalModels from './LocalModels';
 import ModelGroups from './ModelGroups';
 
-export default function ModelZoo({ experimentInfo, tab = 'store' }) {
+export default function ModelZoo({ tab = 'store' }) {
   const navigate = useNavigate();
-  const DEV_MODE = experimentInfo?.name === 'dev';
+  const { experimentInfo } = useExperimentInfo();
 
   return (
     <Sheet

src/renderer/components/Plugins/PluginDetails.tsx

@@ -39,6 +39,7 @@ import fairyflossTheme from '../Shared/fairyfloss.tmTheme.js';
 
 import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
 import useSWR from 'swr';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext.js';
 
 function ListPluginFiles({
   files,
@@ -147,7 +148,8 @@ function setTheme(editor: any, monaco: any) {
 
 const fetcher = (url) => fetch(url).then((res) => res.json());
 
-export default function PluginDetails({ experimentInfo }) {
+export default function PluginDetails() {
+  const { experimentInfo } = useExperimentInfo();
   let { pluginName } = useParams();
   let { state: plugin } = useLocation();
 

src/renderer/components/Plugins/Plugins.tsx

@@ -14,11 +14,13 @@ import { Circle, StoreIcon } from 'lucide-react';
 import { usePluginStatus } from 'renderer/lib/transformerlab-api-sdk';
 import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
 import { useState } from 'react';
+import { useExperimentInfo } from 'renderer/lib/ExperimentInfoContext';
 import PluginGallery from './PluginGallery';
 import LocalPlugins from './LocalPlugins';
 import OneTimePopup from '../Shared/OneTimePopup';
 
-export default function Plugins({ experimentInfo, setLogsDrawerOpen = null }) {
+export default function Plugins({ setLogsDrawerOpen = null }) {
+  const { experimentInfo } = useExperimentInfo();
   const { data: outdatedPlugins, mutate: outdatePluginsMutate } =
     usePluginStatus(experimentInfo);
   const [installing, setInstalling] = useState(null);