lollms-webui/endpoints/lollms_webui_infos.py

"""
project: lollms_webui
file: lollms_webui_infos.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 retrieve details such as the current version number, a list of available databases, and options for 
    updating or restarting the software.

"""

from fastapi import APIRouter, Request
import pkg_resources
from lollms_webui import LOLLMSWebUI
from ascii_colors import ASCIIColors
from lollms.utilities import load_config
from pathlib import Path
from typing import List
import sys
import socketio
# ----------------------- Defining router and main class ------------------------------

# Create an instance of the LoLLMSWebUI class
lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
router = APIRouter()

@router.get("/get_lollms_webui_version")
async def get_lollms_webui_version():
   """Get the version of the LoLLMs Web UI application."""
   # Return the version string
   return lollmsElfServer.version


@router.get("/restart_program")
async def restart_program():
   """Restart the program."""
   # Stop the socketIO server
   lollmsElfServer.sio.shutdown()
   # Sleep for 1 second before rebooting
   lollmsElfServer.sio.sleep(1)
   # Reboot the program
   lollmsElfServer.sio.reboot = True

@router.get("/update_software")
async def update_software():
   """Update the software."""
   # Display an informative message
   ASCIIColors.info("")
   ASCIIColors.info("")
   ASCIIColors.info("")
   ASCIIColors.info("╔══════════════════════════════════════════════════╗")
   ASCIIColors.info("║                Updating backend                  ║")
   ASCIIColors.info("╚══════════════════════════════════════════════════╝")
   ASCIIColors.info("")
   ASCIIColors.info("")
   ASCIIColors.info("")
   # Stop the socketIO server
   lollmsElfServer.sio.shutdown()

   # Run the update script using the provided arguments
   lollmsElfServer.run_update_script(lollmsElfServer.args)
   # Exit the program after successful update
   sys.exit()


@router.get("/check_update")
def check_update():
    """Checks if an update is available"""
    if lollmsElfServer.config.auto_update:
        res = lollmsElfServer.check_update_()
        return {'update_availability':res}
    else:
        return {'update_availability':False}