Update app.py

app.py

@@ -5,6 +5,7 @@ from langchain import FewShotPromptTemplate
 from langchain.prompts.example_selector import LengthBasedExampleSelector
 from dotenv import load_dotenv
 
+load_dotenv() # load the env-sample.txt file
 
 def getLLMResponse(query, age_option,tasktype_option):
     examples = []
@@ -55,15 +56,7 @@ def getLLMResponse(query, age_option,tasktype_option):
     return response
 
 #UI Starts here
-# Load environment variables
-load_dotenv()  # Make sure your .env file path is correct
-
-# Streamlit App Configuration
-st.set_page_config(page_title="PitchPal: Your Friendly Copy Assistant",
-                   page_icon="💻",
-                   layout="wide",
-                   initial_sidebar_state="collapsed")
-
+    
 # Custom CSS for styling
 st.markdown(
     """
@@ -81,10 +74,29 @@ st.markdown(
     </style>
     """, unsafe_allow_html=True)
 
-# Header Section
-st.markdown("<h1 style='text-align: center; color: #1144aa'>PitchPal: Your Efficient Sales Copy Assistant</h1>", unsafe_allow_html=True)
-st.markdown("<h3 style='text-align: center; color: #333'>Craft compelling sales copy with ease</h3>", unsafe_allow_html=True)
-st.markdown("<p style='text-align: right; font-size:14px;'>By <a href='https://entzyeung.github.io/portfolio/index.html'>Lorentz Yeung</a></p>", unsafe_allow_html=True)
+
+st.set_page_config(page_title="PitchPal: Your Friendly Copy Assistant",
+                    page_icon='💻',
+                    layout='centered',
+                    initial_sidebar_state='collapsed')
+
+
+st.markdown("<h1 style='text-align: center'>PitchPal</h1>", unsafe_allow_html=True)
+st.markdown("<h3 style='text-align: center'>Your Efficient Sales Copy Assistant</h2>", unsafe_allow_html=True)
+st.markdown("<p style='text-align: right'>By <a href='https://entzyeung.github.io/portfolio/index.html'>Lorentz Yeung</a></p>", unsafe_allow_html=True)
+
+
+#st.title("PitchPal")
+#st.subheader("Your Friendly Sales Copy Assistant")
+#st.markdown(
+#        """     
+#        - by [Lorentz Yeung]()
+#        """
+#        )
+
+form_input = st.text_area('Enter the name of the product or service you want to promote: ', 'PlayStation 6', height=100)
+# st.write(f'You wrote {len(form_input)} characters.')
+
 
 # User Input Section with Improved Layout
 col1, col2 = st.columns(2)
@@ -95,20 +107,18 @@ with col1:
 with col2:
     tasktype_option = st.selectbox(
         'Marketing copy type:',
-        ('Twitter post', 'Sales copy', 'Product description'),
+        ('Draft a Twitter post', 'Draft a sales copy', 'Draft a product description'),
         index=1)
     age_option = st.selectbox(
-        'Audience age group:',
-        ('Below 18', '18-45', '46-65', '> 65'),
+        'Target customers age group:',
+        ('below age 18', 'age 18-45', 'age 46-65', 'age > 65'),
         index=1)
 
-# Submit Button for Generating Sales Copy
-submit = st.button("Generate Sales Copy")
 
-if submit:
-    response = getLLMResponse(form_input, age_option, tasktype_option)
-    st.markdown("## Generated Sales Copy")
-    st.write(response)  # Display the LLM response
+# numberOfWords= st.slider('Words limit', 1, 200, 25)
+numberOfWords = 40 # the new model doesn't support this.
+
+submit = st.button("Generate Your Sales Copy")
 
-# Note: Ensure that all functions and logic related to LLM response generation are correctly implemented
-# and replace placeholder texts and functions with your actual application code.
+if submit:
+    st.write(getLLMResponse(form_input,tasktype_option,age_option))