Add rendered md option for easier copying.

app.py

@@ -43,7 +43,7 @@ def process_repository(url, model):
         if os.path.exists(repo_name):
             shutil.rmtree(repo_name)
 
-def generate_repo(url, api_key, model):
+def generate_repo(url, api_key, model, output_format):
     if api_key:
         os.environ['OPENAI_API_KEY'] = api_key.strip()
         # if model == 'gpt-4':
@@ -56,7 +56,11 @@ def generate_repo(url, api_key, model):
         #         print("Access to GPT-4 confirmed!")
         #     except:
         #         return "The API key either does not have access to GPT-4 or is not valid."
-        return process_repository(url, model)
+        readme_contents = process_repository(url, model)
+        if output_format == "Rendered Markdown":
+            return readme_contents, None
+        else:
+            return None, readme_contents
     else:
         return "Please add a valid OpenAI API Key (you can get them [here](https://platform.openai.com/account/api-keys))"
 
@@ -83,9 +87,17 @@ with gr.Blocks(css=css_style) as demo:
             label="OpenAI API Key", placeholder="sk-...", type="password")
     url = gr.Textbox(label="GitHub Repository URL")
     model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], type="value", label='Model Type')
-    output = gr.Markdown(label="README.md")
+    output_format = gr.Radio(choices=["Rendered Markdown", "Plain Markdown"], label="Output Format", value="Rendered Markdown")
+    output_md = gr.Markdown(label="Rendered README.md", visible=False)
+    output_text = gr.Text(label="Plain Markdown README.md", visible=False)
     btn = gr.Button("Generate README.md")
-    btn.click(fn=generate_repo, inputs=[url, openai_api_key, model], outputs=[output], api_name="Generate README.md")
+
+    def update_output(*args):
+        md_content, text_content = generate_repo(*args)
+        output_md.update(value=md_content, visible=md_content is not None)
+        output_text.update(value=text_content, visible=text_content is not None)
+
+    btn.click(fn=update_output, inputs=[url, openai_api_key, model, output_format], outputs=[output_md, output_text])
     
 
 demo.queue(concurrency_count=20)