mirror of
https://github.com/ParisNeo/lollms.git
synced 2024-12-24 06:46:40 +00:00
changed names
This commit is contained in:
parent
b671a599e8
commit
be13a3cd41
@ -1,5 +1,5 @@
|
|||||||
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
|
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
|
||||||
version: 98
|
version: 99
|
||||||
binding_name: null
|
binding_name: null
|
||||||
model_name: null
|
model_name: null
|
||||||
model_variant: null
|
model_variant: null
|
||||||
@ -85,7 +85,7 @@ copy_to_clipboard_add_all_details: false
|
|||||||
active_tts_service: "None" # xtts (offline), openai_tts (API key required)
|
active_tts_service: "None" # xtts (offline), openai_tts (API key required)
|
||||||
active_tti_service: "None" # autosd (offline), dall-e (online)
|
active_tti_service: "None" # autosd (offline), dall-e (online)
|
||||||
active_stt_service: "None" # whisper (offline), asr (offline or online), openai_whiosper (API key required)
|
active_stt_service: "None" # whisper (offline), asr (offline or online), openai_whiosper (API key required)
|
||||||
|
active_ttm_service: "None" # musicgen (offline)
|
||||||
# -------------------- Services --------------------------
|
# -------------------- Services --------------------------
|
||||||
|
|
||||||
# ***************** STT *****************
|
# ***************** STT *****************
|
||||||
|
@ -320,7 +320,7 @@ class LollmsApplication(LoLLMsCom):
|
|||||||
self.tts = self.xtts
|
self.tts = self.xtts
|
||||||
|
|
||||||
if self.config.active_stt_service == "openai_whisper":
|
if self.config.active_stt_service == "openai_whisper":
|
||||||
from lollms.services.openai_whisper.lollms_whisper import LollmsOpenAIWhisper
|
from lollms.services.openai_whisper.lollms_openai_whisper import LollmsOpenAIWhisper
|
||||||
self.stt = LollmsOpenAIWhisper(self, self.config.openai_whisper_model, self.config.openai_whisper_key)
|
self.stt = LollmsOpenAIWhisper(self, self.config.openai_whisper_model, self.config.openai_whisper_key)
|
||||||
elif self.config.active_stt_service == "whisper":
|
elif self.config.active_stt_service == "whisper":
|
||||||
from lollms.services.whisper.lollms_whisper import LollmsWhisper
|
from lollms.services.whisper.lollms_whisper import LollmsWhisper
|
||||||
@ -413,7 +413,7 @@ class LollmsApplication(LoLLMsCom):
|
|||||||
self.tts = self.xtts
|
self.tts = self.xtts
|
||||||
|
|
||||||
if self.config.active_stt_service == "openai_whisper":
|
if self.config.active_stt_service == "openai_whisper":
|
||||||
from lollms.services.openai_whisper.lollms_whisper import LollmsOpenAIWhisper
|
from lollms.services.openai_whisper.lollms_openai_whisper import LollmsOpenAIWhisper
|
||||||
self.stt = LollmsOpenAIWhisper(self, self.config.openai_whisper_model, self.config.openai_whisper_key)
|
self.stt = LollmsOpenAIWhisper(self, self.config.openai_whisper_model, self.config.openai_whisper_key)
|
||||||
elif self.config.active_stt_service == "whisper":
|
elif self.config.active_stt_service == "whisper":
|
||||||
from lollms.services.whisper.lollms_whisper import LollmsWhisper
|
from lollms.services.whisper.lollms_whisper import LollmsWhisper
|
||||||
|
@ -1,5 +1,5 @@
|
|||||||
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
|
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
|
||||||
version: 98
|
version: 99
|
||||||
binding_name: null
|
binding_name: null
|
||||||
model_name: null
|
model_name: null
|
||||||
model_variant: null
|
model_variant: null
|
||||||
@ -85,7 +85,7 @@ copy_to_clipboard_add_all_details: false
|
|||||||
active_tts_service: "None" # xtts (offline), openai_tts (API key required)
|
active_tts_service: "None" # xtts (offline), openai_tts (API key required)
|
||||||
active_tti_service: "None" # autosd (offline), dall-e (online)
|
active_tti_service: "None" # autosd (offline), dall-e (online)
|
||||||
active_stt_service: "None" # whisper (offline), asr (offline or online), openai_whiosper (API key required)
|
active_stt_service: "None" # whisper (offline), asr (offline or online), openai_whiosper (API key required)
|
||||||
|
active_ttm_service: "None" # musicgen (offline)
|
||||||
# -------------------- Services --------------------------
|
# -------------------- Services --------------------------
|
||||||
|
|
||||||
# ***************** STT *****************
|
# ***************** STT *****************
|
||||||
|
@ -42,6 +42,18 @@ def list_voices():
|
|||||||
ASCIIColors.yellow("Listing voices")
|
ASCIIColors.yellow("Listing voices")
|
||||||
return {"voices":lollmsElfServer.tts.get_voices()}
|
return {"voices":lollmsElfServer.tts.get_voices()}
|
||||||
|
|
||||||
|
@router.get("/list_stt_models")
|
||||||
|
def list_stt_models():
|
||||||
|
if lollmsElfServer.config.headless_server_mode:
|
||||||
|
return {"status":False,"error":"Code execution is blocked when in headless mode for obvious security reasons!"}
|
||||||
|
|
||||||
|
if lollmsElfServer.config.host!="localhost" and lollmsElfServer.config.host!="127.0.0.1":
|
||||||
|
return {"status":False,"error":"Code execution is blocked when the server is exposed outside for very obvious reasons!"}
|
||||||
|
|
||||||
|
ASCIIColors.yellow("Listing voices")
|
||||||
|
return {"voices":lollmsElfServer.stt.get_models()}
|
||||||
|
|
||||||
|
|
||||||
@router.post("/set_voice")
|
@router.post("/set_voice")
|
||||||
async def set_voice(request: Request):
|
async def set_voice(request: Request):
|
||||||
"""
|
"""
|
||||||
@ -70,7 +82,7 @@ async def set_voice(request: Request):
|
|||||||
|
|
||||||
class LollmsAudio2TextRequest(BaseModel):
|
class LollmsAudio2TextRequest(BaseModel):
|
||||||
wave_file_path: str
|
wave_file_path: str
|
||||||
voice: str = None
|
model: str = None
|
||||||
fn:str = None
|
fn:str = None
|
||||||
|
|
||||||
@router.post("/audio2text")
|
@router.post("/audio2text")
|
@ -26,6 +26,7 @@ from typing import List, Dict, Any
|
|||||||
from ascii_colors import ASCIIColors, trace_exception
|
from ascii_colors import ASCIIColors, trace_exception
|
||||||
from lollms.paths import LollmsPaths
|
from lollms.paths import LollmsPaths
|
||||||
from lollms.utilities import PackageManager, find_next_available_filename
|
from lollms.utilities import PackageManager, find_next_available_filename
|
||||||
|
from lollms.stt import LollmsSTT
|
||||||
import subprocess
|
import subprocess
|
||||||
import shutil
|
import shutil
|
||||||
from tqdm import tqdm
|
from tqdm import tqdm
|
||||||
@ -34,10 +35,7 @@ from io import BytesIO
|
|||||||
from openai import OpenAI
|
from openai import OpenAI
|
||||||
|
|
||||||
|
|
||||||
def get_Whisper(lollms_paths:LollmsPaths):
|
class LollmsOpenAIWhisper(LollmsSTT):
|
||||||
return LollmsOpenAIWhisper
|
|
||||||
|
|
||||||
class LollmsOpenAIWhisper:
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
app:LollmsApplication,
|
app:LollmsApplication,
|
||||||
@ -45,10 +43,8 @@ class LollmsOpenAIWhisper:
|
|||||||
api_key="",
|
api_key="",
|
||||||
output_path=None
|
output_path=None
|
||||||
):
|
):
|
||||||
|
super().__init__(app, model, output_path)
|
||||||
self.client = OpenAI(api_key=api_key)
|
self.client = OpenAI(api_key=api_key)
|
||||||
self.app = app
|
|
||||||
self.model = model
|
|
||||||
self.output_path = output_path
|
|
||||||
self.ready = True
|
self.ready = True
|
||||||
|
|
||||||
def transcribe(
|
def transcribe(
|
@ -30,13 +30,13 @@ class LollmsWhisper(LollmsSTT):
|
|||||||
model="small",
|
model="small",
|
||||||
output_path=None
|
output_path=None
|
||||||
):
|
):
|
||||||
self.app = app
|
super().__init__(app, model, output_path)
|
||||||
self.output_path = output_path
|
|
||||||
self.whisper = whisper.load_model(model)
|
self.whisper = whisper.load_model(model)
|
||||||
|
self.ready = True
|
||||||
|
|
||||||
def transcribe(
|
def transcribe(
|
||||||
self,
|
self,
|
||||||
wav_path: str|Path
|
wave_path: str|Path
|
||||||
):
|
):
|
||||||
result = self.whisper.transcribe(str(wav_path))
|
result = self.whisper.transcribe(str(wave_path))
|
||||||
return result
|
return result
|
||||||
|
@ -24,7 +24,8 @@ class LollmsSTT:
|
|||||||
self,
|
self,
|
||||||
app: LollmsApplication,
|
app: LollmsApplication,
|
||||||
model="",
|
model="",
|
||||||
output_path=None
|
output_path=None,
|
||||||
|
models=[]
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Initializes the LollmsSTT class with the given parameters.
|
Initializes the LollmsSTT class with the given parameters.
|
||||||
@ -38,6 +39,7 @@ class LollmsSTT:
|
|||||||
self.app = app
|
self.app = app
|
||||||
self.output_path = output_path
|
self.output_path = output_path
|
||||||
self.model = model
|
self.model = model
|
||||||
|
self.models = models
|
||||||
|
|
||||||
def transcribe(
|
def transcribe(
|
||||||
self,
|
self,
|
||||||
@ -53,6 +55,9 @@ class LollmsSTT:
|
|||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def get_models(self):
|
||||||
|
return self.models
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def verify(app: LollmsApplication) -> bool:
|
def verify(app: LollmsApplication) -> bool:
|
||||||
"""
|
"""
|
||||||
|
Loading…
Reference in New Issue
Block a user