update app and fix #29

Signed-off-by: Victor Seiji Hariki <victorseijih@gmail.com>
pull/37/head
Victor Seiji Hariki 2023-01-24 19:16:48 -03:00
parent d903006a96
commit 4ebdc9acc0
2 changed files with 13 additions and 3 deletions

2
app

@ -1 +1 @@
Subproject commit 85776d86371cd8d07cf92661499da71dfa4ea53a
Subproject commit f3e869a5ffced67612d25faf26883177e5ae0b80

View File

@ -3,10 +3,14 @@ import gradio as gr
from fastapi import FastAPI
import os
from launch import run
import pathlib
import inspect
import string
import random as rd
extension_dir = pathlib.Path(inspect.getfile(lambda: None)).parent.parent
key_characters = (string.ascii_letters + string.digits)
@ -20,7 +24,7 @@ key = random_string()
def get_files(path):
# Gets all files
directories = set()
for root, _, files in os.walk(path):
for root, _, files in os.walk(path.resolve()):
for file in files:
directories.add(root + '/' + file)
@ -30,7 +34,13 @@ def get_files(path):
def started(demo, app: FastAPI):
try:
# Force allow paths for fixing symlinked extension directory references
force_allow = get_files(f"{os.path.abspath(scripts.basedir())}/app")
force_allow = get_files(extension_dir / "app")
# Add to allowed files list
app.blocks.temp_file_sets.append(force_allow)
# Force allow paths for fixing symlinked extension directory references (base javascript files now)
force_allow = get_files(extension_dir / "javascript")
# Add to allowed files list
app.blocks.temp_file_sets.append(force_allow)