alwrity_facebook.py

import time
import os
import json
import requests
import streamlit as st
from tenacity import retry, stop_after_attempt, wait_random_exponential
import google.generativeai as genai

def generate_facebook_post(business_type, target_audience, post_goal, post_tone, include, avoid, progress_callback):
    """
    Generates a Facebook post prompt for an LLM based on user input.

    Args:
        business_type: The type of business, e.g., fashion retailer, fitness coach.
        target_audience: A description of the target audience.
        post_goal: The goal of the Facebook post.
        post_tone: The desired tone of the post.
        include: Elements to include in the post (e.g., images, videos, links).
        avoid: Elements to avoid in the post (e.g., long paragraphs, technical jargon).
        progress_callback: Function to update progress.

    Returns:
        A string containing the LLM prompt.
    """
    progress_callback(10)
    prompt = f"""
    My business type is {business_type}.

    Please help me write a highly detailed Facebook post, at least 1000-2000 words, that will engage my target audience, {target_audience}.

    Here are some additional details to consider:

    * **Post Goal:** {post_goal}
    * **Post Tone:** {post_tone}
    * **Include:** {include}
    * **Avoid:** {avoid}

    **Example Post Structure:**

    1. **Attention-Grabbing Opening:** Start with a question or a bold statement to capture attention.
    2. **Engaging Content:** Describe the main message or offer, highlighting key benefits or features.
    3. **Call-to-Action (CTA):** Encourage & provide compelling reasons for the audience to take a specific action (e.g., visit a link, comment, share).
    4. **Hashtags:** Include relevant hashtags to increase post visibility.
    """
    progress_callback(30)
    try:
        response = generate_text_with_exception_handling(prompt, progress_callback)
        progress_callback(100)
        return response
    except Exception as err:
        st.error(f"An error occurred while generating the prompt: {err}")
        return None

@retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6))
def generate_text_with_exception_handling(prompt, progress_callback):
    """
    Generates text using the Gemini model with exception handling.

    Args:
        prompt (str): The prompt for text generation.
        progress_callback: Function to update progress.

    Returns:
        str: The generated text.
    """
    progress_callback(50)
    try:
        genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
        generation_config = {
            "temperature": 0.7,
            "top_p": 0.95,
            "top_k": 0,
            "max_output_tokens": 4096,
        }
        safety_settings = [
            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
        ]
        model = genai.GenerativeModel(
            model_name="gemini-1.5-flash",
            generation_config=generation_config,
            safety_settings=safety_settings
        )
        progress_callback(70)
        convo = model.start_chat(history=[])
        convo.send_message(prompt)
        progress_callback(90)
        return convo.last.text
    except Exception as e:
        st.exception(f"An unexpected error occurred: {e}")
        return None

def main():
    st.markdown("""
        <style>
        ::-webkit-scrollbar-track { background: #e1ebf9; }
        ::-webkit-scrollbar-thumb {
            background-color: #90CAF9;
            border-radius: 10px;
            border: 3px solid #e1ebf9;
        }
        ::-webkit-scrollbar-thumb:hover { background: #64B5F6; }
        ::-webkit-scrollbar { width: 16px; }
        div.stButton > button:first-child {
            background: #1565C0;
            color: white;
            border: none;
            padding: 12px 24px;
            border-radius: 8px;
            text-align: center;
            text-decoration: none;
            display: inline-block;
            font-size: 16px;
            margin: 10px 2px;
            cursor: pointer;
            transition: background-color 0.3s ease;
            box-shadow: 2px 2px 5px rgba(0, 0, 0, 0.2);
            font-weight: bold;
        }
        </style>
    """, unsafe_allow_html=True)
    st.title("Alwrity - AI Facebook Post Generator")
    
    with st.expander("**PRO-TIP** - Read the instructions below.", expanded=True):
        col1, col2 = st.columns(2)

        with col1:
            business_type = st.text_input("**🏢 What is your business type?**", placeholder="e.g., fitness coach", help="Enter the type of your business.")
            post_goal = st.selectbox("**🎯 What is the goal of your post?**", ["Promote a new product", "Share valuable content", "Increase engagement", "Other"], index=2)
            
        with col2:
            target_audience = st.text_input("**🎯 Describe your target audience:**", placeholder="e.g., fitness enthusiasts", help="Describe who you want to reach with this post.")
            post_tone = st.selectbox("**🗣️ What tone do you want to use?**", ["Informative", "Humorous", "Inspirational", "Upbeat", "Casual"], index=3)

        include = st.text_input("**📋 What elements do you want to include?**", placeholder="e.g., (Optional) short video with a sneak peek, Image", help="Specify elements to include like images, videos, links.")
        avoid = st.text_input("**🚫 What elements do you want to avoid?**", placeholder="e.g., (Optional) Robotic Tone, long paragraphs, Incorrect information", help="Specify elements to avoid like long paragraphs or technical jargon.")

    progress_bar = st.empty()
    progress_text = st.empty()

    if st.button("**✨ Generate Your FB Post Now!**"):
        if not business_type or not target_audience:
            st.error("🚫 Provide required inputs. Least, you can do..")
        else:
            progress_bar = st.progress(0)
            progress_text = st.empty()

            def progress_callback(progress):
                progress_bar.progress(progress)
                progress_text.text(f"Progress: {progress}%")

            generated_post = generate_facebook_post(business_type, target_audience, post_goal, post_tone, include, avoid, progress_callback)
            if generated_post:
                st.write("**🧕 Verify: Alwrity can make mistakes. To err is Human & AI..**")
                st.markdown(generated_post)
                st.write("\n\n")
            else:
                st.error("Error: Failed to generate Facebook Post.")
        progress_bar.empty()
        progress_text.empty()

if __name__ == "__main__":
    main()