Use system instructions to steer the behavior of a model

lib/ai_writers/twitter_ai_writer.py

@@ -13,59 +13,84 @@ def tweet_writer():
         col1, col2 = st.columns([5, 5])
         with col1:
             hook = st.text_input(
-                    label="What's the tweet about:(Hook)",
-                    placeholder="e.g., Discover the future of tech today!",
-                    help="Provide a compelling opening statement or question to grab attention."
+                label="**What's the tweet about? (Hook)**",
+                placeholder="e.g., Discover the future of tech today!",
+                help="Provide a compelling opening statement or question to grab attention."
             )
 
         with col2:
-            # Collect user inputs with placeholders and help text
             target_audience = st.text_input(
-                label="Target Audience",
+                label="**Target Audience**",
                 placeholder="e.g., technology enthusiasts, travel lovers",
                 help="Describe the audience you want to target with this tweet."
             )
 
+        col3, col4 = st.columns([5, 5])
+        with col3:
+            tweet_tone = st.selectbox(
+                label="**Tweet Tone**",
+                options=["Humorous", "Informative", "Inspirational", "Serious", "Casual"],
+                help="Choose the tone you'd like the tweet to have."
+            )
+
+        with col4:
+            cta = st.text_input(
+                label="**Call to Action (Optional)**",
+                placeholder="e.g., Retweet this if you agree! (Leave blank if not applicable)",
+                help="Provide a call to action if you'd like to include one."
+            )
+
+        col5, col6 = st.columns([5, 5])
+        with col5:
+            keywords_hashtags = st.text_input(
+                label="**Keywords/Hashtags**",
+                placeholder="e.g., #AI #Innovation",
+                help="Provide 2-3 relevant keywords or hashtags."
+            )
+
+        with col6:
+            tweet_length = st.selectbox(
+                "Tweet Length (Optional)",
+                options=["Short (under 100 characters)", "Medium (100-200 characters)", "Long (200+ characters)"],
+                help="Choose the desired tweet length.",
+            )
+
     if st.button('**Write Tweets**'):
         if not target_audience or not hook:
             st.error("🚫 Please provide all required inputs.")
         else:
-            with st.status("Assigning AI professional to write your Google Ads copy..", expanded=True) as status:
-                response = tweet_generator(target_audience, hook)
+            with st.status("Assigning AI professional to write your tweets...", expanded=True) as status:
+                response = tweet_generator(target_audience, hook, tweet_tone, cta, keywords_hashtags, tweet_length)
                 if response:
                     st.subheader(f'**🧕👩: Your Tweets!**')
                     st.markdown(response)
                 else:
-                    st.error("💥**Failed to write Letter. Please try again!**")
+                    st.error("💥**Failed to generate tweets. Please try again!**")
 
 
-def tweet_generator(target_audience, hook):
-    """ Email project_update_writer """
+def tweet_generator(target_audience, hook, tone_style, cta, keywords_hashtags, tweet_length):
+    """ Tweet Generator """
 
     prompt = f"""
-    You are a social media expert creating tweets for an audience interested in {target_audience}. 
-    Write 5 engaging, concise, and visually appealing tweets that each:
-
-    1. Start with a compelling hook based on the following keywords: "{hook}"
-    2. Include a compelling call to action.
-    3. Use 2-3 relevant hashtags. 
-    4. Adopt a tone that matches the following options: 
-        - Humorous 
-        - Informative 
-        - Inspirational 
-        - Serious 
-        - Casual
-    5. Be under 100 characters (including spaces and punctuation). 
-
-    Here are some examples of call-to-actions to include:
-    - Retweet this if you agree!
-    - Share your thoughts in the comments!
-    - Learn more at [link] 
-    - Follow for more [topic] content
-    - Like if you're excited about [topic]
-
-    Output each tweet separated by a newline.
-    """
+	    You are a social media expert creating tweets for an audience interested in {target_audience}. 
+	    Write 5 engaging, concise, and visually appealing tweets that each:
+	
+	    1. Start with a compelling hook based on the following input: "{hook}"
+	    2. Include the following call to action: "{cta}" 
+	    3. Use 2-3 relevant keywords/hashtags, including: "{keywords_hashtags}"
+	    4. Adopt the following tone/style: "{tone_style}"
+	    5. Adhere to the following length requirement: {tweet_length} 
+	    
+	    Make sure to keep the tone consistent with the selected style and platform context.
+	
+	    Here are some examples of call-to-actions to include (if no specific CTA was provided): 
+	    - Retweet this if you agree!
+	    - Share your thoughts in the comments!
+	    - Learn more at [link] 
+	    - Follow for more {target_audience} content.
+	
+	    Output each tweet separated by a newline.
+	    """
 
     try:
         response = llm_text_gen(prompt)