cleaning up my sins

This commit is contained in:
Andrei Stoica 2024-02-26 11:31:01 -05:00
parent 42c605d992
commit 64bb9f9db3
2 changed files with 117 additions and 103 deletions

View File

@ -1,17 +1,7 @@
import { useEffect, useRef, useState } from "react"; import { useState } from "react";
import { import { ChatMsg, Controls, Feed, Header } from "./components.tsx";
TbBrandOpenai,
TbMicrophone2,
TbPlayerPlay,
TbPlayerStop,
} from "react-icons/tb";
import "./App.css"; import "./App.css";
type ChatMsg = {
role: string;
content: string;
};
let audioBlobs: Array<Blob> = []; let audioBlobs: Array<Blob> = [];
let streamBeingCaptured: MediaStream | null = null; let streamBeingCaptured: MediaStream | null = null;
let mediaRecorder: MediaRecorder | null = null; let mediaRecorder: MediaRecorder | null = null;
@ -69,59 +59,12 @@ function playMsg(msg: ChatMsg) {
console.log("loading audio and playing?"); console.log("loading audio and playing?");
audio.play(); audio.play();
} }
function Header() { function App() {
return (
<header className="header p-3">
<div className="title text-5xl font-extrabold">
Speach to Speech AI example
</div>
</header>
);
}
function Feed(props: { chat: Array<ChatMsg>; setChatStateFn: any }) {
const bottomRef = useRef<any>(null);
const scrollToBottom = () => {
if (bottomRef.current) {
bottomRef.current.scrollIntoView({ behavior: "smooth" });
}
};
useEffect(() => {
scrollToBottom();
console.log("scroll?");
});
return (
<div className="feed grow self-center w-5/6 max-w-screen-lg px-6 py-3 overflow-scroll">
<div className="content-center space-y-2 divide-y-4">
{props.chat.filter((m: ChatMsg) => m.role != "system").map((
m: ChatMsg,
i: number,
) => <Msg key={i} msg={m} />)}
</div>
<div ref={bottomRef} />
</div>
);
}
function Msg(props: { msg: ChatMsg }) {
return (
<div className="Messege text-lg">
<span className="font-bold">
{props.msg.role.toUpperCase()}:
</span>
<br />
<span className="ml-8">
{props.msg.content}
</span>
</div>
);
}
function Controls(props: { setChatStateFn: any, chat: Array<ChatMsg> }) {
const [recordState, setRecordState] = useState(false); const [recordState, setRecordState] = useState(false);
const [chatState, setChatState] = useState([{
role: "system",
content: "You are a helpful assistant.",
}]);
function toggleRecord() { function toggleRecord() {
if (recordState == false) { if (recordState == false) {
@ -141,19 +84,19 @@ function Controls(props: { setChatStateFn: any, chat: Array<ChatMsg> }) {
"body": formData, "body": formData,
}).then((res) => res.json()) }).then((res) => res.json())
.then((res) => { .then((res) => {
props.setChatStateFn((curState: Array<ChatMsg>) => [ setChatState((curState: Array<ChatMsg>) => [
...curState, ...curState,
{ "role": "user", "content": res["user-transcript"] }, { "role": "user", "content": res["user-transcript"] },
]); ]);
fetch("http://100.82.51.22:8001/conversation", { fetch("http://100.82.51.22:8001/conversation", {
"method": "POST", "method": "POST",
"body": JSON.stringify([...props.chat, { "body": JSON.stringify([...chatState, {
"role": "user", "role": "user",
"content": res["user-transcript"], "content": res["user-transcript"],
}]), }]),
}).then((res) => res.json()) }).then((res) => res.json())
.then((res) => { .then((res) => {
props.setChatStateFn(( setChatState((
curState: Array<ChatMsg>, curState: Array<ChatMsg>,
) => [...curState, res]); ) => [...curState, res]);
console.log("attempting to play result"); console.log("attempting to play result");
@ -162,41 +105,6 @@ function Controls(props: { setChatStateFn: any, chat: Array<ChatMsg> }) {
}); });
} }
return (
<div className="controls self-center flex justify-evenly p-5 text-5xl border-2 border-b-0 w-1/2 max-w-screen-sm min-w-fit">
<button
onClick={() => toggleRecord()}
className={"inline-flex " + (recordState ? "text-red-500" : "")}
>
{recordState ? <TbPlayerStop /> : <TbMicrophone2 />}
{recordState ? "STOP" : "REC"}
</button>
<button
onClick={() => playRecord()}
className="inline-flex text-green-500"
>
<TbPlayerPlay /> PLAY
</button>
<button
onClick={() => {
sendAudio();
}}
className="inline-flex"
>
<TbBrandOpenai /> SEND
</button>
</div>
);
}
function App() {
const [chatState, setChatState] = useState([{
role: "system",
content: "You are a helpful assistant.",
}]);
return ( return (
<> <>
<div className="h-screen center flex flex-col"> <div className="h-screen center flex flex-col">
@ -205,7 +113,12 @@ function App() {
<hr className="mx-3 border-t-4" /> <hr className="mx-3 border-t-4" />
</div> </div>
<Feed chat={chatState} setChatStateFn={setChatState} /> <Feed chat={chatState} setChatStateFn={setChatState} />
<Controls setChatStateFn={setChatState} chat={chatState} /> <Controls
recButtonOnClick={toggleRecord}
recordState={recordState}
playButtonOnClick={playRecord}
sendButtonOnClick={sendAudio}
/>
</div> </div>
</> </>
); );

View File

@ -0,0 +1,101 @@
import { useEffect, useRef } from "react";
import {
TbBrandOpenai,
TbMicrophone2,
TbPlayerPlay,
TbPlayerStop,
} from "react-icons/tb";
export type ChatMsg = {
role: string;
content: string;
};
export function Header() {
return (
<header className="header p-3">
<div className="title text-5xl font-extrabold">
Speach to Speech AI example
</div>
</header>
);
}
export function Feed(props: { chat: Array<ChatMsg>; setChatStateFn: any }) {
const bottomRef = useRef<any>(null);
const scrollToBottom = () => {
if (bottomRef.current) {
bottomRef.current.scrollIntoView({ behavior: "smooth" });
}
};
useEffect(() => {
scrollToBottom();
console.log("scroll?");
});
return (
<div className="feed grow self-center w-5/6 max-w-screen-lg px-6 py-3 overflow-scroll">
<div className="content-center space-y-2 divide-y-4">
{props.chat.filter((m: ChatMsg) => m.role != "system").map((
m: ChatMsg,
i: number,
) => <Msg key={i} msg={m} />)}
</div>
<div ref={bottomRef} />
</div>
);
}
export function Msg(props: { msg: ChatMsg }) {
return (
<div className="Messege text-lg">
<span className="font-bold">
{props.msg.role.toUpperCase()}:
</span>
<br />
<span className="ml-8">
{props.msg.content}
</span>
</div>
);
}
export function Controls(
props: {
recButtonOnClick: Function;
recordState: Boolean;
playButtonOnClick: Function;
sendButtonOnClick: Function;
},
) {
return (
<div className="controls self-center flex justify-evenly p-5 text-5xl border-2 border-b-0 w-1/2 max-w-screen-sm min-w-fit">
<button
onClick={() => props.recButtonOnClick()}
className={"inline-flex " + (props.recordState ? "text-red-500" : "")}
>
{props.recordState ? <TbPlayerStop /> : <TbMicrophone2 />}
{props.recordState ? "STOP" : "REC"}
</button>
<button
onClick={() => props.playButtonOnClick()}
className="inline-flex text-green-500"
>
<TbPlayerPlay /> PLAY
</button>
<button
onClick={() => {
props.sendButtonOnClick();
}}
className="inline-flex"
>
<TbBrandOpenai /> SEND
</button>
</div>
);
}