mirror of
https://github.com/HumanAIGC-Engineering/gradio-webrtc.git
synced 2026-02-05 18:09:23 +08:00
[feat] update some feature
sync code of fastrtc, add text support through datachannel, fix safari connect problem support chat without camera or mic
This commit is contained in:
@@ -0,0 +1,123 @@
|
||||
"use client"
|
||||
|
||||
import { useState, useEffect, useRef, useCallback } from "react";
|
||||
import { BackgroundCircles } from "@/components/ui/background-circles";
|
||||
import { AIVoiceInput } from "@/components/ui/ai-voice-input";
|
||||
import { WebRTCClient } from "@/lib/webrtc-client";
|
||||
|
||||
export function BackgroundCircleProvider() {
|
||||
const [currentVariant, setCurrentVariant] =
|
||||
useState<keyof typeof COLOR_VARIANTS>("octonary");
|
||||
const [isConnected, setIsConnected] = useState(false);
|
||||
const [webrtcClient, setWebrtcClient] = useState<WebRTCClient | null>(null);
|
||||
const [audioLevel, setAudioLevel] = useState(0);
|
||||
const audioRef = useRef<HTMLAudioElement>(null);
|
||||
|
||||
// Memoize callbacks to prevent recreation on each render
|
||||
const handleConnected = useCallback(() => setIsConnected(true), []);
|
||||
const handleDisconnected = useCallback(() => setIsConnected(false), []);
|
||||
|
||||
const handleAudioStream = useCallback((stream: MediaStream) => {
|
||||
if (audioRef.current) {
|
||||
audioRef.current.srcObject = stream;
|
||||
}
|
||||
}, []);
|
||||
|
||||
const handleAudioLevel = useCallback((level: number) => {
|
||||
// Apply some smoothing to the audio level
|
||||
setAudioLevel(prev => prev * 0.7 + level * 0.3);
|
||||
}, []);
|
||||
|
||||
// Get all available variants
|
||||
const variants = Object.keys(
|
||||
COLOR_VARIANTS
|
||||
) as (keyof typeof COLOR_VARIANTS)[];
|
||||
|
||||
// Function to change to the next color variant
|
||||
const changeVariant = () => {
|
||||
const currentIndex = variants.indexOf(currentVariant);
|
||||
const nextVariant = variants[(currentIndex + 1) % variants.length];
|
||||
setCurrentVariant(nextVariant);
|
||||
};
|
||||
|
||||
useEffect(() => {
|
||||
// Initialize WebRTC client with memoized callbacks
|
||||
const client = new WebRTCClient({
|
||||
onConnected: handleConnected,
|
||||
onDisconnected: handleDisconnected,
|
||||
onAudioStream: handleAudioStream,
|
||||
onAudioLevel: handleAudioLevel
|
||||
});
|
||||
setWebrtcClient(client);
|
||||
|
||||
return () => {
|
||||
client.disconnect();
|
||||
};
|
||||
}, [handleConnected, handleDisconnected, handleAudioStream, handleAudioLevel]);
|
||||
|
||||
const handleStart = () => {
|
||||
webrtcClient?.connect();
|
||||
};
|
||||
|
||||
const handleStop = () => {
|
||||
webrtcClient?.disconnect();
|
||||
};
|
||||
|
||||
return (
|
||||
<div
|
||||
className="relative w-full h-full"
|
||||
onClick={changeVariant} // Add click handler to change color
|
||||
>
|
||||
<BackgroundCircles
|
||||
variant={currentVariant}
|
||||
audioLevel={audioLevel}
|
||||
isActive={isConnected}
|
||||
/>
|
||||
<div className="absolute inset-0 flex items-center justify-center">
|
||||
<AIVoiceInput
|
||||
onStart={handleStart}
|
||||
onStop={handleStop}
|
||||
isConnected={isConnected}
|
||||
/>
|
||||
</div>
|
||||
<audio ref={audioRef} autoPlay hidden />
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default { BackgroundCircleProvider }
|
||||
|
||||
const COLOR_VARIANTS = {
|
||||
primary: {
|
||||
border: [
|
||||
"border-emerald-500/60",
|
||||
"border-cyan-400/50",
|
||||
"border-slate-600/30",
|
||||
],
|
||||
gradient: "from-emerald-500/30",
|
||||
},
|
||||
secondary: {
|
||||
border: [
|
||||
"border-violet-500/60",
|
||||
"border-fuchsia-400/50",
|
||||
"border-slate-600/30",
|
||||
],
|
||||
gradient: "from-violet-500/30",
|
||||
},
|
||||
senary: {
|
||||
border: [
|
||||
"border-blue-500/60",
|
||||
"border-sky-400/50",
|
||||
"border-slate-600/30",
|
||||
],
|
||||
gradient: "from-blue-500/30",
|
||||
}, // blue
|
||||
octonary: {
|
||||
border: [
|
||||
"border-red-500/60",
|
||||
"border-rose-400/50",
|
||||
"border-slate-600/30",
|
||||
],
|
||||
gradient: "from-red-500/30",
|
||||
},
|
||||
} as const;
|
||||
Reference in New Issue
Block a user