Update post_generator.py

post_generator.py

@@ -1,9 +1,6 @@
-import few_shot
 from llm_helper import llm
 from few_shot import FewShotPosts
 
-
-
 def get_length_str(length):
     if length == "Short":
         return "1 to 5 lines"
@@ -12,15 +9,20 @@ def get_length_str(length):
     if length == "Long":
         return "11 to 15 lines"
 
-
 def generate_post(persona_name, length, language, tag):
+    """Generate a LinkedIn post based on persona and given criteria."""
+    
+    # Create an instance of FewShotPosts with the persona name
     few_shot = FewShotPosts(persona_name)
-    prompt = get_prompt(length, language, tag)
+
+    # Pass the instance to get_prompt
+    prompt = get_prompt(few_shot, length, language, tag)
     response = llm.invoke(prompt)
     return response.content
 
-
-def get_prompt(length, language, tag):
+def get_prompt(few_shot_instance, length, language, tag):
+    """Generate the prompt for LLM based on few-shot examples."""
+    
     length_str = get_length_str(length)
 
     prompt = f'''
@@ -31,9 +33,9 @@ def get_prompt(length, language, tag):
     3) Language: {language} 
     The script for the generated post should always be English.
     '''
-    # prompt = prompt.format(post_topic=tag, post_length=length_str, post_language=language)
 
-    examples = few_shot.get_filtered_posts(length, language, tag)
+    # ✅ Fix: Call get_filtered_posts() on few_shot_instance, not module `few_shot`
+    examples = few_shot_instance.get_filtered_posts(length, language, tag)
 
     if len(examples) > 0:
         prompt += "4) Use the writing style as per the following examples."
@@ -42,11 +44,7 @@ def get_prompt(length, language, tag):
         post_text = post['text_blocks']
         prompt += f'\n\n Example {i+1}: \n\n {post_text}'
 
-        if i == 1: # Use max two samples
+        if i == 1:  # Use max two samples
             break
 
     return prompt
-
-
-if __name__ == "__main__":
-    print(generate_post("Medium", "English", "Economy"))