Upload 6 files

app.py

@@ -0,0 +1,69 @@
+import streamlit as st
+from post_generator import *
+
+# Options for length and language
+length_options = ["Short", "Medium", "Long"]
+language_options = ["English", "Hinglish"]
+
+# Main app layout
+def main():
+    st.title("🕵️ LinkedIn Content Generator")
+    st.markdown(
+        """<style>
+        .css-1oe6wy4, .css-1y4p8pa {
+            background-color: #f8f9fa;
+            border-radius: 15px;
+            box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+        }
+        .css-1v0mbdj {
+            background-color: #ffffff;
+            padding: 20px;
+            border-radius: 10px;
+            box-shadow: 0 2px 5px rgba(0, 0, 0, 0.1);
+        }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
+
+    st.markdown("**The code replicates the writing style of the selected author and generates posts accordingly. You can customize the raw data to match any author of your choice.**")
+    st.markdown("**Using llama-3.3-70b-versatile model with groc API and Prompt Engineering**")
+
+    st.markdown("---")
+
+    # Organize UI into sections
+    st.header("Choose Post Parameters")
+
+    fs = FewShotPosts()
+    tags = fs.get_tags()
+
+    # Create three columns for inputs
+    col1, col2, col3 = st.columns([1.5, 1, 1])
+    with col1:
+        selected_tag = st.selectbox("Select a Topic", options=tags)
+
+    with col2:
+        selected_length = st.selectbox("Select Length", options=length_options)
+
+    with col3:
+        selected_language = st.selectbox("Select Language", options=language_options)
+
+    st.markdown("---")
+
+    st.header("Generate Post")
+
+    # Generate Button and Display
+    if st.button("Generate"):
+        post = generate_post(selected_length, selected_language, selected_tag)
+
+        # Display the generated post
+        st.write(post)
+    else:
+        st.write("Click the button above to generate a LinkedIn post.")
+
+    st.markdown("---")
+
+
+# Run the app
+if __name__ == "__main__":
+    main()

post_generator.py

@@ -0,0 +1,92 @@
+from langchain_groq import ChatGroq
+import os
+from dotenv import load_dotenv
+import pandas as pd
+import json
+
+
+class FewShotPosts:
+    def __init__(self, file_path="processed_posts.json"):
+        self.df = None
+        self.unique_tags = None
+        self.load_posts(file_path)
+
+    def load_posts(self, file_path):
+        with open(file_path, encoding="utf-8") as f:
+            posts = json.load(f)
+            self.df = pd.json_normalize(posts)
+            self.df['length'] = self.df['line_count'].apply(self.categorize_length)
+            # collect unique tags
+            all_tags = self.df['tags'].apply(lambda x: x).sum()
+            self.unique_tags = list(set(all_tags))
+
+    def get_filtered_posts(self, length, language, tag):
+        df_filtered = self.df[
+            (self.df['tags'].apply(lambda tags: tag in tags)) &  # Tags contain 'Influencer'
+            (self.df['language'] == language) &  # Language is 'English'
+            (self.df['length'] == length)  # Line count is less than 5
+        ]
+        return df_filtered.to_dict(orient='records')
+
+    def categorize_length(self, line_count):
+        if line_count < 5:
+            return "Short"
+        elif 5 <= line_count <= 10:
+            return "Medium"
+        else:
+            return "Long"
+
+    def get_tags(self):
+        return self.unique_tags
+
+
+load_dotenv()
+llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model_name="llama-3.3-70b-versatile")
+few_shot = FewShotPosts()
+
+def get_length_str(length):
+    if length == "Short":
+        return "1 to 5 lines"
+    if length == "Medium":
+        return "6 to 10 lines"
+    if length == "Long":
+        return "11 to 15 lines"
+
+
+def generate_post(length, language, tag):
+    prompt = get_prompt(length, language, tag)
+    response = llm.invoke(prompt)
+    return response.content
+
+
+def get_prompt(length, language, tag):
+    length_str = get_length_str(length)
+
+    prompt = f'''
+    Generate a LinkedIn post using the below information. No preamble.
+
+    1) Topic: {tag}
+    2) Length: {length_str}
+    3) Language: {language}
+    If Language is Hinglish then it means it is a mix of Hindi and English. 
+    The script for the generated post should always be English.
+    '''
+    # prompt = prompt.format(post_topic=tag, post_length=length_str, post_language=language)
+
+    examples = few_shot.get_filtered_posts(length, language, tag)
+
+    if len(examples) > 0:
+        prompt += "4) Use the writing style as per the following examples."
+
+    for i, post in enumerate(examples):
+        post_text = post['text']
+        prompt += f'\n\n Example {i+1}: \n\n {post_text}'
+
+        if i == 1: # Use max two samples
+            break
+
+    return prompt
+
+
+if __name__ == "__main__":
+    print(generate_post("Medium", "English", "Mental Health"))

preprocess.py

@@ -0,0 +1,85 @@
+import json
+from llm_helper import llm
+from langchain_core.prompts import PromptTemplate
+from langchain_core.output_parsers import JsonOutputParser
+from langchain_core.exceptions import OutputParserException
+
+
+def process_posts(raw_file_path, processed_file_path=None):
+    with open(raw_file_path, encoding='utf-8') as file:
+        posts = json.load(file)
+        enriched_posts = []
+        for post in posts:
+            metadata = extract_metadata(post['text'])
+            post_with_metadata = post | metadata
+            enriched_posts.append(post_with_metadata)
+
+    unified_tags = get_unified_tags(enriched_posts)
+    for post in enriched_posts:
+        current_tags = post['tags']
+        new_tags = {unified_tags[tag] for tag in current_tags}
+        post['tags'] = list(new_tags)
+
+    with open(processed_file_path, encoding='utf-8', mode="w") as outfile:
+        json.dump(enriched_posts, outfile, indent=4)
+
+
+def extract_metadata(post):
+    template = '''
+    You are given a LinkedIn post. You need to extract number of lines, language of the post and tags.
+    1. Return a valid JSON. No preamble. 
+    2. JSON object should have exactly three keys: line_count, language and tags. 
+    3. tags is an array of text tags. Extract maximum two tags.
+    4. Language should be English or Hinglish (Hinglish means hindi + english)
+    
+    Here is the actual post on which you need to perform this task:  
+    {post}
+    '''
+
+    pt = PromptTemplate.from_template(template)
+    chain = pt | llm
+    response = chain.invoke(input={"post": post})
+
+    try:
+        json_parser = JsonOutputParser()
+        res = json_parser.parse(response.content)
+    except OutputParserException:
+        raise OutputParserException("Context too big. Unable to parse jobs.")
+    return res
+
+
+def get_unified_tags(posts_with_metadata):
+    unique_tags = set()
+    # Loop through each post and extract the tags
+    for post in posts_with_metadata:
+        unique_tags.update(post['tags'])  # Add the tags to the set
+
+    unique_tags_list = ','.join(unique_tags)
+
+    template = '''I will give you a list of tags. You need to unify tags with the following requirements,
+    1. Tags are unified and merged to create a shorter list. 
+       Example 1: "Jobseekers", "Job Hunting" can be all merged into a single tag "Job Search". 
+       Example 2: "Motivation", "Inspiration", "Drive" can be mapped to "Motivation"
+       Example 3: "Personal Growth", "Personal Development", "Self Improvement" can be mapped to "Self Improvement"
+       Example 4: "Scam Alert", "Job Scam" etc. can be mapped to "Scams"
+    2. Each tag should be follow title case convention. example: "Motivation", "Job Search"
+    3. Output should be a JSON object, No preamble
+    3. Output should have mapping of original tag and the unified tag. 
+       For example: {{"Jobseekers": "Job Search",  "Job Hunting": "Job Search", "Motivation": "Motivation}}
+    
+    Here is the list of tags: 
+    {tags}
+    '''
+    pt = PromptTemplate.from_template(template)
+    chain = pt | llm
+    response = chain.invoke(input={"tags": str(unique_tags_list)})
+    try:
+        json_parser = JsonOutputParser()
+        res = json_parser.parse(response.content)
+    except OutputParserException:
+        raise OutputParserException("Context too big. Unable to parse jobs.")
+    return res
+
+
+if __name__ == "__main__":
+    process_posts("raw_posts.json", "processed_posts.json")

processed_posts.json

@@ -0,0 +1,100 @@
+[
+    {
+        "text": "Just saw a LinkedIn Influencer with 'Organic Growth' written in the profile with 65K+ followers claiming that he can help you in growing your platform, copying the posts from other influencers.",
+        "engagement": 90,
+        "line_count": 1,
+        "language": "English",
+        "tags": [
+            "Influencer",
+            "Organic Growth"
+        ]
+    },
+    {
+        "text": "Jobseekers, this one\u2019s for you.\n Every application, every interview, every follow-up\u2026 the pressure is immense.\n And I know what you're thinking: Am I not good enough? \n But let me tell you, this isn\u2019t about you or your skills. It\u2019s about a broken system where 60% of applicants never hear back. \n Your mental health is not worth sacrificing for a system that doesn\u2019t acknowledge your worth. \n Please remember, taking care of yourself is the real priority. \n Your dream job will come, but for now, breathe. \ud83c\udf3b",
+        "engagement": 347,
+        "line_count": 7,
+        "language": "English",
+        "tags": [
+            "Job Search",
+            "Mental Health"
+        ]
+    },
+    {
+        "text": "Looking for jobs on LinkedIn is like online dating: Full of promises, but in the end, you\u2019re just left ghosted.",
+        "engagement": 109,
+        "line_count": 1,
+        "language": "English",
+        "tags": [
+            "Job Search",
+            "Online Dating"
+        ]
+    },
+    {
+        "text": "LinkedIn scams be like: 'Congratulations, you've been selected for a role you didn\u2019t even apply for!' \n The catch? Pay Rs. 50,000 for the honor.",
+        "engagement": 115,
+        "line_count": 2,
+        "language": "English",
+        "tags": [
+            "Scams"
+        ]
+    },
+    {
+        "text": "sapne dekhna achi baat hai,\nlekin job ka sapna dekh ke 'interested' likhna,\nyeh toh achi baat nahi hai na?",
+        "engagement": 545,
+        "line_count": 3,
+        "language": "Hinglish",
+        "tags": [
+            "Job Search",
+            "Sapne"
+        ]
+    },
+    {
+        "text": "Next time when I'll be reading some LinkedIn Influencer's story, I am starting from the last line.\nIf there's a link attached to it, it's most probably a fake one.\nSaves me time!",
+        "engagement": 188,
+        "line_count": 3,
+        "language": "English",
+        "tags": [
+            "Productivity",
+            "Time Management"
+        ]
+    },
+    {
+        "text": "Every time I poured my heart into 5-6 rounds of interviews and faced rejection, it felt like a punch in the gut. The sleepless nights, the endless preparation, all for nothing.\n\nBut looking back, I realize it wasn\u2019t nothing. It was the Universe\u2019s way of saying, \u201cNot this one, something better is on the way.\u201d\n\nEvery single time, I\u2019ve been shown why that rejection happened.\n\nDoors I thought I wanted to walk through were shut, only to have the right ones swing open.\n\nThe kind that aligned with my growth, my values, and my happiness.\n\nAt first, it stung. It hurt deeply. But now, when things don\u2019t go as planned, I don\u2019t panic.\n\nI don\u2019t question my worth. I sit back, breathe, and trust. The Universe knows.\n\nI know there's another plan waiting. Something bigger, better, and just for me.\n\nTo anyone feeling the weight of rejection: trust that the closed doors are protecting you from something you can\u2019t see right now.\n\nYour path is being cleared for something even more beautiful.",
+        "engagement": 206,
+        "line_count": 16,
+        "language": "English",
+        "tags": [
+            "Motivation"
+        ]
+    },
+    {
+        "text": "To everyone who's still looking for a job...\n\nI see you. I feel you. \ud83d\udc94\n\nEvery rejection email feels like a punch in the gut, and every 'We'll get back to you' sounds more like 'You'll never hear from us.'\n\nBut I want you to know, you're not alone in this. \ud83c\udf38\n\nAccording to a study, 80% of jobseekers struggle with anxiety and self-doubt during their search. It's normal to feel lost, but it's not the end.\n\nTake breaks, breathe, and remember, this doesn't define you. Your worth is not tied to an offer letter. \ud83d\udca5\n\nYour mental health matters more than any job.",
+        "engagement": 899,
+        "line_count": 9,
+        "language": "English",
+        "tags": [
+            "Job Search",
+            "Mental Health"
+        ]
+    },
+    {
+        "text": "Sometimes, we forget that a company\u2019s brand name doesn\u2019t define someone\u2019s talent. It\u2019s easy to get caught up in the 'big company = big talent' mindset, but that's not always the case.\n\nI\u2019ve had the privilege of working with people from smaller companies (lesser known) who blow me away with their skills and dedication. They don\u2019t need a fancy title or a famous brand behind them to prove their worth.\n\nI've seen the other side too\u2014people in top-tier companies feeling lost, overwhelmed, or stuck, even though the world sees them as 'successful'.\n\nLet\u2019s stop attaching someone\u2019s value to the company they work for. Freshers especially need to hear this\u2014skills are what matter, not the size of the company behind them.\n\nAt the end of the day, happiness and growth don\u2019t come from a brand name, they come from doing what you love and constantly improving your craft.",
+        "engagement": 166,
+        "line_count": 11,
+        "language": "English",
+        "tags": [
+            "Self Improvement",
+            "Career Advice"
+        ]
+    },
+    {
+        "text": "So when I left a toxic work environment, I told my manager a simple thing and felt so good \ud83d\ude2f\n\nI just said-\n\n'Hope your son gets a manager like you.\nI hope that the manager behaves the same way as you did with me.\nThank you.'\n\nNow tell me 1 thing-\n\nShe always said that she was a great manager.\nWhy will she get offended?\n\nI just told her that I wish her son would get a manager like she was.\n\nIf you felt bad, then that means you were a bad manager and now you know it. \ud83e\b80\n\nIf you feel good, then take it as a blessing for your son and you'll really want someone to treat your son/daughter in the same way.\n\nShe cannot be even angry with me else it'll prove that she was not a 'great' manager.\n\nMuskan - 1\nManager - 0\n\nMuskan -> Aura +100000000\n\n(Fictional message unfortunately :(\n)\n\nHope you all become the people that your sons/daughters will like to work under \ud83d\ude4f\n\nThere are a lot of bad people/things, bring a small change and break the chain :)",
+        "engagement": 1111,
+        "line_count": 19,
+        "language": "English",
+        "tags": [
+            "Motivation",
+            "Leadership"
+        ]
+    }
+]

raw_posts.json

@@ -0,0 +1,42 @@
+[
+	{
+		"text": "Just saw a LinkedIn Influencer with 'Organic Growth' written in the profile with 65K+ followers claiming that he can help you in growing your platform, copying the posts from other influencers.",
+		"engagement": 90
+	},
+	{	
+		"text": "Jobseekers, this one’s for you.\n Every application, every interview, every follow-up… the pressure is immense.\n And I know what you're thinking: Am I not good enough? \n But let me tell you, this isn’t about you or your skills. It’s about a broken system where 60% of applicants never hear back. \n Your mental health is not worth sacrificing for a system that doesn’t acknowledge your worth. \n Please remember, taking care of yourself is the real priority. \n Your dream job will come, but for now, breathe. 🌻",
+		"engagement": 347
+	},
+	{
+		"text": "Looking for jobs on LinkedIn is like online dating: Full of promises, but in the end, you’re just left ghosted.",
+		"engagement": 109
+	},
+	{
+		"text": "LinkedIn scams be like: 'Congratulations, you've been selected for a role you didn’t even apply for!' \n The catch? Pay Rs. 50,000 for the honor.",
+		"engagement": 115
+	},
+	{
+		"text": "sapne dekhna achi baat hai,\nlekin job ka sapna dekh ke 'interested' likhna,\nyeh toh achi baat nahi hai na?",
+		"engagement": 545
+	},
+	{
+		"text": "Next time when I'll be reading some LinkedIn Influencer's story, I am starting from the last line.\nIf there's a link attached to it, it's most probably a fake one.\nSaves me time!",
+		"engagement": 188
+	},
+	{
+		"text": "Every time I poured my heart into 5-6 rounds of interviews and faced rejection, it felt like a punch in the gut. The sleepless nights, the endless preparation, all for nothing.\n\nBut looking back, I realize it wasn’t nothing. It was the Universe’s way of saying, “Not this one, something better is on the way.”\n\nEvery single time, I’ve been shown why that rejection happened.\n\nDoors I thought I wanted to walk through were shut, only to have the right ones swing open.\n\nThe kind that aligned with my growth, my values, and my happiness.\n\nAt first, it stung. It hurt deeply. But now, when things don’t go as planned, I don’t panic.\n\nI don’t question my worth. I sit back, breathe, and trust. The Universe knows.\n\nI know there's another plan waiting. Something bigger, better, and just for me.\n\nTo anyone feeling the weight of rejection: trust that the closed doors are protecting you from something you can’t see right now.\n\nYour path is being cleared for something even more beautiful.",
+		"engagement": 206
+	},
+	{
+		"text": "To everyone who's still looking for a job...\n\nI see you. I feel you. \ud83d\udc94\n\nEvery rejection email feels like a punch in the gut, and every 'We'll get back to you' sounds more like 'You'll never hear from us.'\n\nBut I want you to know, you're not alone in this. \ud83c\udf38\n\nAccording to a study, 80% of jobseekers struggle with anxiety and self-doubt during their search. It's normal to feel lost, but it's not the end.\n\nTake breaks, breathe, and remember, this doesn't define you. Your worth is not tied to an offer letter. \ud83d\udca5\n\nYour mental health matters more than any job.",
+		"engagement": 899
+	},
+	{
+		"text": "Sometimes, we forget that a company’s brand name doesn’t define someone’s talent. It’s easy to get caught up in the 'big company = big talent' mindset, but that's not always the case.\n\nI’ve had the privilege of working with people from smaller companies (lesser known) who blow me away with their skills and dedication. They don’t need a fancy title or a famous brand behind them to prove their worth.\n\nI've seen the other side too—people in top-tier companies feeling lost, overwhelmed, or stuck, even though the world sees them as 'successful'.\n\nLet’s stop attaching someone’s value to the company they work for. Freshers especially need to hear this—skills are what matter, not the size of the company behind them.\n\nAt the end of the day, happiness and growth don’t come from a brand name, they come from doing what you love and constantly improving your craft.",
+		"engagement": 166
+	},
+	{
+		"text": "So when I left a toxic work environment, I told my manager a simple thing and felt so good \ud83d\ude2f\n\nI just said-\n\n'Hope your son gets a manager like you.\nI hope that the manager behaves the same way as you did with me.\nThank you.'\n\nNow tell me 1 thing-\n\nShe always said that she was a great manager.\nWhy will she get offended?\n\nI just told her that I wish her son would get a manager like she was.\n\nIf you felt bad, then that means you were a bad manager and now you know it. \ud83e\b80\n\nIf you feel good, then take it as a blessing for your son and you'll really want someone to treat your son/daughter in the same way.\n\nShe cannot be even angry with me else it'll prove that she was not a 'great' manager.\n\nMuskan - 1\nManager - 0\n\nMuskan -> Aura +100000000\n\n(Fictional message unfortunately :(\n)\n\nHope you all become the people that your sons/daughters will like to work under \ud83d\ude4f\n\nThere are a lot of bad people/things, bring a small change and break the chain :)",
+		"engagement": 1111
+	}
+]

requirements.txt

@@ -0,0 +1,7 @@
+streamlit==1.35.0
+langchain==0.2.14
+langchain-core==0.2.39
+langchain-community==0.2.12
+langchain_groq==0.1.9
+pandas==2.0.2
+python-dotenv==1.0.1