update application

app.py

@@ -1,7 +1,123 @@
+import os
+
 import gradio as gr
+from boilerplate_x.generator import ProjectGenerator
+
+from blocks import customisation_block, github_repo_block
+
+INTRO_MD = """
+<center>
+
+# Boilerplate-X
+
+Create project boilerplate for any programming language in minutes, with just an idea. Powered by Langchain and chatGPT API.
+
+</center>
+"""
+
+OUTPUT_PATH = "outputs/"
+OPT_MAP = {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
+    output_path = OUTPUT_PATH
+
+    customisation_kwargs = {
+        "unit_tests": OPT_MAP[unit_tests],
+        "dockerization": OPT_MAP[dockerization],
+        "github_actions": OPT_MAP[github_actions],
+        "pre_commit_hooks": OPT_MAP[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=False,
+        customisation_kwargs=customisation_kwargs,
+        github_repo_creator_kwargs=github_repo_creator_kwargs,
+    )
+
+    try:
+        generator.generate_template()
+    except Exception as e:
+        gr.Error(e)
+
+    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
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+if __name__ == "__main__":
+    app = build_app()
+    app.queue().launch()

blocks.py

@@ -0,0 +1,104 @@
+import gradio as gr
+
+
+def select_custom_options(customisation):
+    """Selects the custom options."""
+    if customisation:
+        return (
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(visible=True),
+        )
+    return (
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=False),
+    )
+
+
+def select_github_repo_options(github_repo):
+    """Selects the Github repo options."""
+    if github_repo:
+        return (
+            gr.update(visible=True),
+            gr.update(visible=True),
+            gr.update(visible=True),
+        )
+    return (
+        gr.update(visible=False),
+        gr.update(visible=False),
+        gr.update(visible=False),
+    )
+
+
+def customisation_block() -> dict:
+    with gr.Column():
+        enable_customisation = gr.Checkbox(
+            label="Enable customisation (Click to reveal options)",
+            default=False,
+            interactive=True,
+        )
+        with gr.Row():
+            unit_tests = gr.Checkbox(
+                label="Add Unit tests",
+                default=False,
+                interactive=True,
+                visible=False,
+            )
+            dockerization = gr.Checkbox(
+                label="Add Docker support",
+                default=False,
+                interactive=True,
+                visible=False,
+            )
+            github_actions = gr.Checkbox(
+                label="Add Github Actions support",
+                default=False,
+                interactive=True,
+                visible=False,
+            )
+            pre_commit_hooks = gr.Checkbox(
+                label="Add Pre-commit hooks",
+                default=False,
+                interactive=True,
+                visible=False,
+            )
+
+        enable_customisation.change(
+            fn=select_custom_options,
+            inputs=enable_customisation,
+            outputs=[unit_tests, dockerization, github_actions, pre_commit_hooks],
+        )
+
+    return (
+        unit_tests,
+        dockerization,
+        github_actions,
+        pre_commit_hooks,
+    )
+
+
+def github_repo_block() -> dict:
+    with gr.Column():
+        with gr.Row():
+            gh_token = gr.Textbox(
+                placeholder="Paste your Github Personal Access Token (Required scopes: repo, workflow)",
+                label="Github Personal Access Token",
+                lines=1,
+                type="password",
+            )
+            gh_repo_name = gr.Textbox(
+                placeholder="Enter Github repo name",
+                label="Github repo name",
+                lines=1,
+            )
+
+        private = gr.Checkbox(
+            label="Make repo private?",
+            default=False,
+            interactive=True,
+        )
+
+    return gh_token, gh_repo_name, private