Peter commited on
Commit
51de900
1 Parent(s): b1deccc

support email links

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  import torch
6
  from transformers import pipeline
7
 
8
- from utils import postprocess, clear
9
 
10
  logging.basicConfig(
11
  level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -68,7 +68,7 @@ def generate_text(
68
  if verbose:
69
  logging.info(f"Generated text: {response}")
70
  logging.info(f"Generation time: {rt:.2f}s")
71
- return postprocess(response)
72
 
73
 
74
  def get_parser():
@@ -153,10 +153,12 @@ if __name__ == "__main__":
153
  label="Generated Result",
154
  placeholder="The completed email will appear here",
155
  )
156
- generate_button = gr.Button(
157
- value="Generate!",
158
- variant="primary",
159
- )
 
 
160
 
161
  gr.Markdown("## Advanced Options")
162
  gr.Markdown(
@@ -208,7 +210,7 @@ if __name__ == "__main__":
208
  no_repeat_ngram_size,
209
  length_penalty,
210
  ],
211
- outputs=[generated_email],
212
  )
213
 
214
  demo.launch(
 
5
  import torch
6
  from transformers import pipeline
7
 
8
+ from utils import postprocess, clear, make_email_link
9
 
10
  logging.basicConfig(
11
  level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
 
68
  if verbose:
69
  logging.info(f"Generated text: {response}")
70
  logging.info(f"Generation time: {rt:.2f}s")
71
+ return postprocess(response), make_email_link(body=response)
72
 
73
 
74
  def get_parser():
 
153
  label="Generated Result",
154
  placeholder="The completed email will appear here",
155
  )
156
+ with gr.Row():
157
+ generate_button = gr.Button(
158
+ value="Generate!",
159
+ variant="primary",
160
+ )
161
+ email_link = gr.HTML("<p><em>A mailto: link will appear here</em></p>")
162
 
163
  gr.Markdown("## Advanced Options")
164
  gr.Markdown(
 
210
  no_repeat_ngram_size,
211
  length_penalty,
212
  ],
213
+ outputs=[generated_email, email_link],
214
  )
215
 
216
  demo.launch(