Update post_generator.py

post_generator.py

@@ -18,43 +18,79 @@ def generate_post(length, language, tag, selected_tone=None):
     response = llm.invoke(prompt)
     post_content = response.content
 
-    # Add closing statements dynamically based on tone
+    # Add closing statements dynamically based on tone and language
     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. 😊"
-        ]
+        "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! 🔥"
+            ]
+        },
+        "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. 💡"
+            ]
+        },
+        "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! 😎"
+            ]
+        },
+        "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. 😊"
+            ]
+        }
     }
 
     if selected_tone and selected_tone in closing_lines:
-        # Randomly select a closing line from the appropriate tone category
+        if language == "Hinglish" and "Hinglish" in closing_lines[selected_tone]:
+            closing_line = closing_lines[selected_tone]["Hinglish"]
+        else:
+            closing_line = closing_lines[selected_tone]["English"]
+
+        # Randomly select a closing line
         import random
-        post_content += f"\n\n{random.choice(closing_lines[selected_tone])}"
+        post_content += f"\n\n{random.choice(closing_line)}"
 
     return post_content
 
 
-
 def get_prompt(length, language, tag):
     length_str = get_length_str(length)