Spaces:

postbot
/

autocomplete-emails

Running

App Files Files Community

pszemraj commited on Oct 19, 2022

Commit

a0d66c9

1 Parent(s): 9dd82fc

✨ add better output

Browse files

Signed-off-by: peter szemraj <[email protected]>

Files changed (2) hide show

app.py +10 -10
utils.py +66 -0

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import gradio as gr
 import torch
 from transformers import pipeline
-from utils import postprocess, clear, make_email_link
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -68,7 +68,7 @@ def generate_text(
     logging.info(f"Generation time: {rt:.2f}s")
     formatted_email = postprocess(response)
-    return formatted_email, make_email_link(body=formatted_email)
 def load_emailgen_model(model_tag:str):
     """
@@ -171,7 +171,7 @@ 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!",
@@ -192,13 +192,13 @@ if __name__ == "__main__":
                     'Load Model',
                     variant='secondary',
                 )
-                num_beams = gr.Radio(
-                    choices=[4, 8, 12, 16],
-                    label="Number of Beams",
-                    value=8,
-                )
-            with gr.Row():
                 no_repeat_ngram_size = gr.Radio(
                     choices=[1, 2, 3, 4],
                     label="no repeat ngram size",
@@ -238,7 +238,7 @@ if __name__ == "__main__":
                 no_repeat_ngram_size,
                 length_penalty,
             ],
-            outputs=[generated_email, email_link],
         )
         load_model_button.click(

 import torch
 from transformers import pipeline
+from utils import make_mailto_form, postprocess, clear, make_email_link
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
     logging.info(f"Generation time: {rt:.2f}s")
     formatted_email = postprocess(response)
+    return make_mailto_form(body=formatted_email)
 def load_emailgen_model(model_tag:str):
     """
                 label="Generated Result",
                 placeholder="The completed email will appear here",
             )
+            email_template = gr.HTML("<p><em>A mailto: link will appear here</em></p>")
             generate_button = gr.Button(
                 value="Generate!",
                     'Load Model',
                     variant='secondary',
                 )
+            with gr.Row():
+                num_beams = gr.Radio(
+                                    choices=[4, 8, 12, 16],
+                                    label="Number of Beams",
+                                    value=4,
+                                )
                 no_repeat_ngram_size = gr.Radio(
                     choices=[1, 2, 3, 4],
                     label="no repeat ngram size",
                 no_repeat_ngram_size,
                 length_penalty,
             ],
+            outputs=[generated_email, email_template],
         )
         load_model_button.click(

utils.py CHANGED Viewed

@@ -72,3 +72,69 @@ def make_email_link(
     body = body.replace("\n", " ")
     return f'<a href="mailto:%20?subject={subject}&body={body}">{link_text}</a>'

     body = body.replace("\n", " ")
     return f'<a href="mailto:%20?subject={subject}&body={body}">{link_text}</a>'
+def make_mailto_form(body: str=None, subject: str="This email was generated by Postbot with AI!", cc_email: str="",):
+    """Returns a mailto link with the given parameters"""
+    if body is None:
+        body = "hmm - no body. Replace me or try rerunning the model."
+    template = f"""<!DOCTYPE html>
+    <html>
+    <head>
+        <title>Generated Email</title>
+    <style>
+        body {{
+            font-family: sans-serif;
+            font-size: 1.2em;
+        }}
+        .container {{
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            height: 100vh;
+        }}
+        .button {{
+            background-color: #6CCEC6;
+            border: none;
+            color: white;
+            padding: 15px 32px;
+            text-align: center;
+            text-decoration: none;
+            display: inline-block;
+            font-size: 16px;
+            margin: 4px 2px;
+            cursor: pointer;
+            value: "Send Email";
+        }}
+    </style>
+    </head>
+    <body>
+        <h1>Adjust and Open in your mail client:</h1>
+        <form action="mailto:" method="get" enctype="text/plain">
+        <div>
+            <label for="cc">CC Email:
+            <input type="text" name="cc" id="cc" value="{cc_email}"/>
+            </label>
+        </div>
+        <div>
+            <label for="subject">Subject:
+            <input type="text" name="subject" id="subject" value="{subject}"/>
+            </label>
+        </div>
+        <div>
+            <label>Email Body:</label>
+            <br />
+            <textarea name="body" id="body" rows="12" cols="35">{body}</textarea>
+        </div>
+        <div>
+            <input type="submit" name="submit" value="Open in Email App" class="button"/>
+        </div>
+        </form>
+    </body>
+    </html>"""
+    return template