Spaces:
Build error
Build error
File size: 1,945 Bytes
a37b99e 7f41b48 a37b99e 7f41b48 a37b99e 7f41b48 a37b99e 7f41b48 a37b99e 7f41b48 a37b99e 7f41b48 a37b99e 7f41b48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import streamlit as st
import openai
openai.api_key = st.secrets["openai-api-key"]
def generate_email(prompt: str, max_tokens: int = 256) -> str:
"""
Returns a generated an email using GPT3 with a certain prompt and starting sentence
"""
completions = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
temperature=0.7,
max_tokens=max_tokens,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
message = completions.choices[0].text
return message
TEMPLATE = "Write a {tone} email to the customers of a {company_type} offering {offer}"
def main():
st.title("Email Generator")
st.text("by Marc Puig")
st.sidebar.markdown("### :arrow_right: Parameters")
email_tone = st.sidebar.selectbox(
label="tone",
options=("serious", "funny", "formal"),
)
email_company_type = st.sidebar.selectbox(
label="Company type",
options=("telco company", "bank", "insurance")
)
offer = st.sidebar.text_area(
label="offer description",
value="100 minutes free if they buy 1000 minutes"
)
prompt_input = "lorem ipsum"
if email_tone and email_company_type and offer:
prompt_input = TEMPLATE.format(tone=email_tone, company_type=email_company_type, offer=offer)
max_tokens_input = st.slider(
label="How many characters do you want your email to be? ",
help="A typical email is usually 100-500 characters",
min_value=64,
max_value=750,
value=200
)
with st.form(key="form"):
if st.form_submit_button(label='Generate email', disabled=len(prompt_input) == 0):
with st.spinner("Generating email..."):
output = generate_email(prompt_input, max_tokens=max_tokens_input)
st.markdown("----")
st.markdown(output)
if __name__ == "__main__":
main()
|