Update post_generator.py

post_generator.py

@@ -12,202 +12,52 @@ def get_length_str(length):
         return "11 to 15 lines"
 
 
+def generate_closing_line(language, tag, tone):
+    """
+    Generate a closing line using the LLM based on language, tag, and tone.
+    """
+    closing_prompt = f"""
+    You are writing a LinkedIn post. Create a concise and engaging closing line.
+    - The closing line should reflect the topic: "{tag}".
+    - Use the tone/style: "{tone}".
+    - The closing line must encourage engagement or provide a call to action, relevant to the topic.
+    - Use the language: "{language}" (Hinglish means Hindi phrases written in English script).
+
+    Examples:
+    - Topic: "Job Search", Tone: "Motivational", Language: "English"
+      Closing Line: "Your dream job is closer than you think. Stay determined! 🚀"
+
+    - Topic: "Mental Health", Tone: "Professional", Language: "English"
+      Closing Line: "Your mental well-being is essential. Let’s discuss ways to manage stress. 💡"
+
+    - Topic: "Dating", Tone: "Informal", Language: "Hinglish"
+      Closing Line: "Apka perfect date idea kya hai? Neeche share karein! 😄"
+
+    Now, write a relevant closing line for the following inputs:
+    Topic: "{tag}"
+    Tone: "{tone}"
+    Language: "{language}"
+    """
+    response = llm.invoke(closing_prompt)
+    return response.content.strip()
+
+
 def generate_post(length, language, tag, selected_tone=None):
     """
-    Generate a LinkedIn post based on the given length, language, topic (tag), and tone.
+    Generate a LinkedIn post dynamically with LLM including a generated closing line.
     """
-    # Prepare the prompt for the LLM
     prompt = get_prompt(length, language, tag)
     response = llm.invoke(prompt)
     post_content = response.content
 
-    # Fetch all available topics dynamically from FewShotPosts
-    fs = FewShotPosts()
-    all_topics = [topic.lower() for topic in fs.get_tags()]
-
-    # Closing lines structure
-    closing_lines = {
-        "Motivational": {
-            "English": {
-                "default": [
-                    "Stay inspired and keep moving forward! 💪",
-                    "Your journey to greatness begins today. 🌟",
-                    "Remember, small steps lead to big achievements. 🚀",
-                    "Keep pushing your boundaries. You’ve got this! 🔥"
-                ],
-                "dating": [
-                    "Love yourself first; the rest will follow! 💖",
-                    "Every journey begins with a spark. Keep believing! ✨",
-                    "Trust the process, and love will find its way. 💕",
-                    "Stay true to yourself, and the right person will connect. 🌸"
-                ],
-                "mental health": [
-                    "Your mental health matters. Take care of yourself! 💚",
-                    "Small steps lead to great mental wellness. 🌿",
-                    "Prioritize self-care; it’s your greatest investment. 🌼",
-                    "Remember, it’s okay to seek help when needed. 🌟"
-                ],
-                "job search": [
-                    "Your dream job is closer than you think. Keep going! 🚀",
-                    "Stay focused and determined; success is near. 🌟",
-                    "Every application is a step toward your goal. ✨",
-                    "Believe in your skills; the right opportunity awaits! 💼"
-                ],
-                "technology": [
-                    "Innovation starts with curiosity. Explore! 💡",
-                    "Stay updated and keep learning in this ever-changing field. 📱",
-                    "Technology drives the future. Be part of it! 🚀",
-                    "Every breakthrough starts with an idea. What’s yours? 🌟"
-                ],
-                "wellness": [
-                    "Health is wealth. Invest in yourself! 💪",
-                    "Take time to nurture your body and soul. 🌸",
-                    "Wellness is a journey, not a destination. 🌿",
-                    "Small habits lead to a healthier, happier you. 🌼"
-                ]
-            },
-            "Hinglish": {
-                "default": [
-                    "Dhire dhire aage badho aur safalta pao! 💪",
-                    "Sapne bade rakho, unhe pura karo. 🌟",
-                    "Chhoti chhoti koshis se bade lakshya hasil hote hain! 🚀",
-                    "Apne boundaries todho, aap kar sakte ho! 🔥"
-                ],
-                "dating": [
-                    "Khud se pyaar karo, baaki sab aasan hoga! 💖",
-                    "Har safar ek chhoti chamak se shuru hoti hai! ✨",
-                    "Apne upar bharosa rakho, pyaar apna rasta dhund lega. 💕",
-                    "Apne asli roop mein raho, sahi insaan connect karega. 🌸"
-                ],
-                "mental health": [
-                    "Aapki mental health mahatvapurn hai. Apna dhyan rakhein! 💚",
-                    "Chhoti chhoti koshis se acchi sehat milti hai. 🌿",
-                    "Self-care ko prioritize karein; yeh aapka sabse bada asset hai. 🌼",
-                    "Yaad rakhein, zarurat padne par madad lena bilkul sahi hai. 🌟"
-                ],
-                "job search": [
-                    "Aapka sapno ka naukri aapke kareeb hai. Lage raho! 🚀",
-                    "Focus banaye rakhein; safalta paas hai. 🌟",
-                    "Har application aapke goal ki taraf ek kadam hai. ✨",
-                    "Apni skills par bharosa rakhein; sahi opportunity intezaar kar rahi hai! 💼"
-                ],
-                "technology": [
-                    "Innovation curiosity se shuru hoti hai. Khojiye! 💡",
-                    "Stay updated, ye field hamesha badal rahi hai. 📱",
-                    "Technology future ko chalati hai. Hissa baniye! 🚀",
-                    "Har breakthrough ek idea se shuru hota hai. Aapka kya hai? 🌟"
-                ],
-                "wellness": [
-                    "Health hi wealth hai. Khud par invest karein! 💪",
-                    "Apne body aur soul ko nurture karne ka samay nikalien. 🌸",
-                    "Wellness ek journey hai, destination nahi. 🌿",
-                    "Chhoti aadatein ek healthy aur happy aap banati hain. 🌼"
-                ]
-            }
-        },
-        "Professional": {
-            "English": {
-                "default": [
-                    "Looking forward to hearing your thoughts! 👔",
-                    "Let’s collaborate and make an impact together. 🌐",
-                    "Feel free to share your perspective below. 📢",
-                    "Insights and feedback are always welcome. 💡"
-                ],
-                "job search": [
-                    "Networking is key to success. Let’s connect! 👔",
-                    "Stay prepared and open to opportunities. 🌐",
-                    "Professional growth starts with the right connections. 📈",
-                    "Every step counts in your career journey. Keep moving! 💼"
-                ],
-                "technology": [
-                    "Technology thrives on collaboration. Let’s innovate together! 💻",
-                    "Stay curious and keep challenging the status quo. 🌐",
-                    "Your insights can drive the next big breakthrough. Share below! 💡",
-                    "Adaptation and growth go hand in hand in tech. Let’s excel! 🚀"
-                ]
-            },
-            "Hinglish": {
-                "default": [
-                    "Aapke vichar sunne ka intezaar hai! 👔",
-                    "Chaliye milkar kuch asar karte hain. 🌐",
-                    "Neeche apne vichar zaroor share karein. 📢",
-                    "Aapka feedback hamare liye mahatvapurn hai. 💡"
-                ],
-                "job search": [
-                    "Networking safalta ki kunji hai. Chaliye judte hain! 👔",
-                    "Tayyar rahiye aur naye opportunities ke liye khule rahiye. 🌐",
-                    "Professional growth sahi connections se shuru hoti hai. 📈",
-                    "Aapke career journey mein har kadam mahatvapurn hai. Lage raho! 💼"
-                ],
-                "technology": [
-                    "Technology collaboration se phalti phoolti hai. Milkar innovate karein! 💻",
-                    "Curiosity banaye rakhein aur naye challenges ko accept karein. 🌐",
-                    "Aapke insights agla bada breakthrough laa sakte hain. Neeche share karein! 💡",
-                    "Tech mein adaptation aur growth saath saath chalte hain. Chaliye excellence achieve karein! 🚀"
-                ]
-            }
-        },
-        "Informal": {
-            "English": {
-                "default": [
-                    "What do you think? Let's chat in the comments! 😄",
-                    "Drop your thoughts below, would love to hear from you! 🗨️",
-                    "Let’s keep the conversation rolling. Share your views! 🚀",
-                    "Got something to add? Don’t hold back! 😎"
-                ],
-                "dating": [
-                    "What’s your idea of a perfect date? Let’s talk! 😍",
-                    "Got a dating tip? Share it below! 👫",
-                    "Dating stories? Share your fun experiences! ❤️",
-                    "Let’s chat about love and connections! 💕"
-                ]
-            },
-            "Hinglish": {
-                "default": [
-                    "Kya sochte ho? Comments mein baat karte hain! 😄",
-                    "Apne vichar neeche likho, sunne ke liye excited hoon! 🗨️",
-                    "Charcha ko jaari rakhein, apne vichar share karein! 🚀",
-                    "Kuch add karna hai? Sharmana mat, batao! 😎"
-                ],
-                "dating": [
-                    "Aapke liye perfect date kya hai? Bataiye! 😍",
-                    "Koi dating tip hai? Share karein! 👫",
-                    "Dating stories? Apne mazedar anubhav share karein! ❤️",
-                    "Chaliye pyaar aur connection ke baare mein baat karein! 💕"
-                ]
-            }
-        },
-        "Neutral": {
-            "English": {
-                "default": [
-                    "Thank you for reading. Your feedback is valued! 🙌",
-                    "Let’s connect and share ideas. 🌟",
-                    "Looking forward to engaging with you in the comments. 💬",
-                    "Your thoughts are always appreciated. 😊"
-                ]
-            },
-            "Hinglish": {
-                "default": [
-                    "Padhne ke liye dhanyavaad. Aapka feedback mahatvapurn hai! 🙌",
-                    "Chaliye judein aur naye ideas share karein. 🌟",
-                    "Aapke vichar comments mein padhne ka intezaar hai. 💬",
-                    "Aapke sujhav hamesha sarankrit hote hain. 😊"
-                ]
-            }
-        }
-    }
-
-    # Determine the appropriate closing lines
-    if selected_tone and selected_tone in closing_lines:
-        tone_lines = closing_lines[selected_tone][language]
-        if tag.lower() in all_topics and tag.lower() in tone_lines:
-            topic_closing = tone_lines[tag.lower()]
-        else:
-            topic_closing = tone_lines["default"]
-
-        # Randomly select a closing line
-        import random
-        post_content += f"\n\n{random.choice(topic_closing)}"
+    # Generate a dynamic closing line using LLM
+    if selected_tone and tag:
+        try:
+            closing_line = generate_closing_line(language, tag, selected_tone)
+            post_content += f"\n\n{closing_line}"
+        except Exception as e:
+            # Fallback in case of LLM failure
+            post_content += f"\n\nThank you for reading. Your feedback is valued! 🙌"
 
     return post_content