Update post_generator.py

post_generator.py

@@ -1,10 +1,8 @@
-import re  # For handling hashtags
 from llm_helper import llm
 from few_shot import FewShotPosts
 
 few_shot = FewShotPosts()
 
-
 def get_length_str(length):
     if length == "Short":
         return "1 to 5 lines"
@@ -41,53 +39,24 @@ def generate_closing_line(language, tag, tone):
     return response.content.strip()
 
 
-def extract_hashtags(content):
-    """
-    Extract all hashtags from the given content.
-    """
-    return re.findall(r"#\w+", content)
-
-
-def remove_hashtags(content):
-    """
-    Remove all hashtags from the given content.
-    """
-    return re.sub(r"#\w+", "", content).strip()
-
-
 def generate_post(length, language, tag, selected_tone=None):
     """
-    Generate a LinkedIn post dynamically with LLM, ensuring hashtags are added only at the end.
+    Generate a LinkedIn post dynamically with LLM including a generated closing line.
     """
-    # Generate the main content
     prompt = get_prompt(length, language, tag)
     response = llm.invoke(prompt)
-    post_content = response.content.strip()  # Clean up whitespace
+    post_content = response.content
 
-    # Extract and remove hashtags from the main content
-    hashtags = extract_hashtags(post_content)
-    post_content = remove_hashtags(post_content)
-
-    # Generate the closing line
-    closing_line = ""
+    # Generate a dynamic closing line using LLM
     if selected_tone and tag:
         try:
             closing_line = generate_closing_line(language, tag, selected_tone)
-            # Extract and remove hashtags from the closing line
-            hashtags += extract_hashtags(closing_line)
-            closing_line = remove_hashtags(closing_line).strip()
+            post_content += f"\n\n\n{closing_line}"
         except Exception as e:
-            closing_line = "Thank you for reading. Your feedback is valued! 🙌"
-
-    # Combine the cleaned main content and closing line
-    full_post = f"{post_content}\n\n{closing_line}"
-
-    # Add unique hashtags at the end of the post
-    if hashtags:
-        unique_hashtags = " ".join(set(hashtags))  # Remove duplicates
-        full_post += f"\n\n{unique_hashtags}"
+            # Fallback in case of LLM failure
+            post_content += f"\n\nThank you for reading. Your feedback is valued! 🙌"
 
-    return full_post
+    return post_content
 
 
 def get_prompt(length, language, tag):
@@ -113,4 +82,4 @@ def get_prompt(length, language, tag):
 
 
 if __name__ == "__main__":
-    print(generate_post("Medium", "English", "Mental Health"))
+    print(generate_post("Medium", "English", "Mental Health"))