import logging import os import tempfile import time import gradio as gr from boilerplate_x.generator import ProjectGenerator from blocks import customisation_block, github_repo_block logging.basicConfig(level=logging.INFO) logger = logging.getLogger("gradio") INTRO_MD = """
# Boilerplate-X Create project boilerplate for any programming language in minutes, with just an idea. Powered by Langchain and chatGPT API. [![GitHub](https://img.shields.io/badge/github-%23121011.svg?style=for-the-badge&logo=github&logoColor=white)](https://github.com/ajndkr/boilerplate-x)
""" BOOL2STR = {True: "yes", False: "no"} def generate_boilerplate( api_key: str, prompt: str, unit_tests: bool, dockerization: bool, github_actions: bool, pre_commit_hooks: bool, gh_token: str, gh_repo_name: str, private: bool, ): """Generates project boilerplate.""" if not api_key: gr.Error("Please enter your OpenAI API key!") os.environ["OPENAI_API_KEY"] = api_key with tempfile.TemporaryDirectory() as output_path: customisation_kwargs = { "unit_tests": BOOL2STR[unit_tests], "dockerization": BOOL2STR[dockerization], "github_actions": BOOL2STR[github_actions], "pre_commit_hooks": BOOL2STR[pre_commit_hooks], } if not gh_token: gr.Error("Please enter your GitHub token!") if not gh_repo_name: gr.Error("Please enter your GitHub repository name!") github_repo_creator_kwargs = { "token": gh_token, "repo_name": gh_repo_name, "private": private, "target_folder": output_path, } generator = ProjectGenerator( prompt=prompt, output_path=output_path, verbose=True, customisation_kwargs=customisation_kwargs, github_repo_creator_kwargs=github_repo_creator_kwargs, ) try: logger.info("Generating project boilerplate...") generator.generate_template() except Exception as e: logger.error(e) raise gr.Error(e) # wait for GitHub repo to be created time.sleep(1) return f"Your project is now available on {generator.github_repo_url} 🚀 !" def build_app(): """Builds the Gradio UI.""" block = gr.Blocks(title="Boilerplate X") with block: gr.Markdown(INTRO_MD) openai_api_key = gr.Textbox( placeholder="Paste your OpenAI API key", label="OpenAI API Key", lines=1, type="password", ) prompt = gr.Textbox( placeholder="Enter your project idea", label="Prompt", lines=1, ) ( unit_tests, dockerization, github_actions, pre_commit_hooks, ) = customisation_block() gh_token, gh_repo_name, private = github_repo_block() generate = gr.Button("⚡ Generate Boilerplate ⚡") output = gr.Markdown() generate.click( fn=generate_boilerplate, inputs=[ openai_api_key, prompt, unit_tests, dockerization, github_actions, pre_commit_hooks, gh_token, gh_repo_name, private, ], outputs=[output], ) return block if __name__ == "__main__": app = build_app() app.queue().launch()