mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-12 20:49:04 +08:00
### What problem does this PR solve? feat: Supports pronunciation while outputting text #2088 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
parent
87a998e9e5
commit
07de36ec86
@ -22,12 +22,14 @@ interface IProps {
|
||||
content: string;
|
||||
prompt?: string;
|
||||
showLikeButton: boolean;
|
||||
audioBinary?: string;
|
||||
}
|
||||
|
||||
export const AssistantGroupButton = ({
|
||||
messageId,
|
||||
content,
|
||||
prompt,
|
||||
audioBinary,
|
||||
showLikeButton,
|
||||
}: IProps) => {
|
||||
const { visible, hideModal, showModal, onFeedbackOk, loading } =
|
||||
@ -38,7 +40,7 @@ export const AssistantGroupButton = ({
|
||||
showModal: showPromptModal,
|
||||
} = useSetModalState();
|
||||
const { t } = useTranslation();
|
||||
const { handleRead, ref, isPlaying } = useSpeech(content);
|
||||
const { handleRead, ref, isPlaying } = useSpeech(content, audioBinary);
|
||||
|
||||
const handleLike = useCallback(() => {
|
||||
onFeedbackOk({ thumbup: true });
|
||||
|
@ -52,7 +52,7 @@ export const useRemoveMessage = (
|
||||
return { onRemoveMessage, loading };
|
||||
};
|
||||
|
||||
export const useSpeech = (content: string) => {
|
||||
export const useSpeech = (content: string, audioBinary?: string) => {
|
||||
const ref = useRef<HTMLAudioElement>(null);
|
||||
const { read } = useSpeechWithSse();
|
||||
const player = useRef<SpeechPlayer>();
|
||||
@ -94,6 +94,15 @@ export const useSpeech = (content: string) => {
|
||||
}
|
||||
}, [setIsPlaying, speech, isPlaying, pause]);
|
||||
|
||||
// useEffect(() => {
|
||||
// if (audioBinary) {
|
||||
// const units = hexStringToUint8Array(audioBinary);
|
||||
// if (units) {
|
||||
// player.current?.feed(units);
|
||||
// }
|
||||
// }
|
||||
// }, [audioBinary]);
|
||||
|
||||
useEffect(() => {
|
||||
initialize();
|
||||
}, [initialize]);
|
||||
|
@ -131,6 +131,7 @@ const MessageItem = ({
|
||||
content={item.content}
|
||||
prompt={item.prompt}
|
||||
showLikeButton={showLikeButton}
|
||||
audioBinary={item.audio_binary}
|
||||
></AssistantGroupButton>
|
||||
)
|
||||
) : (
|
||||
|
@ -207,13 +207,17 @@ export const useUploadFile = () => {
|
||||
formData.append('file', file);
|
||||
formData.append('path', pathList[index]);
|
||||
});
|
||||
const { data } = await fileManagerService.uploadFile(formData);
|
||||
if (data.retcode === 0) {
|
||||
message.success(t('message.uploaded'));
|
||||
setPaginationParams(1);
|
||||
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
|
||||
try {
|
||||
const { data } = await fileManagerService.uploadFile(formData);
|
||||
if (data.retcode === 0) {
|
||||
message.success(t('message.uploaded'));
|
||||
setPaginationParams(1);
|
||||
queryClient.invalidateQueries({ queryKey: ['fetchFileList'] });
|
||||
}
|
||||
return data.retcode;
|
||||
} catch (error) {
|
||||
console.log('🚀 ~ useUploadFile ~ error:', error);
|
||||
}
|
||||
return data.retcode;
|
||||
},
|
||||
});
|
||||
|
||||
|
@ -433,6 +433,7 @@ export const useSelectDerivedMessages = () => {
|
||||
role: MessageType.Assistant,
|
||||
}),
|
||||
prompt: answer.prompt,
|
||||
audio_binary: answer.audio_binary,
|
||||
},
|
||||
];
|
||||
});
|
||||
|
@ -70,6 +70,7 @@ export interface Message {
|
||||
doc_ids?: string[];
|
||||
prompt?: string;
|
||||
id?: string;
|
||||
audio_binary?: string;
|
||||
}
|
||||
|
||||
export interface IReference {
|
||||
@ -84,6 +85,7 @@ export interface IAnswer {
|
||||
conversationId?: string;
|
||||
prompt?: string;
|
||||
id?: string;
|
||||
audio_binary?: string;
|
||||
}
|
||||
|
||||
export interface Docagg {
|
||||
|
@ -72,3 +72,44 @@ export const toFixed = (value: unknown, fixed = 2) => {
|
||||
}
|
||||
return value;
|
||||
};
|
||||
|
||||
export const stringToUint8Array = (str: string) => {
|
||||
// const byteString = str.replace(/b'|'/g, '');
|
||||
const byteString = str.slice(2, -1);
|
||||
|
||||
const uint8Array = new Uint8Array(byteString.length);
|
||||
for (let i = 0; i < byteString.length; i++) {
|
||||
uint8Array[i] = byteString.charCodeAt(i);
|
||||
}
|
||||
|
||||
return uint8Array;
|
||||
};
|
||||
|
||||
export const hexStringToUint8Array = (hex: string) => {
|
||||
const arr = hex.match(/[\da-f]{2}/gi);
|
||||
if (Array.isArray(arr)) {
|
||||
return new Uint8Array(
|
||||
arr.map(function (h) {
|
||||
return parseInt(h, 16);
|
||||
}),
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
export function hexToArrayBuffer(input: string) {
|
||||
if (typeof input !== 'string') {
|
||||
throw new TypeError('Expected input to be a string');
|
||||
}
|
||||
|
||||
if (input.length % 2 !== 0) {
|
||||
throw new RangeError('Expected string to be an even number of characters');
|
||||
}
|
||||
|
||||
const view = new Uint8Array(input.length / 2);
|
||||
|
||||
for (let i = 0; i < input.length; i += 2) {
|
||||
view[i / 2] = parseInt(input.substring(i, i + 2), 16);
|
||||
}
|
||||
|
||||
return view.buffer;
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user