ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-03-01 04:06:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

addded some texts

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-05-19 13:13:20 +02:00
parent 5e0f38da1b
commit d81f5bf28e
2 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/app.py
View File

@ -71,7 +71,7 @@ class LollmsApplication(LoLLMsCom):
self.generate_msg_with_internet: Callable[[str, Dict], None] = None self.generate_msg_with_internet: Callable[[str, Dict], None] = None
self.handle_continue_generate_msg_from: Callable[[str, Dict], None] = None self.handle_continue_generate_msg_from: Callable[[str, Dict], None] = None
self.rt_com = None
if not free_mode: if not free_mode:
try: try:
if config.auto_update: if config.auto_update:

20
lollms/media.py
View File

@ -94,14 +94,27 @@ from datetime import datetime
import math import math
class AudioRecorder: class RTCom:
def __init__( def __init__(
self, self,
lc:LollmsApplication, lc:LollmsApplication,
sio:socketio.Client, sio:socketio.Client,
personality:AIPersonality, personality:AIPersonality,
client:Client, client:Client,
threshold=1000, silence_duration=2, sound_threshold_percentage=10, gain=1.0, rate=44100, channels=1, buffer_size=10, model="small.en", snd_device=None, logs_folder="logs", voice=None, block_while_talking=True, context_size=4096): threshold=1000,
silence_duration=2,
sound_threshold_percentage=10,
gain=1.0,
rate=44100,
channels=1,
buffer_size=10,
model="small.en",
snd_device=None,
logs_folder="logs",
voice=None,
block_while_talking=True,
context_size=4096
):
self.sio = sio self.sio = sio
self.lc = lc self.lc = lc
self.client = client self.client = client
@ -353,6 +366,7 @@ class AudioRecorder:
user_description = "\n!@>user information:" + self.lc.config.user_description if self.lc.config.use_user_informations_in_discussion else "" user_description = "\n!@>user information:" + self.lc.config.user_description if self.lc.config.use_user_informations_in_discussion else ""
# TODO: send signal # TODO: send signal
# self.transcription_signal.update_status.emit("Transcribing") # self.transcription_signal.update_status.emit("Transcribing")
self.lc.info("Transcribing")
ASCIIColors.green("<<TRANSCRIBING>>") ASCIIColors.green("<<TRANSCRIBING>>")
result = self.whisper.transcribe(str(Path(self.logs_folder)/filename)) result = self.whisper.transcribe(str(Path(self.logs_folder)/filename))
transcription_fn = str(Path(self.logs_folder)/filename) + ".txt" transcription_fn = str(Path(self.logs_folder)/filename) + ".txt"
@ -369,6 +383,7 @@ class AudioRecorder:
# TODO : send the output # TODO : send the output
# self.transcription_signal.update_status.emit("Generating answer") # self.transcription_signal.update_status.emit("Generating answer")
ASCIIColors.green("<<RESPONDING>>") ASCIIColors.green("<<RESPONDING>>")
self.lc.info("Responding")
self.lc.handle_generate_msg(self.client.client_id, {"prompt": current_prompt}) self.lc.handle_generate_msg(self.client.client_id, {"prompt": current_prompt})
while self.lc.busy: while self.lc.busy:
time.sleep(0.01) time.sleep(0.01)
@ -383,6 +398,7 @@ class AudioRecorder:
trace_exception(ex) trace_exception(ex)
self.block_listening = False self.block_listening = False
ASCIIColors.green("<<LISTENING>>") ASCIIColors.green("<<LISTENING>>")
self.lc.info("Listening")
# TODO : send the output # TODO : send the output
#self.transcription_signal.update_status.emit("Listening") #self.transcription_signal.update_status.emit("Listening")