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

fixed error

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-05-20 13:22:01 +02:00
parent cb090d32fd
commit b5b92865b6
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lollms/media.py
View File

@ -354,18 +354,18 @@ class RTCom:
# self.transcription_signal.update_status.emit("Transcribing")
self.lc.info("Transcribing")
ASCIIColors.green("<<TRANSCRIBING>>")
result = self.lc.stt.transcribe(str(Path(self.logs_folder)/filename))
transcription = self.lc.stt.transcribe(str(Path(self.logs_folder)/filename))
transcription_fn = str(Path(self.logs_folder)/filename) + ".txt"
with open(transcription_fn, "w", encoding="utf-8") as f:
f.write(result["text"])
f.write(transcription)
with self.transcribed_lock:
self.transcribed_files.append((filename, result["text"]))
self.transcribed_files.append((filename, transcription))
self.transcribed_lock.notify()
if result["text"]!="":
current_prompt = result["text"]
if transcription!="":
current_prompt = transcription
# TODO : send the output
# self.transcription_signal.new_user_transcription.emit(filename, result["text"])
# self.transcription_signal.new_user_transcription.emit(filename, transcription)
# TODO : send the output
# self.transcription_signal.update_status.emit("Generating answer")
self.lc.new_block(client_id=self.client.client_id,sender=self.lc.config.user_name, content=current_prompt)
@ -567,7 +567,7 @@ class RealTimeTranscription:
# If the result is not empty, call the callback
if result:
self.callback(result["text"])
self.callback(transcription)
except KeyboardInterrupt:
# If the user hits Ctrl+C, stop the stream
self.stop()

4
lollms/services/whisper/lollms_whisper.py
View File

@ -37,6 +37,6 @@ class LollmsWhisper(LollmsSTT):
def transcribe(
self,
wave_path: str|Path
):
)->str:
result = self.whisper.transcribe(str(wave_path))
return result
return result["text"]

4
lollms/stt.py
View File

@ -44,8 +44,8 @@ class LollmsSTT:
def transcribe(
self,
wav_path: str | Path,
prompt=""
):
prompt:str=""
)->str:
"""
Transcribes the given audio file to text.