Update post_generator.py

post_generator.py

@@ -13,10 +13,46 @@ def get_length_str(length):
         return "11 to 15 lines"
 
 
-def generate_post(length, language, tag):
+def generate_post(length, language, tag, selected_tone=None):
     prompt = get_prompt(length, language, tag)
     response = llm.invoke(prompt)
-    return response.content
+    post_content = response.content
+
+    # Add closing statements dynamically based on tone
+    closing_lines = {
+        "Motivational": [
+            "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! 🔥"
+        ],
+        "Professional": [
+            "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. 💡"
+        ],
+        "Informal": [
+            "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! 😎"
+        ],
+        "Neutral": [
+            "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. 😊"
+        ]
+    }
+
+    if selected_tone and selected_tone in closing_lines:
+        # Randomly select a closing line from the appropriate tone category
+        import random
+        post_content += f"\n\n{random.choice(closing_lines[selected_tone])}"
+
+    return post_content
+
 
 
 def get_prompt(length, language, tag):