mirror of
https://github.com/ParisNeo/lollms-webui.git
synced 2024-12-24 06:36:37 +00:00
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""
|
|
project: lollms_user
|
|
file: lollms_user.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 manipulate user information.
|
|
|
|
"""
|
|
from fastapi import APIRouter
|
|
from lollms_webui import LOLLMSWebUI
|
|
from pydantic import BaseModel
|
|
from starlette.responses import StreamingResponse
|
|
from lollms.types import MSG_TYPE
|
|
from lollms.main_config import BaseConfig
|
|
from lollms.utilities import detect_antiprompt, remove_text_from_string
|
|
from ascii_colors import ASCIIColors
|
|
from api.db import DiscussionsDB
|
|
from pathlib import Path
|
|
from safe_store.text_vectorizer import TextVectorizer, VectorizationMethod, VisualizationMethod
|
|
import tqdm
|
|
from fastapi import FastAPI, UploadFile, File
|
|
import shutil
|
|
class PersonalPathParameters(BaseModel):
|
|
path:str
|
|
|
|
# ----------------------- Defining router and main class ------------------------------
|
|
|
|
router = APIRouter()
|
|
lollmsElfServer = LOLLMSWebUI.get_instance()
|
|
|
|
@router.get("/switch_personal_path")
|
|
def switch_personal_path(data:PersonalPathParameters):
|
|
path = data.path
|
|
global_paths_cfg = Path("./global_paths_cfg.yaml")
|
|
if global_paths_cfg.exists():
|
|
try:
|
|
cfg = BaseConfig()
|
|
cfg.load_config(global_paths_cfg)
|
|
cfg.lollms_personal_path = path
|
|
cfg.save_config(global_paths_cfg)
|
|
return {"status": True}
|
|
except Exception as ex:
|
|
print(ex)
|
|
return {"status": False, 'error':f"Couldn't switch path: {ex}"}
|
|
|
|
@router.post("/upload_avatar")
|
|
def upload_avatar(avatar: UploadFile = File(...)):
|
|
with open(lollmsElfServer.lollms_paths.personal_user_infos_path/avatar.filename, "wb") as buffer:
|
|
shutil.copyfileobj(avatar.file, buffer)
|
|
return {"status": True,"fileName":avatar.filename}
|