Spaces:
Runtime error
Runtime error
File size: 3,494 Bytes
5f5d58c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
"""
Contains the functions that run when `gradio` is called from the command line. Specifically, allows
$ gradio app.py, to run app.py in reload mode where any changes in the app.py file or Gradio library reloads the demo.
$ gradio app.py my_demo, to use variable names other than "demo"
"""
from __future__ import annotations
import inspect
import os
import re
import subprocess
import sys
import threading
from pathlib import Path
from typing import List, Optional
import typer
from rich import print
import gradio
from gradio import utils
reload_thread = threading.local()
def _setup_config(
demo_path: Path,
demo_name: str = "demo",
additional_watch_dirs: list[str] | None = None,
encoding: str = "utf-8",
):
original_path = Path(demo_path)
app_text = original_path.read_text(encoding=encoding)
patterns = [
f"with gr\\.Blocks\\(.*\\) as {demo_name}",
f"{demo_name} = gr\\.Blocks",
f"{demo_name} = gr\\.Interface",
f"{demo_name} = gr\\.ChatInterface",
f"{demo_name} = gr\\.TabbedInterface",
]
if not any(re.search(p, app_text, flags=re.DOTALL) for p in patterns):
print(
f"\n[bold red]Warning[/]: Cannot statically find a gradio demo called {demo_name}. "
"Reload work may fail."
)
abs_original_path = utils.abspath(original_path)
if original_path.is_absolute():
relpath = original_path.relative_to(Path.cwd())
else:
relpath = original_path
module_name = str(relpath.parent / relpath.stem).replace(os.path.sep, ".")
gradio_folder = Path(inspect.getfile(gradio)).parent
message = "Watching:"
message_change_count = 0
watching_dirs = []
if str(gradio_folder).strip():
watching_dirs.append(gradio_folder)
message += f" '{gradio_folder}'"
message_change_count += 1
abs_parent = abs_original_path.parent
if str(abs_parent).strip():
watching_dirs.append(abs_parent)
if message_change_count == 1:
message += ","
message += f" '{abs_parent}'"
abs_current = Path.cwd().absolute()
if str(abs_current).strip():
watching_dirs.append(abs_current)
if message_change_count == 1:
message += ","
message += f" '{abs_current}'"
for wd in additional_watch_dirs or []:
if Path(wd) not in watching_dirs:
watching_dirs.append(wd)
if message_change_count == 1:
message += ","
message += f" '{wd}'"
print(message + "\n")
# guaranty access to the module of an app
sys.path.insert(0, os.getcwd())
return module_name, abs_original_path, [str(s) for s in watching_dirs], demo_name
def main(
demo_path: Path,
demo_name: str = "demo",
watch_dirs: Optional[List[str]] = None,
encoding: str = "utf-8",
):
# default execution pattern to start the server and watch changes
module_name, path, watch_dirs, demo_name = _setup_config(
demo_path, demo_name, watch_dirs, encoding
)
# extra_args = args[1:] if len(args) == 1 or args[1].startswith("--") else args[2:]
popen = subprocess.Popen(
[sys.executable, "-u", path],
env=dict(
os.environ,
GRADIO_WATCH_DIRS=",".join(watch_dirs),
GRADIO_WATCH_MODULE_NAME=module_name,
GRADIO_WATCH_DEMO_NAME=demo_name,
),
)
popen.wait()
if __name__ == "__main__":
typer.run(main)
|