Merge branch 'main' of https://huggingface.co/spaces/postbot/autocomplete-emails

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import torch
 from transformers import pipeline
 
-from utils import postprocess, clear
+from utils import postprocess, clear, make_email_link
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -68,7 +68,9 @@ def generate_text(
     if verbose:
         logging.info(f"Generated text: {response}")
     logging.info(f"Generation time: {rt:.2f}s")
-    return postprocess(response)
+
+    formatted_email = postprocess(response)
+    return formatted_email, make_email_link(body=formatted_email)
 
 
 def get_parser():
@@ -153,6 +155,8 @@ if __name__ == "__main__":
                 label="Generated Result",
                 placeholder="The completed email will appear here",
             )
+            email_link = gr.HTML("<p><em>A mailto: link will appear here</em></p>")
+
             generate_button = gr.Button(
                 value="Generate!",
                 variant="primary",
@@ -208,7 +212,7 @@ if __name__ == "__main__":
                 no_repeat_ngram_size,
                 length_penalty,
             ],
-            outputs=[generated_email],
+            outputs=[generated_email, email_link],
         )
 
     demo.launch(

utils.py

@@ -3,6 +3,7 @@
 """
 import logging
 
+
 def postprocess(text: str):
     """
     postprocess - remove common values in scraped dataset
@@ -40,3 +41,25 @@ def clear(text, verbose=False, **kwargs):
     if verbose:
         logging.info(f"Clearing text: {text}")
     return ""
+
+
+def make_email_link(
+    subject: str = "Email subject - This was generated by Postbot",
+    link_text: str = "click to open in your email client",
+    body: str = None,
+):
+    """
+    email_link - generate an email link
+
+    Args:
+        subject (str, optional): the subject of the email. Defaults to "Email subject - This was generated by Postbot".
+        link_text (str, optional): the text of the link. Defaults to "click to open in your email client".
+        body (_type_, optional): the body of the email. Defaults to None.
+
+    Returns:
+        str: the email link
+    """
+
+    if body is None:
+        body = "hmm - no body. replace me"
+    return f'<a href="mailto:%20?subject={subject}&body={body}">{link_text}</a>'