mirror of
https://github.com/ParisNeo/lollms-webui.git
synced 2024-12-30 09:08:51 +00:00
138 lines
5.1 KiB
Python
138 lines
5.1 KiB
Python
"""
|
|
project: lollms_webui
|
|
file: lollms_xtts.py
|
|
author: ParisNeo
|
|
description:
|
|
This module contains a set of FastAPI routes that provide information about the Lord of Large Language and Multimodal Systems (LoLLMs) Web UI
|
|
application. These routes allow users to
|
|
|
|
"""
|
|
from fastapi import APIRouter, Request
|
|
from fastapi import HTTPException
|
|
from lollms_webui import LOLLMSWebUI
|
|
from pydantic import BaseModel, Field
|
|
from starlette.responses import StreamingResponse
|
|
from lollms.types import MSG_OPERATION_TYPE
|
|
from lollms.main_config import BaseConfig
|
|
from lollms.utilities import detect_antiprompt, remove_text_from_string, trace_exception, find_first_available_file_index, add_period, PackageManager
|
|
from lollms.security import sanitize_path_from_endpoint, validate_path, forbid_remote_access, check_access
|
|
from pathlib import Path
|
|
from ascii_colors import ASCIIColors
|
|
import os
|
|
import platform
|
|
import yaml, json
|
|
# ----------------------- Defining router and main class ------------------------------
|
|
|
|
router = APIRouter()
|
|
lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
|
|
|
|
|
|
# ----------------------- voice ------------------------------
|
|
@router.get("/get_presets")
|
|
def get_presets():
|
|
presets = []
|
|
presets_folder = Path("__file__").parent/"presets"
|
|
for filename in presets_folder.glob('*.yaml'):
|
|
with open(filename, 'r', encoding='utf-8') as file:
|
|
try:
|
|
print(filename)
|
|
preset = yaml.safe_load(file)
|
|
if preset is not None:
|
|
presets.append(preset)
|
|
except Exception as ex:
|
|
trace_exception(ex)
|
|
presets_folder = lollmsElfServer.lollms_paths.personal_discussions_path/"lollms_playground_presets"
|
|
presets_folder.mkdir(exist_ok=True, parents=True)
|
|
for filename in presets_folder.glob('*.yaml'):
|
|
with open(filename, 'r', encoding='utf-8') as file:
|
|
preset = yaml.safe_load(file)
|
|
if preset is not None:
|
|
presets.append(preset)
|
|
return presets
|
|
|
|
class PresetData(BaseModel):
|
|
client_id: str
|
|
name: str = Field(..., min_length=1)
|
|
|
|
@router.post("/add_preset")
|
|
async def add_preset(preset_data: PresetData):
|
|
"""
|
|
Changes current voice
|
|
|
|
:param request: The HTTP request object.
|
|
:return: A JSON response with the status of the operation.
|
|
"""
|
|
forbid_remote_access(lollmsElfServer)
|
|
check_access(lollmsElfServer, preset_data.client_id)
|
|
try:
|
|
|
|
presets_folder = lollmsElfServer.lollms_paths.personal_discussions_path/"lollms_playground_presets"
|
|
if not presets_folder.exists():
|
|
presets_folder.mkdir(exist_ok=True, parents=True)
|
|
|
|
# Ensure the name doesn't contain any path manipulation characters
|
|
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
|
|
|
|
fn = preset_data.name.lower().replace(" ","_")
|
|
filename = presets_folder/f"{fn}.yaml"
|
|
with open(filename, 'w', encoding='utf-8') as file:
|
|
yaml.dump(preset_data, file)
|
|
return {"status": True}
|
|
except Exception as ex:
|
|
trace_exception(ex) # Assuming 'trace_exception' function logs the error
|
|
return {"status": False, "error": "There was an error adding the preset"}
|
|
|
|
@router.post("/del_preset")
|
|
async def del_preset(preset_data: PresetData):
|
|
"""
|
|
Saves a preset to a file.
|
|
|
|
:param preset_data: The data of the preset.
|
|
:return: A JSON response with the status of the operation.
|
|
"""
|
|
forbid_remote_access(lollmsElfServer)
|
|
check_access(lollmsElfServer, preset_data.client_id)
|
|
# Get the JSON data from the POST request.
|
|
if preset_data.name is None:
|
|
raise HTTPException(status_code=400, detail="Preset name is missing in the request")
|
|
|
|
# Ensure the name doesn't contain any path manipulation characters
|
|
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
|
|
|
|
presets_file = lollmsElfServer.lollms_paths.personal_discussions_path/"lollms_playground_presets"/preset_data.name
|
|
try:
|
|
presets_file.unlink()
|
|
return {"status":True}
|
|
except:
|
|
return {"status":False}
|
|
|
|
|
|
class PresetDataWithValue(BaseModel):
|
|
client_id: str
|
|
name: str = Field(..., min_length=1)
|
|
preset: str
|
|
|
|
@router.post("/save_presets")
|
|
async def save_presets(preset_data: PresetDataWithValue):
|
|
"""
|
|
Saves a preset to a file.
|
|
|
|
:param preset_data: The data of the preset.
|
|
:return: A JSON response with the status of the operation.
|
|
"""
|
|
forbid_remote_access(lollmsElfServer)
|
|
check_access(lollmsElfServer, preset_data.client_id)
|
|
# Get the JSON data from the POST request.
|
|
if preset_data.preset is None:
|
|
raise HTTPException(status_code=400, detail="Preset data is missing in the request")
|
|
|
|
|
|
# Ensure the name doesn't contain any path manipulation characters
|
|
sanitize_path_from_endpoint(preset_data.name,exception_text="Invalid preset name")
|
|
|
|
presets_file = lollmsElfServer.lollms_paths.personal_discussions_path/"presets.json"
|
|
# Save the JSON data to a file.
|
|
with open(presets_file, "w") as f:
|
|
json.dump(preset_data.preset, f, indent=4)
|
|
|
|
return {"status":True,"message":"Preset saved successfully!"} |