gradio/render_readme.py
aliabid94 b4d9825409
Website: WIP (#328)
Ported gradio website into gradio repository, now launched as a docker service from gradio/website
2021-12-13 22:02:19 -08:00

41 lines
1.5 KiB
Python

import re
from os.path import join, exists, getmtime
from jinja2 import Environment, BaseLoader, TemplateNotFound
README_TEMPLATE = "readme_template.md"
GETTING_STARTED_TEMPLATE = "getting_started.md"
with open(join("guides", GETTING_STARTED_TEMPLATE)) as getting_started_file:
getting_started = getting_started_file.read()
code_tags = re.findall(r'\{\{ code\["([^\s]*)"\] \}\}', getting_started)
demo_tags = re.findall(r'\{\{ demos\["([^\s]*)"\] \}\}', getting_started)
code, demos = {}, {}
for code_src in code_tags:
with open(join("demo", code_src, "run.py")) as code_file:
python_code = code_file.read()
python_code = python_code.replace('if __name__ == "__main__":\n iface.launch()', "iface.launch()")
code[code_src] = "```python\n" + python_code + "\n```"
for demo_src in demo_tags:
demos[demo_src] = "![" + demo_src + " interface](demo/" + demo_src + "/screenshot.gif)"
class GuidesLoader(BaseLoader):
def __init__(self, path):
self.path = path
def get_source(self, environment, template):
path = join(self.path, template)
if not exists(path):
raise TemplateNotFound(template)
mtime = getmtime(path)
with open(path) as f:
source = f.read()
return source, path, lambda: mtime == getmtime(path)
readme_template = Environment(loader=GuidesLoader("guides")).get_template(README_TEMPLATE)
output_readme = readme_template.render(code=code, demos=demos)
with open("README.md", "w") as readme_md:
readme_md.write(output_readme)