main #22
7 changed files with 167 additions and 7 deletions
|
@ -1,13 +1,15 @@
|
||||||
// AI.tsx
|
// AI.tsx
|
||||||
import React from 'react';
|
import React from 'react';
|
||||||
import InputBackend from './InputBackend';
|
import InputOutputBackend from './InputOutputHandler';
|
||||||
|
|
||||||
const AI: React.FC = () => {
|
const AI: React.FC = () => {
|
||||||
return (
|
return (
|
||||||
<div>
|
<div>
|
||||||
<InputBackend />
|
<div className="ai-container">
|
||||||
|
<InputOutputBackend />
|
||||||
|
</div>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
};
|
};
|
||||||
|
|
||||||
export default AI;
|
export default AI;
|
|
@ -23,7 +23,7 @@ const ConversationFrontend = React.forwardRef<HTMLDivElement, ConversationProps>
|
||||||
<div className="conversation resize" id="conversation" ref={ref}>
|
<div className="conversation resize" id="conversation" ref={ref}>
|
||||||
{messages.map((message, index) => {
|
{messages.map((message, index) => {
|
||||||
const isUserMessage = message.startsWith('User:');
|
const isUserMessage = message.startsWith('User:');
|
||||||
console.log("output: "+messages)
|
console.log(messages)
|
||||||
return (
|
return (
|
||||||
<div
|
<div
|
||||||
key={index}
|
key={index}
|
||||||
|
|
|
@ -40,7 +40,6 @@ const InputBackend: React.FC = () => {
|
||||||
if (chatResponse && chatResponse.choices && chatResponse.choices.length > 0) {
|
if (chatResponse && chatResponse.choices && chatResponse.choices.length > 0) {
|
||||||
if (chatResponse.choices[0].message.content) {
|
if (chatResponse.choices[0].message.content) {
|
||||||
addMessage('AI: ' + chatResponse.choices[0].message.content);
|
addMessage('AI: ' + chatResponse.choices[0].message.content);
|
||||||
console.log(messages)
|
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
console.error('Error: Unexpected API response:', chatResponse);
|
console.error('Error: Unexpected API response:', chatResponse);
|
||||||
|
|
109
app/InputOutputHandler.tsx
Normal file
109
app/InputOutputHandler.tsx
Normal file
|
@ -0,0 +1,109 @@
|
||||||
|
"use client"
|
||||||
|
import React, { useEffect, useRef, useState } from "react";
|
||||||
|
import ConversationFrontend from "./ConversationFrontend";
|
||||||
|
import InputFrontend from "./InputFrontend";
|
||||||
|
|
||||||
|
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
|
||||||
|
};
|
||||||
|
|
||||||
|
const InputOutputBackend: React.FC = () => {
|
||||||
|
const [accessToken, setAccessToken] = useState("")
|
||||||
|
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: I’m good, thank you!'
|
||||||
|
]);
|
||||||
|
|
||||||
|
const addMessage = (message: string) => {
|
||||||
|
setMessages((prevMessages) => [...prevMessages, message]);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
workerRef.current = new Worker(new URL("./ProcessAPI.js", import.meta.url))
|
||||||
|
workerRef.current.postMessage({})
|
||||||
|
workerRef.current.onmessage = (e) => {
|
||||||
|
setAccessToken(e.data)
|
||||||
|
}
|
||||||
|
|
||||||
|
return () => {
|
||||||
|
if (workerRef.current) {
|
||||||
|
workerRef.current.terminate()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},[])
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div>
|
||||||
|
<ConversationFrontend
|
||||||
|
messages={messages}
|
||||||
|
onResendClick={handleResendClick}
|
||||||
|
onEditClick={handleEditClick}
|
||||||
|
onCopyClick={handleCopyClick}
|
||||||
|
/>
|
||||||
|
<InputFrontend
|
||||||
|
message=""
|
||||||
|
onSendClick={handleSendClick}
|
||||||
|
onMicClick={handleMicClick}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export default InputOutputBackend
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
39
app/ProcessAPI.js
Normal file
39
app/ProcessAPI.js
Normal file
|
@ -0,0 +1,39 @@
|
||||||
|
import axios from 'axios'
|
||||||
|
|
||||||
|
onmessage = function (e) {
|
||||||
|
const { functionName = "getAccess", access_token = "", message = "", ai_model = "phi3.5", system_prompt = "You are a helpful assistant" } = e.data
|
||||||
|
const data = {
|
||||||
|
"ai_model": ai_model,
|
||||||
|
"message": message,
|
||||||
|
"system_prompt": system_prompt,
|
||||||
|
"access_token": access_token
|
||||||
|
};
|
||||||
|
switch (functionName) {
|
||||||
|
case "getAccess":
|
||||||
|
axios.get('https://localhost:5000/interstellar/api/ai_create')
|
||||||
|
.then(Response => {
|
||||||
|
postMessage(Response.data.access_token)
|
||||||
|
}).catch(error => {
|
||||||
|
console.error("Error with GET Token request:", error)
|
||||||
|
})
|
||||||
|
break
|
||||||
|
case "postRequest":
|
||||||
|
axios.post('https://localhost:5000/interstellar/api/ai_send', data)
|
||||||
|
.then(Response => {
|
||||||
|
postMessage(Response.data)
|
||||||
|
}).catch(error => {
|
||||||
|
console.error("Error:", error)
|
||||||
|
})
|
||||||
|
break
|
||||||
|
case "getResponse":
|
||||||
|
axios.get('https://localhost:5000/interstellar/api/ai_get?access_token='+access_token)
|
||||||
|
.then(Response => {
|
||||||
|
postMessage(Response.data.response)
|
||||||
|
}).catch(error => {
|
||||||
|
console.error("Error with GET response request:", error)
|
||||||
|
})
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
|
@ -1,2 +0,0 @@
|
||||||
import React from "react";
|
|
||||||
import axios from 'axios';
|
|
13
app/ProcessMemory.tsx
Normal file
13
app/ProcessMemory.tsx
Normal file
|
@ -0,0 +1,13 @@
|
||||||
|
import React from 'react'
|
||||||
|
|
||||||
|
type Chat = {
|
||||||
|
name:string
|
||||||
|
messages:string[]
|
||||||
|
}
|
||||||
|
|
||||||
|
type History = {
|
||||||
|
chats:Chat[]
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue