Spaces:
Running
Running
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 = """ | |
<center> | |
# 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) | |
</center>""" | |
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() | |