ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-02-06 19:09:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

added the list of avatars in order

Browse Source
This commit is contained in:
saloui 2023-05-22 16:40:39 +02:00
parent 709c0a79fc
commit 19b5b26e1b
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
app.py
View File

@ -258,13 +258,24 @@ class Gpt4AllWebUI(GPT4AllAPI):
scripts_path = personality_folder / 'scripts' scripts_path = personality_folder / 'scripts'
personality_info['has_scripts'] = scripts_path.is_dir() personality_info['has_scripts'] = scripts_path.is_dir()
assets_path = personality_folder / 'assets' assets_path = personality_folder / 'assets'
logo_path = assets_path / 'logo.png'
gif_logo_path = assets_path / 'logo.gif' gif_logo_path = assets_path / 'logo.gif'
personality_info['has_logo'] = logo_path.is_file() or gif_logo_path.is_file() webp_logo_path = assets_path / 'logo.webp'
png_logo_path = assets_path / 'logo.png'
jpg_logo_path = assets_path / 'logo.jpg'
jpeg_logo_path = assets_path / 'logo.jpeg'
bmp_logo_path = assets_path / 'logo.bmp'
personality_info['has_logo'] = png_logo_path.is_file() or gif_logo_path.is_file()
if gif_logo_path.is_file(): if gif_logo_path.is_file():
personality_info['avatar'] = str(gif_logo_path).replace("\\","/") personality_info['avatar'] = str(gif_logo_path).replace("\\","/")
elif logo_path.is_file(): elif webp_logo_path.is_file():
personality_info['avatar'] = str(logo_path).replace("\\","/") personality_info['avatar'] = str(webp_logo_path).replace("\\","/")
elif png_logo_path.is_file():
personality_info['avatar'] = str(png_logo_path).replace("\\","/")
elif jpg_logo_path.is_file():
personality_info['avatar'] = str(jpg_logo_path).replace("\\","/")
elif bmp_logo_path.is_file():
personality_info['avatar'] = str(bmp_logo_path).replace("\\","/")
else: else:
personality_info['avatar'] = "" personality_info['avatar'] = ""
personalities[language_folder.name][category_folder.name].append(personality_info) personalities[language_folder.name][category_folder.name].append(personality_info)