290 lines
9.3 KiB
TypeScript
290 lines
9.3 KiB
TypeScript
import { Dialog, Transition } from "@headlessui/react";
|
||
import { Fragment, useState } from "react";
|
||
import { FiSettings } from "react-icons/fi";
|
||
import { ImStatsBars } from "react-icons/im";
|
||
import styled from "styled-components";
|
||
import { InferenceResult, PromptInput } from "../types";
|
||
import DebugView from "./DebugView";
|
||
|
||
const ModalContainer = styled.div`
|
||
position: absolute;
|
||
top: 0;
|
||
left: 0;
|
||
width: 100vw;
|
||
height: 100vh;
|
||
background: rgba(0, 0, 0, 0.5);
|
||
display: flex;
|
||
align-items: center;
|
||
justify-content: center;
|
||
`;
|
||
|
||
interface DebugViewProps {
|
||
promptInputs: PromptInput[];
|
||
inferenceResults: InferenceResult[];
|
||
nowPlayingResult: InferenceResult;
|
||
denoising: number;
|
||
setDenoising: (denoising: number) => void;
|
||
seedImage: string;
|
||
setSeedImage: (seedImage: string) => void;
|
||
}
|
||
|
||
export default function Settings({
|
||
promptInputs,
|
||
inferenceResults,
|
||
nowPlayingResult,
|
||
denoising,
|
||
setDenoising,
|
||
seedImage,
|
||
setSeedImage,
|
||
}: DebugViewProps) {
|
||
const [open, setOpen] = useState(false);
|
||
|
||
var classNameCondition = "";
|
||
if (open) {
|
||
classNameCondition =
|
||
"fixed z-20 top-44 right-4 md:top-48 md:right-8 bg-sky-400 w-14 h-14 rounded-full drop-shadow-lg flex justify-center items-center text-white text-2xl hover:bg-sky-500 hover:drop-shadow-2xl";
|
||
} else {
|
||
classNameCondition =
|
||
"fixed z-20 top-44 right-4 md:top-48 md:right-8 bg-slate-100 w-14 h-14 rounded-full drop-shadow-lg flex justify-center items-center text-sky-900 text-2xl hover:text-white hover:bg-sky-600 hover:drop-shadow-2xl";
|
||
}
|
||
|
||
return (
|
||
<>
|
||
<button
|
||
title="Settings"
|
||
className={classNameCondition}
|
||
onClick={() => setOpen(true)}
|
||
>
|
||
<FiSettings />
|
||
</button>
|
||
|
||
<Transition appear show={open} as={Fragment}>
|
||
<Dialog
|
||
as="div"
|
||
className="fixed inset-0 z-20 overflow-y-auto"
|
||
onClose={() => setOpen(false)}
|
||
>
|
||
<div className="min-h-screen px-4 text-center">
|
||
<Transition.Child
|
||
as={Fragment}
|
||
enter="ease-out duration-300"
|
||
enterFrom="opacity-0"
|
||
enterTo="opacity-100"
|
||
leave="ease-in duration-200"
|
||
leaveFrom="opacity-100"
|
||
leaveTo="opacity-0"
|
||
>
|
||
<Dialog.Overlay className="fixed inset-0" />
|
||
</Transition.Child>
|
||
|
||
<span
|
||
className="inline-block h-screen align-middle"
|
||
aria-hidden="true"
|
||
>
|
||
​
|
||
</span>
|
||
<Transition.Child
|
||
as={Fragment}
|
||
enter="ease-out duration-300"
|
||
enterFrom="opacity-0 scale-95"
|
||
enterTo="opacity-100 scale-100"
|
||
leave="ease-in duration-200"
|
||
leaveFrom="opacity-100 scale-100"
|
||
leaveTo="opacity-0 scale-95"
|
||
>
|
||
<ModalContainer key="settings-modal-container">
|
||
<div className="my-8 inline-block w-full max-w-md transform overflow-hidden rounded-2xl bg-white p-6 text-left align-middle shadow-xl transition-all">
|
||
<Dialog.Title
|
||
as="h1"
|
||
className="text-3xl font-medium leading-6 text-gray-900 pb-4"
|
||
>
|
||
Settings
|
||
</Dialog.Title>
|
||
<div className="mt-1">
|
||
<p className=" text-gray-700 text-sm">
|
||
Riffusion generates music from text prompts. Try your
|
||
favorite artists, instruments like saxophone or violin,
|
||
modifiers like arabic or jamaican, genres like jazz or
|
||
gospel, sounds like church bells or rain, or any
|
||
combination. Play with the settings below to explore the
|
||
latent space of sound.
|
||
</p>
|
||
|
||
{/* <input type="range" min="0" max="100" value="40" className="range" /> */}
|
||
|
||
{SeedImageSelector(seedImage, setSeedImage)}
|
||
|
||
{DenoisingSelector(denoising, setDenoising)}
|
||
|
||
{DebugButton(
|
||
promptInputs,
|
||
inferenceResults,
|
||
nowPlayingResult
|
||
)}
|
||
</div>
|
||
|
||
<div className="mt-6">
|
||
<button
|
||
className="relative inline-flex items-center justify-center p-0.5 mb-2 mr-2 overflow-hidden text-sm font-medium text-gray-900 rounded-lg group bg-sky-500 group-hover:from-sky-600 group-hover:to-sky-500 hover:text-white"
|
||
onClick={() => {
|
||
window.open("/about", "_blank");
|
||
setOpen(false);
|
||
}}
|
||
>
|
||
<span className="relative px-5 py-2 transition-all ease-in duration-75 bg-white rounded-md group-hover:bg-opacity-0">
|
||
💁🏻♀️ How does it work?
|
||
</span>
|
||
</button>
|
||
|
||
<button
|
||
type="button"
|
||
className="text-white bg-gradient-to-br from-purple-600 to-sky-500 hover:bg-gradient-to-bl font-medium rounded-lg text-sm px-5 py-2.5 text-center mr-2 mb-2"
|
||
onClick={() => {
|
||
setOpen(false);
|
||
}}
|
||
>
|
||
🎧 Riff
|
||
</button>
|
||
</div>
|
||
</div>
|
||
</ModalContainer>
|
||
</Transition.Child>
|
||
</div>
|
||
</Dialog>
|
||
</Transition>
|
||
</>
|
||
);
|
||
}
|
||
|
||
export function SeedImageSelector(
|
||
seedImage: string,
|
||
setSeedImage: (seedImage: string) => void
|
||
) {
|
||
let selectOptions = [
|
||
["OG Beat", "og_beat"],
|
||
["Agile", "agile"],
|
||
["Marim", "marim"],
|
||
["Motorway", "motorway"],
|
||
["Vibes", "vibes"],
|
||
];
|
||
|
||
let matchedOption = selectOptions.find((x) => x[1] === seedImage);
|
||
if (matchedOption === undefined) {
|
||
matchedOption = [`Custom (${seedImage})`, seedImage];
|
||
selectOptions.push(matchedOption);
|
||
}
|
||
|
||
return (
|
||
<div className="form-control w-full">
|
||
<label className="label">
|
||
<span className="label-text text-gray-700">Seed Image</span>
|
||
</label>
|
||
<select
|
||
className="select select-bordered select-sm"
|
||
onChange={(e) => {
|
||
const newValue = selectOptions.find(
|
||
(x) => x[0] === e.target.value
|
||
)[1];
|
||
|
||
console.log("Setting seed image: ", newValue);
|
||
setSeedImage(newValue);
|
||
}}
|
||
defaultValue={matchedOption[0]}
|
||
>
|
||
<option disabled>Change it up</option>
|
||
{selectOptions.map((x, i) => (
|
||
<option key={i}>{x[0]}</option>
|
||
))}
|
||
</select>
|
||
<p className="label-text-alt text-gray-700 pt-2">
|
||
Used as the base for img2img diffusion. This keeps your riff on beat and
|
||
impacts melodic patterns.
|
||
</p>
|
||
</div>
|
||
);
|
||
}
|
||
|
||
export function DenoisingSelector(
|
||
denoising: number,
|
||
setDenoising: (d: number) => void
|
||
) {
|
||
let selectOptions = [
|
||
["Keep it on beat (0.75)", 0.75],
|
||
["Get a little crazy (0.8)", 0.8],
|
||
["I'm feeling lucky (0.85)", 0.85],
|
||
["What is tempo? (0.95)", 0.95],
|
||
];
|
||
|
||
let matchedOption = selectOptions.find((x) => x[1] === denoising);
|
||
if (matchedOption === undefined) {
|
||
matchedOption = [`Custom (${denoising})`, denoising];
|
||
selectOptions.push(matchedOption);
|
||
}
|
||
|
||
return (
|
||
<div className="form-control w-full">
|
||
<label className="label">
|
||
<span className="label-text text-gray-700">Denoising</span>
|
||
</label>
|
||
<select
|
||
className="select select-bordered select-sm"
|
||
onChange={(e) => {
|
||
const newValue = selectOptions.find(
|
||
(x) => x[0] === e.target.value
|
||
)[1] as number;
|
||
|
||
console.log("Setting denoising: ", newValue);
|
||
setDenoising(newValue);
|
||
}}
|
||
defaultValue={matchedOption[0]}
|
||
>
|
||
<option disabled>How wild to get</option>
|
||
{selectOptions.map((x, i) => (
|
||
<option key={i}>{x[0]}</option>
|
||
))}
|
||
</select>
|
||
<p className="label-text-alt text-gray-700 pt-2">
|
||
The higher the denoising, the more creative the output, and the more
|
||
likely you are to get off beat.
|
||
</p>
|
||
</div>
|
||
);
|
||
}
|
||
|
||
export function DebugButton(promptInputs, inferenceResults, nowPlayingResult) {
|
||
const [debugOpen, debugSetOpen] = useState(false);
|
||
|
||
let buttonClassName = "";
|
||
if (debugOpen) {
|
||
buttonClassName =
|
||
"fixed z-20 top-4 right-6 bg-sky-400 w-10 h-10 rounded-full flex justify-center items-center text-white text-xl hover:bg-sky-500 hover:drop-shadow-2xl";
|
||
} else {
|
||
buttonClassName =
|
||
"fixed z-20 top-4 right-6 bg-sky-100 w-10 h-10 rounded-full flex justify-center items-center text-sky-900 text-xl hover:text-white hover:bg-sky-500 hover:drop-shadow-2xl";
|
||
}
|
||
|
||
return (
|
||
<>
|
||
<button
|
||
title="Debug"
|
||
key="debug-button"
|
||
className={buttonClassName}
|
||
onClick={() => {
|
||
debugSetOpen(true);
|
||
}}
|
||
>
|
||
<ImStatsBars />
|
||
</button>
|
||
|
||
<DebugView
|
||
promptInputs={promptInputs}
|
||
inferenceResults={inferenceResults}
|
||
nowPlayingResult={nowPlayingResult}
|
||
open={debugOpen}
|
||
setOpen={debugSetOpen}
|
||
key="debug-view"
|
||
/>
|
||
</>
|
||
);
|
||
}
|