JohanDL commited on
Commit
e425200
·
verified ·
1 Parent(s): e01f2c9

Update to use tabs.

Browse files
Files changed (1) hide show
  1. app.py +14 -12
app.py CHANGED
@@ -57,10 +57,7 @@ def generate_repo(url, api_key, model, output_format):
57
  # except:
58
  # return "The API key either does not have access to GPT-4 or is not valid."
59
  readme_contents = process_repository(url, model)
60
- if output_format == "Rendered Markdown":
61
- return readme_contents, None
62
- else:
63
- return None, readme_contents
64
  else:
65
  return "Please add a valid OpenAI API Key (you can get them [here](https://platform.openai.com/account/api-keys))"
66
 
@@ -87,17 +84,22 @@ with gr.Blocks(css=css_style) as demo:
87
  label="OpenAI API Key", placeholder="sk-...", type="password")
88
  url = gr.Textbox(label="GitHub Repository URL")
89
  model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], type="value", label='Model Type')
90
- output_format = gr.Radio(choices=["Rendered Markdown", "Plain Markdown"], label="Output Format", value="Rendered Markdown")
91
- output_md = gr.Markdown(label="Rendered README.md", visible=False)
92
- output_text = gr.Text(label="Plain Markdown README.md", visible=False)
93
  btn = gr.Button("Generate README.md")
94
 
95
- def update_output(*args):
96
- md_content, text_content = generate_repo(*args)
97
- output_md.update(value=md_content, visible=md_content is not None)
98
- output_text.update(value=text_content, visible=text_content is not None)
 
 
99
 
100
- btn.click(fn=update_output, inputs=[url, openai_api_key, model, output_format], outputs=[output_md, output_text])
 
 
 
 
101
 
102
 
103
  demo.queue(concurrency_count=20)
 
57
  # except:
58
  # return "The API key either does not have access to GPT-4 or is not valid."
59
  readme_contents = process_repository(url, model)
60
+ return readme_contents
 
 
 
61
  else:
62
  return "Please add a valid OpenAI API Key (you can get them [here](https://platform.openai.com/account/api-keys))"
63
 
 
84
  label="OpenAI API Key", placeholder="sk-...", type="password")
85
  url = gr.Textbox(label="GitHub Repository URL")
86
  model = gr.Dropdown(["gpt-3.5-turbo", "gpt-4"], type="value", label='Model Type')
87
+ output_md = gr.Markdown()
88
+ output_text = gr.Text()
 
89
  btn = gr.Button("Generate README.md")
90
 
91
+ def update_output(url, api_key, model):
92
+ content = generate_repo(url, api_key, model)
93
+ output_md.update(value=content)
94
+ output_text.update(value=content)
95
+
96
+ btn.click(fn=update_output, inputs=[url, openai_api_key, model], outputs=[])
97
 
98
+ with gr.Tabs() as tabs:
99
+ with gr.TabItem("Rendered Markdown"):
100
+ output_md
101
+ with gr.TabItem("Plain Markdown"):
102
+ output_text
103
 
104
 
105
  demo.queue(concurrency_count=20)