Update post_generator.py

post_generator.py

@@ -45,13 +45,14 @@ def generate_post(length, language, tag, selected_tone=None):
     """
     prompt = get_prompt(length, language, tag)
     response = llm.invoke(prompt)
-    post_content = response.content
+    post_content = response.content.strip()
 
     # 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}"  # Append the closing line with hashtags
+            # Append the closing line with hashtags
+            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! 🙌"
@@ -68,6 +69,7 @@ def get_prompt(length, language, tag):
     2. Post Length: "{length_str}"
     3. Language: "{language}" (Hinglish means Hindi phrases written in English script).
     4. Incorporate creativity, enthusiasm, emotional appeal, and actionable advice.
+    5. Do not include hashtags in the main content.
     '''
 
     examples = few_shot.get_filtered_posts(length, language, tag)
@@ -75,7 +77,9 @@ def get_prompt(length, language, tag):
         prompt += "\nExamples of great posts:\n"
         for i, post in enumerate(examples[:2]):  # Limit to 2 examples
             post_text = post['text']
-            prompt += f"Example {i + 1}: {post_text}\n"
+            # Remove hashtags from examples
+            cleaned_post_text = " ".join(word for word in post_text.split() if not word.startswith("#"))
+            prompt += f"Example {i + 1}: {cleaned_post_text}\n"
     
     prompt += "\nNow write the post."
     return prompt