gradio

1
import argparse
2
import os
3
import pathlib
4
import shutil
5
import textwrap
6

7

8
def copy_all_demos(source_dir: str, dest_dir: str):
9
    demos_to_copy = [
10
        "audio_debugger",
11
        "altair_plot",
12
        "blocks_essay",
13
        "blocks_group",
14
        "blocks_js_methods",
15
        "blocks_layout",
16
        "blocks_multiple_event_triggers",
17
        "blocks_update",
18
        "calculator",
19
        "cancel_events",
20
        "chatbot_multimodal",
21
        "chatinterface_streaming_echo",
22
        "clear_components",
23
        "code",
24
        "fake_gan",
25
        "fake_diffusion_with_gif",
26
        "file_explorer_component_events",
27
        "image_mod_default_image",
28
        "image_segmentation",
29
        "interface_random_slider",
30
        "kitchen_sink",
31
        "kitchen_sink_random",
32
        "matrix_transpose",
33
        "model3D",
34
        "native_plots",
35
        "reverse_audio",
36
        "stt_or_tts",
37
        "stream_audio",
38
        "stream_frames",
39
        "video_component",
40
        "zip_files",
41
    ]
42
    for demo in demos_to_copy:
43
        shutil.copytree(
44
            os.path.join(source_dir, demo),
45
            os.path.join(dest_dir, demo),
46
            dirs_exist_ok=True,
47
        )
48

49

50
if __name__ == "__main__":
51
    parser = argparse.ArgumentParser(
52
        description="Copy all demos to all_demos and update requirements"
53
    )
54
    parser.add_argument("gradio_version", type=str, help="Gradio")
55
    parser.add_argument("gradio_client_version", type=str, help="Gradio Client Version")
56
    args = parser.parse_args()
57

58
    source_dir = pathlib.Path(pathlib.Path(__file__).parent, "..", "demo")
59
    dest_dir = pathlib.Path(
60
        pathlib.Path(__file__).parent, "..", "demo", "all_demos", "demos"
61
    )
62
    copy_all_demos(source_dir, dest_dir)
63
    reqs_file_path = pathlib.Path(
64
        pathlib.Path(__file__).parent, "..", "demo", "all_demos", "requirements.txt"
65
    )
66
    requirements = f"""
67
    {args.gradio_client_version}
68
    {args.gradio_version}
69
    pypistats==1.1.0
70
    plotly==5.10.0
71
    opencv-python==4.6.0.66
72
    transformers==4.21.1
73
    torch==1.12.1
74
    altair
75
    vega_datasets
76
    """
77
    open(reqs_file_path, "w").write(textwrap.dedent(requirements))
78