এখানে সবচেয়ে গুরুত্বপূর্ণ উপাদান হল Large Language Model (LLM) ব্যাকএন্ড, যার জন্য আমরা ওল্লামা ব্যবহার করব। অফলাইনে LLM চালানো এবং পরিবেশন করার জন্য একটি জনপ্রিয় হাতিয়ার হিসাবে ব্যাপকভাবে স্বীকৃত। ওল্লামা আপনার কাছে নতুন হলে, আমি অফলাইন RAG-তে আমার আগের নিবন্ধটি চেক করার পরামর্শ দিচ্ছি: মূলত, আপনাকে শুধু ওল্লামা অ্যাপ্লিকেশনটি ডাউনলোড করতে হবে, আপনার পছন্দের মডেলটি টানতে হবে এবং এটি চালাতে হবে।
হুইস্পার, ওল্লামা এবং বার্কের সাথে ভয়েস সহকারীর জন্য সিকোয়েন্স ডায়াগ্রাম।
বাস্তবায়নটি বার্কের উপর ভিত্তি করে একটি TextToSpeechService
তৈরির মাধ্যমে শুরু হয়, পাঠ্য থেকে বক্তৃতা সংশ্লেষণ করার পদ্ধতিগুলি অন্তর্ভুক্ত করে এবং দীর্ঘ টেক্সট ইনপুটগুলিকে নির্বিঘ্নে পরিচালনা করে:
import nltk import torch import warnings import numpy as np from transformers import AutoProcessor, BarkModel warnings.filterwarnings( "ignore", message="torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.", ) class TextToSpeechService: def __init__(self, device: str = "cuda" if torch.cuda.is_available() else "cpu"): """ Initializes the TextToSpeechService class. Args: device (str, optional): The device to be used for the model, either "cuda" if a GPU is available or "cpu". Defaults to "cuda" if available, otherwise "cpu". """ self.device = device self.processor = AutoProcessor.from_pretrained("suno/bark-small") self.model = BarkModel.from_pretrained("suno/bark-small") self.model.to(self.device) def synthesize(self, text: str, voice_preset: str = "v2/en_speaker_1"): """ Synthesizes audio from the given text using the specified voice preset. Args: text (str): The input text to be synthesized. voice_preset (str, optional): The voice preset to be used for the synthesis. Defaults to "v2/en_speaker_1". Returns: tuple: A tuple containing the sample rate and the generated audio array. """ inputs = self.processor(text, voice_preset=voice_preset, return_tensors="pt") inputs = {k: v.to(self.device) for k, v in inputs.items()} with torch.no_grad(): audio_array = self.model.generate(**inputs, pad_token_id=10000) audio_array = audio_array.cpu().numpy().squeeze() sample_rate = self.model.generation_config.sample_rate return sample_rate, audio_array def long_form_synthesize(self, text: str, voice_preset: str = "v2/en_speaker_1"): """ Synthesizes audio from the given long-form text using the specified voice preset. Args: text (str): The input text to be synthesized. voice_preset (str, optional): The voice preset to be used for the synthesis. Defaults to "v2/en_speaker_1". Returns: tuple: A tuple containing the sample rate and the generated audio array. """ pieces = [] sentences = nltk.sent_tokenize(text) silence = np.zeros(int(0.25 * self.model.generation_config.sample_rate)) for sent in sentences: sample_rate, audio_array = self.synthesize(sent, voice_preset) pieces += [audio_array, silence.copy()] return self.model.generation_config.sample_rate, np.concatenate(pieces)
__init__
) : ক্লাসটি একটি ঐচ্ছিক device
প্যারামিটার নেয়, যা মডেলের জন্য ব্যবহার করা ডিভাইসটি নির্দিষ্ট করে (হয় cuda
যদি একটি GPU পাওয়া যায়, অথবা cpu
)। এটি suno/bark-small
প্রাক-প্রশিক্ষিত মডেল থেকে বার্ক মডেল এবং সংশ্লিষ্ট প্রসেসর লোড করে। আপনি মডেল লোডারের জন্য suno/bark
উল্লেখ করে বড় সংস্করণটিও ব্যবহার করতে পারেন।
synthesize
) : এই পদ্ধতিটি একটি text
ইনপুট এবং একটি voice_preset
প্যারামিটার নেয়, যা সংশ্লেষণের জন্য ব্যবহৃত ভয়েসটি নির্দিষ্ট করে। আপনি অন্য voice_preset
মান পরীক্ষা করতে পারেন। এটি ইনপুট টেক্সট এবং ভয়েস প্রিসেট প্রস্তুত করতে processor
ব্যবহার করে এবং তারপর model.generate()
পদ্ধতি ব্যবহার করে অডিও অ্যারে তৈরি করে। জেনারেট করা অডিও অ্যারে একটি NumPy অ্যারেতে রূপান্তরিত হয় এবং অডিও অ্যারের সাথে নমুনা হার ফেরত দেওয়া হয়।
long_form_synthesize
) : এই পদ্ধতিটি দীর্ঘ টেক্সট ইনপুট সংশ্লেষণের জন্য ব্যবহৃত হয়। এটি প্রথমে nltk.sent_tokenize
ফাংশন ব্যবহার করে বাক্যে ইনপুট টেক্সটকে টোকেনাইজ করে। প্রতিটি বাক্যের জন্য, এটি অডিও অ্যারে তৈরি করতে synthesize
পদ্ধতিকে কল করে। এটি তারপর প্রতিটি বাক্যের মধ্যে একটি ছোট নীরবতা (0.25 সেকেন্ড) যোগ করে জেনারেট করা অডিও অ্যারেগুলিকে সংযুক্ত করে।
এখন যেহেতু আমরা TextToSpeechService
সেট আপ করেছি, আমাদের ওল্লামা সার্ভারটিকে বড় ভাষা মডেল (LLM) পরিবেশনের জন্য প্রস্তুত করতে হবে। এটি করার জন্য, আপনাকে এই পদক্ষেপগুলি অনুসরণ করতে হবে:
ollama pull llama2
।
ollama serve
।
base.en
) ব্যবহার করব।
ConversationalChain
ব্যবহার করব যা কথোপকথন প্রবাহ পরিচালনা করার জন্য একটি টেমপ্লেট প্রদান করে। আমরা এটিকে ওল্লামা ব্যাকএন্ডের সাথে Llama-2 ভাষার মডেল ব্যবহার করার জন্য কনফিগার করব। import time import threading import numpy as np import whisper import sounddevice as sd from queue import Queue from rich.console import Console from langchain.memory import ConversationBufferMemory from langchain.chains import ConversationChain from langchain.prompts import PromptTemplate from langchain_community.llms import Ollama from tts import TextToSpeechService console = Console() stt = whisper.load_model("base.en") tts = TextToSpeechService() template = """ You are a helpful and friendly AI assistant. You are polite, respectful, and aim to provide concise responses of less than 20 words. The conversation transcript is as follows: {history} And here is the user's follow-up: {input} Your response: """ PROMPT = PromptTemplate(input_variables=["history", "input"], template=template) chain = ConversationChain( prompt=PROMPT, verbose=False, memory=ConversationBufferMemory(ai_prefix="Assistant:"), llm=Ollama(), )
এখন, প্রয়োজনীয় ফাংশন সংজ্ঞায়িত করা যাক:record_audio
: sounddevice.RawInputStream
ব্যবহার করে ব্যবহারকারীর মাইক্রোফোন থেকে অডিও ডেটা ক্যাপচার করতে এই ফাংশনটি একটি পৃথক থ্রেডে চলে।RawInputStream। যখনই নতুন অডিও ডেটা পাওয়া যায় তখন কলব্যাক ফাংশনটিকে বলা হয় এবং এটি আরও প্রক্রিয়াকরণের জন্য ডেটাটিকে data_queue
রাখে।
transcribe
: এই ফাংশনটি data_queue
থেকে অডিও ডেটাকে টেক্সটে ট্রান্সক্রাইব করতে হুইস্পার ইনস্ট্যান্স ব্যবহার করে।
get_llm_response
: এই ফাংশনটি Llama-2 ভাষার মডেলে বর্তমান কথোপকথনের প্রসঙ্গ ফিড করে (Langchain ConversationalChain
এর মাধ্যমে) এবং জেনারেট করা পাঠ্য প্রতিক্রিয়া পুনরুদ্ধার করে।
play_audio
: এই ফাংশনটি বার্ক টেক্সট-টু-স্পিচ ইঞ্জিন দ্বারা তৈরি অডিও তরঙ্গরূপ নেয় এবং এটি একটি সাউন্ড প্লেব্যাক লাইব্রেরি (যেমন, sounddevice
) ব্যবহার করে ব্যবহারকারীর কাছে ফিরিয়ে দেয়। def record_audio(stop_event, data_queue): """ Captures audio data from the user's microphone and adds it to a queue for further processing. Args: stop_event (threading.Event): An event that, when set, signals the function to stop recording. data_queue (queue.Queue): A queue to which the recorded audio data will be added. Returns: None """ def callback(indata, frames, time, status): if status: console.print(status) data_queue.put(bytes(indata)) with sd.RawInputStream( samplerate=16000, dtype="int16", channels=1, callback=callback ): while not stop_event.is_set(): time.sleep(0.1) def transcribe(audio_np: np.ndarray) -> str: """ Transcribes the given audio data using the Whisper speech recognition model. Args: audio_np (numpy.ndarray): The audio data to be transcribed. Returns: str: The transcribed text. """ result = stt.transcribe(audio_np, fp16=False) # Set fp16=True if using a GPU text = result["text"].strip() return text def get_llm_response(text: str) -> str: """ Generates a response to the given text using the Llama-2 language model. Args: text (str): The input text to be processed. Returns: str: The generated response. """ response = chain.predict(input=text) if response.startswith("Assistant:"): response = response[len("Assistant:") :].strip() return response def play_audio(sample_rate, audio_array): """ Plays the given audio data using the sounddevice library. Args: sample_rate (int): The sample rate of the audio data. audio_array (numpy.ndarray): The audio data to be played. Returns: None """ sd.play(audio_array, sample_rate) sd.wait()
তারপর, আমরা প্রধান অ্যাপ্লিকেশন লুপ সংজ্ঞায়িত করি। প্রধান অ্যাপ্লিকেশন লুপ ব্যবহারকারীকে কথোপকথনমূলক মিথস্ক্রিয়া দ্বারা নির্দেশিত করে:
ব্যবহারকারী একবার এন্টার চাপলে, ব্যবহারকারীর অডিও ইনপুট ক্যাপচার করার জন্য record_audio
ফাংশনটিকে একটি পৃথক থ্রেডে ডাকা হয়।
যখন ব্যবহারকারী রেকর্ডিং বন্ধ করতে আবার এন্টার চাপেন, তখন transcribe
ফাংশন ব্যবহার করে অডিও ডেটা প্রতিলিপি করা হয়।
প্রতিলিপিকৃত পাঠ্যটি তারপর get_llm_response
ফাংশনে প্রেরণ করা হয়, যা Llama-2 ভাষা মডেল ব্যবহার করে একটি প্রতিক্রিয়া তৈরি করে।
উৎপন্ন প্রতিক্রিয়া কনসোলে প্রিন্ট করা হয় এবং play_audio
ফাংশন ব্যবহার করে ব্যবহারকারীর কাছে ফিরে আসে।
if __name__ == "__main__": console.print("[cyan]Assistant started! Press Ctrl+C to exit.") try: while True: console.input( "Press Enter to start recording, then press Enter again to stop." ) data_queue = Queue() # type: ignore[var-annotated] stop_event = threading.Event() recording_thread = threading.Thread( target=record_audio, args=(stop_event, data_queue), ) recording_thread.start() input() stop_event.set() recording_thread.join() audio_data = b"".join(list(data_queue.queue)) audio_np = ( np.frombuffer(audio_data, dtype=np.int16).astype(np.float32) / 32768.0 ) if audio_np.size > 0: with console.status("Transcribing...", spinner="earth"): text = transcribe(audio_np) console.print(f"[yellow]You: {text}") with console.status("Generating response...", spinner="earth"): response = get_llm_response(text) sample_rate, audio_array = tts.long_form_synthesize(response) console.print(f"[cyan]Assistant: {response}") play_audio(sample_rate, audio_array) else: console.print( "[red]No audio recorded. Please ensure your microphone is working." ) except KeyboardInterrupt: console.print("\n[red]Exiting...") console.print("[blue]Session ended.")
এছাড়াও প্রকাশিত