interstellar_ai/app/backend/InputOutputHandler.tsx

109 lines
3 KiB
TypeScript
Raw Normal View History

2024-09-19 13:02:56 +02:00
"use client"
2024-09-19 15:56:26 +02:00
import React, { useEffect, useRef, useState } from "react";
2024-09-20 10:34:16 +02:00
import ConversationFrontend from "../components/ConversationFrontend";
import InputFrontend from "../components/InputFrontend";
2024-09-19 15:56:26 +02:00
const handleMicClick = () => {
console.log('Mic clicked!');
// Do something when the mic button is clicked
};
const handleResendClick = () => {
console.log('Resend button clicked');
// Handle resend action
};
const handleEditClick = () => {
console.log('Edit button clicked');
// Handle edit action
};
const handleCopyClick = () => {
console.log('Copy button clicked');
// Handle copy action
};
2024-09-19 13:02:56 +02:00
const InputOutputBackend: React.FC = () => {
const [accessToken, setAccessToken] = useState("")
2024-09-19 15:56:26 +02:00
const workerRef = useRef<Worker | null>(null)
const handleSendClick = (message: string) => {
var system = "You give really short answers (maximum of 30 sentences). The following is the chat history."
for (let index = 0; index < messages.length; index++) {
system += messages[index] + " ";
};
HandlePostRequest(message, "phi3.5", system)
addMessage('User: ' + message);
};
const [messages, setMessages] = useState([
'User: Hello!',
'AI: Hi there!',
'User: How are you?',
'AI: Im good, thank you!'
]);
const addMessage = (message: string) => {
setMessages((prevMessages) => [...prevMessages, message]);
};
2024-09-19 13:02:56 +02:00
useEffect(() => {
2024-09-19 15:56:26 +02:00
workerRef.current = new Worker(new URL("./ProcessAPI.js", import.meta.url))
workerRef.current.postMessage({})
workerRef.current.onmessage = (e) => {
setAccessToken(e.data)
2024-09-19 13:02:56 +02:00
}
2024-09-19 15:56:26 +02:00
2024-09-19 13:02:56 +02:00
return () => {
2024-09-19 15:56:26 +02:00
if (workerRef.current) {
workerRef.current.terminate()
}
2024-09-19 13:02:56 +02:00
}
2024-09-19 15:56:26 +02:00
},[])
2024-09-19 13:02:56 +02:00
2024-09-19 15:56:26 +02:00
const HandleGetRequest = (message: string, ai_model: string, system_prompt: string) => {
if (workerRef.current) {
workerRef.current.postMessage({ functionName: "getResponse", access_token: accessToken, message: message, ai_model: ai_model, system_prompt: system_prompt })
workerRef.current.onmessage = (e) => {
addMessage("AI: " + e.data)
}
}
2024-09-19 13:02:56 +02:00
}
2024-09-19 15:56:26 +02:00
const HandlePostRequest = (message: string, ai_model: string, system_prompt: string) => {
if (workerRef.current) {
workerRef.current.postMessage({ functionName: "postRequest", access_token: accessToken, message: message, ai_model: ai_model, system_prompt: system_prompt })
workerRef.current.onmessage = (e) => {
HandleGetRequest(message,ai_model,system_prompt)
}
}
2024-09-19 13:02:56 +02:00
}
2024-09-19 15:56:26 +02:00
return (
<div>
<ConversationFrontend
messages={messages}
onResendClick={handleResendClick}
onEditClick={handleEditClick}
onCopyClick={handleCopyClick}
/>
<InputFrontend
message=""
onSendClick={handleSendClick}
onMicClick={handleMicClick}
/>
</div>
);
2024-09-19 13:02:56 +02:00
}
export default InputOutputBackend