mirror of
https://github.com/ParisNeo/lollms-webui.git
synced 2024-12-18 12:16:22 +00:00
47 lines
1.6 KiB
Python
47 lines
1.6 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 do advanced stuff like executing code.
|
|
|
|
"""
|
|
from fastapi import APIRouter, Request, HTTPException
|
|
from fastapi.responses import FileResponse
|
|
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, show_yes_no_dialog, add_period
|
|
from lollms.security import sanitize_path, forbid_remote_access, check_access, sanitize_svg, sanitize_path_from_endpoint
|
|
from ascii_colors import ASCIIColors
|
|
from lollms.databases.discussions_database import DiscussionsDB
|
|
from lollms.client_session import Client
|
|
from pathlib import Path
|
|
import tqdm
|
|
from fastapi import FastAPI, UploadFile, File
|
|
import shutil
|
|
import os
|
|
import platform
|
|
import string
|
|
import re
|
|
import subprocess
|
|
from typing import Optional
|
|
|
|
def validate_file_path(path):
|
|
try:
|
|
sanitized_path = sanitize_path(path, allow_absolute_path=False)
|
|
return sanitized_path is not None
|
|
except Exception as e:
|
|
print(f"Path validation error: {str(e)}")
|
|
return False
|
|
|
|
|
|
|
|
# ----------------------- Defining router and main class ------------------------------
|
|
|
|
router = APIRouter()
|
|
lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
|