Update post_generator.py

post_generator.py

@@ -13,79 +13,201 @@ def get_length_str(length):
 
 
 def generate_post(length, language, tag, selected_tone=None):
+    """
+    Generate a LinkedIn post based on the given length, language, topic (tag), and tone.
+    """
+    # Prepare the prompt for the LLM
     prompt = get_prompt(length, language, tag)
     response = llm.invoke(prompt)
     post_content = response.content
 
-    # Add closing statements dynamically based on tone and language
+    # 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": [
-                "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! 🔥"
-            ],
-            "Hinglish": [
-                "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! 🔥"
-            ]
+            "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": [
-                "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. 💡"
-            ],
-            "Hinglish": [
-                "Aapke vichar sunne ka intezaar hai! 👔",
-                "Chaliye milkar kuch asar karte hain. 🌐",
-                "Neeche apne vichar zaroor share karein. 📢",
-                "Aapka feedback hamare liye mahatvapurn hai. 💡"
-            ]
+            "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": [
-                "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! 😎"
-            ],
-            "Hinglish": [
-                "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! 😎"
-            ]
+            "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": [
-                "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": [
-                "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. 😊"
-            ]
+            "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:
-        if language == "Hinglish" and "Hinglish" in closing_lines[selected_tone]:
-            closing_line = closing_lines[selected_tone]["Hinglish"]
+        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:
-            closing_line = closing_lines[selected_tone]["English"]
+            topic_closing = tone_lines["default"]
 
         # Randomly select a closing line
         import random
-        post_content += f"\n\n{random.choice(closing_line)}"
+        post_content += f"\n\n{random.choice(topic_closing)}"
 
     return post_content