tcsenpai/agenticSeek
1
0
Fork 0
mirror of https://github.com/tcsenpai/agenticSeek.git synced 2025-06-06 19:15:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:Fosowl/agenticSeek

Browse Source
This commit is contained in:
martin legrand 2025-02-23 12:41:50 +01:00
commit e552e7cd04
4 changed files with 49 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -10,3 +10,5 @@ ollama==0.4.7
scipy==1.15.1 scipy==1.15.1
kokoro==0.7.12 kokoro==0.7.12
flask==3.1.0 flask==3.1.0
soundfile==0.13.1
termcolor

11
sources/text_to_speech.py
View File

@ -3,6 +3,8 @@ from IPython.display import display, Audio
import soundfile as sf import soundfile as sf
import subprocess import subprocess
import re import re
import platform
class Speech(): class Speech():
def __init__(self, language = "english") -> None: def __init__(self, language = "english") -> None:
@ -26,10 +28,15 @@ class Speech():
speed=1, split_pattern=r'\n+' speed=1, split_pattern=r'\n+'
) )
for i, (gs, ps, audio) in enumerate(generator): for i, (gs, ps, audio) in enumerate(generator):
audio_file = f'sample.wav' audio_file = 'sample.wav'
print(audio_file)
display(Audio(data=audio, rate=24000, autoplay=i==0)) display(Audio(data=audio, rate=24000, autoplay=i==0))
sf.write(audio_file, audio, 24000) # save each audio file sf.write(audio_file, audio, 24000) # save each audio file
subprocess.call(["afplay", audio_file]) if platform.system().lower() != "windows":
subprocess.call(["afplay", audio_file])
else:
import winsound
winsound.PlaySound(audio_file, winsound.SND_FILENAME)
def clean_sentence(self, sentence): def clean_sentence(self, sentence):
sentence = re.sub(r'`.*?`', '', sentence) sentence = re.sub(r'`.*?`', '', sentence)

43
sources/utility.py
View File

@ -1,20 +1,37 @@
from colorama import Fore from colorama import Fore
from termcolor import colored
import platform
def pretty_print(text, color = "info"): def pretty_print(text, color = "info"):
""" """
print text with color print text with color
""" """
color_map = { if platform.system().lower() != "windows":
"success": Fore.GREEN, color_map = {
"failure": Fore.RED, "success": Fore.GREEN,
"status": Fore.LIGHTGREEN_EX, "failure": Fore.RED,
"code": Fore.LIGHTBLUE_EX, "status": Fore.LIGHTGREEN_EX,
"warning": Fore.YELLOW, "code": Fore.LIGHTBLUE_EX,
"output": Fore.LIGHTCYAN_EX, "warning": Fore.YELLOW,
} "output": Fore.LIGHTCYAN_EX,
if color not in color_map: }
print(text) if color not in color_map:
pretty_print("Invalid color in pretty_print", "warning") print(text)
return pretty_print("Invalid color in pretty_print", "warning")
print(color_map[color], text, Fore.RESET) return
print(color_map[color], text, Fore.RESET)
else:
color_map = {
"success": "green",
"failure": "red",
"status": "light_green",
"code": "light_blue",
"warning": "yello",
"output": "cyan",
"default": "black"
}
if color not in color_map:
color = "default"
print(colored(text, color_map[color]))