Adding parameters

app.py

@@ -1,12 +1,6 @@
 import streamlit as st
 import openai
 
-EXAMPLES = [
-    "Write a professional email to the customers of a telco company offering 100 minutes free if they buy 1000 minutes",
-    "Write a formal email to the customers of a bank offering a new home insurance",
-    "Write an informal email to the customers of a bank, offering a new car insurance, using emojis in the subject",
-]
-
 openai.api_key = st.secrets["openai-api-key"]
 
 
@@ -28,38 +22,50 @@ def generate_email(prompt: str, max_tokens: int = 256) -> str:
     return message
 
 
-def example_selected():
-    st.session_state['prompt'] = st.session_state['selected_example']
+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")
+    )
 
-st.title("Email Generator")
-st.text("by Marc Puig")
+    offer = st.sidebar.text_area(
+        label="offer description",
+        value="100 minutes free if they buy 1000 minutes"
+    )
 
-st.selectbox(
-    label="Examples",
-    options=EXAMPLES,
-    on_change=example_selected,
-    key="selected_example"
-)
+    prompt_input = "lorem ipsum"
 
-prompt_input = st.text_area(
-    label="Describe the type of email you want to be written.",
-    key="prompt"
-)
+    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
+    )
 
-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)
 
-with st.form(key="form"):
-    submit_button = st.form_submit_button(label='Generate email', disabled=len(st.session_state["prompt"]) == 0)
 
-    if submit_button:
-        with st.spinner("Generating email..."):
-            output = generate_email(prompt_input, max_tokens=max_tokens_input)
-        st.markdown("----")
-        st.markdown(output)
+if __name__ == "__main__":
+    main()