101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
import time #Iwish
|
|
import os
|
|
import json
|
|
import requests
|
|
import streamlit as st
|
|
|
|
from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
|
|
|
|
|
|
def tweet_writer():
|
|
""" AI Tweet Generator """
|
|
with st.expander("**PRO-TIP** - Read the instructions below.", expanded=True):
|
|
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."
|
|
)
|
|
|
|
with col2:
|
|
target_audience = st.text_input(
|
|
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 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 generate tweets. Please try again!**")
|
|
|
|
|
|
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 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)
|
|
return response
|
|
except Exception as err:
|
|
st.error(f"Exit: Failed to get response from LLM: {err}")
|
|
exit(1)
|