| import argparse | |
| import os | |
| import pathlib | |
| import shutil | |
| import textwrap | |
| def copy_all_demos(source_dir: str, dest_dir: str): | |
| demos_to_copy = [ | |
| "audio_debugger", | |
| "altair_plot", | |
| "blocks_essay", | |
| "blocks_group", | |
| "blocks_js_methods", | |
| "blocks_layout", | |
| "blocks_multiple_event_triggers", | |
| "blocks_update", | |
| "calculator", | |
| "cancel_events", | |
| "chatbot_multimodal", | |
| "chatinterface_streaming_echo", | |
| "clear_components", | |
| "code", | |
| "fake_gan", | |
| "fake_diffusion_with_gif", | |
| "image_mod_default_image", | |
| "image_segmentation", | |
| "interface_random_slider", | |
| "kitchen_sink", | |
| "kitchen_sink_random", | |
| "matrix_transpose", | |
| "model3D", | |
| "native_plots", | |
| "reverse_audio", | |
| "stt_or_tts", | |
| "stream_audio", | |
| "stream_frames", | |
| "video_component", | |
| "zip_files", | |
| ] | |
| for demo in demos_to_copy: | |
| shutil.copytree( | |
| os.path.join(source_dir, demo), | |
| os.path.join(dest_dir, demo), | |
| dirs_exist_ok=True, | |
| ) | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser( | |
| description="Copy all demos to all_demos and update requirements" | |
| ) | |
| parser.add_argument("gradio_version", type=str, help="Gradio") | |
| parser.add_argument("gradio_client_version", type=str, help="Gradio Client Version") | |
| args = parser.parse_args() | |
| source_dir = pathlib.Path(pathlib.Path(__file__).parent, "..", "demo") | |
| dest_dir = pathlib.Path( | |
| pathlib.Path(__file__).parent, "..", "demo", "all_demos", "demos" | |
| ) | |
| copy_all_demos(source_dir, dest_dir) | |
| reqs_file_path = pathlib.Path( | |
| pathlib.Path(__file__).parent, "..", "demo", "all_demos", "requirements.txt" | |
| ) | |
| requirements = f""" | |
| {args.gradio_client_version} | |
| {args.gradio_version} | |
| pypistats==1.1.0 | |
| plotly==5.10.0 | |
| opencv-python==4.6.0.66 | |
| transformers==4.21.1 | |
| torch==1.12.1 | |
| altair | |
| vega_datasets | |
| """ | |
| open(reqs_file_path, "w").write(textwrap.dedent(requirements)) | |