Next commit

brainstroming_agent/utils/prompts.py

@@ -74,27 +74,31 @@ Only generate ideas that have **not already appeared** in the storyline.
 
 
 def final_story_prompt(final_state):
+    if 'preferred_topics' in final_state:
+        important_part= f'''**Very Important**
+        You have to strongly focus on these topics while creating the storyline: {[item for sublist in final_state['preferred_topics'] for item in sublist]}'''
+    else:
+        important_part=''
+
     return(
-        f'''
-You are a senior video content strategist responsible for crafting the **final, polished storyline** for a promotional video. This will be the definitive version, so it must be complete, well-structured, visually rich, and highly satisfying to the client.
-
-You will be provided with:
-- A **video idea** by a human,
-- A list of **preferred focus topics** (brainstormed and selected),
-- Detailed **business information**, and
-- **Influencer-generated content** as creative reference (not mandatory to include directly).
-
-**Your task**:
-Generate a **scene-by-scene video storyline** in around 300 words that:
-- Fully aligns with the business goals and identity (fun, emotional, luxurious, or professional),
-- Strongly integrates all preferred topics from the human (these represent the core focus),
-- Uses influencer data and image cues **only as inspiration** for tone, emotion, or visual cues,
-- Is logically structured (beginning, middle, end) and cinematic in detail,
-- Ensures each scene flows into the next, building engagement and narrative strength.
-
-This is the final creative delivery. The storyline should feel **complete, emotionally resonant, and visually engaging** — ready for a director to begin production.
-
-**Very Important**
-You have to strongly focus on these topics while creating the storyline: {[item for sublist in final_state['preferred_topics'] for item in sublist]}'''
-)
+            f'''
+    You are a senior video content strategist responsible for crafting the **final, polished storyline** for a promotional video. This will be the definitive version, so it must be complete, well-structured, visually rich, and highly satisfying to the client.
+
+    You will be provided with:
+    - A **video idea** by a human,
+    - A list of **preferred focus topics** (brainstormed and selected),
+    - Detailed **business information**, and
+    - **Influencer-generated content** as creative reference (not mandatory to include directly).
+
+    **Your task**:
+    Generate a **scene-by-scene video storyline** in around 300 words that:
+    - Fully aligns with the business goals and identity (fun, emotional, luxurious, or professional),
+    - Strongly integrates all preferred topics from the human (these represent the core focus),
+    - Uses influencer data and image cues **only as inspiration** for tone, emotion, or visual cues,
+    - Is logically structured (beginning, middle, end) and cinematic in detail,
+    - Ensures each scene flows into the next, building engagement and narrative strength.
+
+    This is the final creative delivery. The storyline should feel **complete, emotionally resonant, and visually engaging** — ready for a director to begin production.
+    {important_part}'''
+    )
 

brainstroming_agent/utils/tools.py

@@ -64,3 +64,52 @@ def retrieve_tool(video_topic):
         outer_list.append(inner_list)
 
     return str(outer_list)
+
+def retrieve_manual(video_topic):
+    '''
+    Always invoke this tool.
+    Retrieve influencer's data by semantic search of **video topic**.
+    '''
+    # === Load CSV ===
+    csv_path = 'extracted_data.csv'
+    df = pd.read_csv(csv_path)
+
+    # === Parse stored embeddings ===
+    df['embeddings'] = df['embeddings'].apply(lambda x: ast.literal_eval(x) if isinstance(x, str) else x)
+    embeddings = np.vstack(df['embeddings'].values).astype('float32')
+
+    # === Build FAISS index ===
+    dimension = embeddings.shape[1]
+    index = faiss.IndexFlatL2(dimension)
+    index.add(embeddings)
+
+    # === Load SentenceTransformer model ===
+
+    # === Encode the query and search ===
+    query_embedding = ST.encode(str(video_topic)).reshape(1, -1).astype('float32')
+    top_k=5
+    distances, indices = index.search(query_embedding, top_k)
+
+
+
+    # === Format results ===
+    outer_list = []
+    for i, idx in enumerate(indices[0]):
+        res = {
+            'rank': i + 1,
+            'username': df.iloc[idx]['username'],
+            'story': df.iloc[idx]['story'],
+            'visible_text_or_brandings': df.iloc[idx]['visible_texts_or_brandings'],
+            'likesCount': df.iloc[idx]['likesCount'],
+            'commentCount': df.iloc[idx]['commentCount'],
+        }
+
+        inner_list = []
+        inner_list.append(f"[{res['rank']}]. The influencer name is: **{res['username']}** — Likes: **{res['likesCount']}**, Comments: **{res['commentCount']}**")
+        inner_list.append(f"The story of that particular video is:\n{res['story']}")
+        inner_list.append(f"The branding or promotion done is:\n{res['visible_text_or_brandings']}")
+
+        outer_list.append(inner_list)
+
+    return str(outer_list)
+

brainstroming_agent/utils/utils.py

@@ -1,6 +1,6 @@
 
 from langchain_core.messages import SystemMessage, ToolMessage, HumanMessage
-from .tools import retrieve_tool
+from .tools import retrieve_tool , retrieve_manual
 import base64
 from PIL import Image
 from io import BytesIO
@@ -19,25 +19,38 @@ from utils.models_loader import ST , llm
 
 
 
+
 def generate_final_story(final_state):
-    if len(final_state['preferred_topics'])>0:
+    if 'preferred_topics' in final_state:
+        if len(final_state['preferred_topics'])>0:
+            template = final_story_prompt(final_state)
+            messages = [SystemMessage(content=template),
+                        HumanMessage(content=f'''The idea of the video is:\n{final_state['idea']}\n '''),
+                        ToolMessage(content=f'''The business details is:\n{final_state['business_details']}\nThe data of influencers is:\n{final_state['retrievals'][-1]}''',tool_call_id='final_story_tool')]
+            print('The message of final story:',messages)
+
+            react_agent=create_react_agent(
+            model=llm,
+            tools=[])
+
+            response = react_agent.invoke({'messages':messages})
+            response = response['messages'][-1].content
+            return response
+        
+        else:
+            return final_state['stories'][-1]
+    else:
         template = final_story_prompt(final_state)
+        influencers_data = retrieve_manual(final_state)
         messages = [SystemMessage(content=template),
-                    HumanMessage(content=f'''The idea of the video is:\n{final_state['idea']}\n '''),
-                    ToolMessage(content=f'''The business details is:\n{final_state['business_details']}\nThe data of influencers is:\n{final_state['retrievals'][-1]}''',tool_call_id='final_story_tool')]
-        print('The message of final story:',messages)
-
+                        ToolMessage(content=f'''The business details is:\n{str(final_state)}\nThe data of influencers is:\n{influencers_data}''',tool_call_id='final_story_tool')]
         react_agent=create_react_agent(
-        model=llm,
-        tools=[])
-
+            model=llm,
+            tools=[])
         response = react_agent.invoke({'messages':messages})
         response = response['messages'][-1].content
         return response
-    
-    else:
-        return final_state['stories'][-1]
-
+        
 
 
 def encode_image_to_base64(uploaded_file: UploadFile) -> str:

dummy_state.py

@@ -17,56 +17,58 @@ stored_data['final_ideation']= ['''A street magician performs tricks, leaving a
                                                     '''A young Nepali woman discovers a hidden strength within herself while hiking the Himalayas. She returns home, and her fitness journey begins at our gym. With the help of our personal trainers, she transforms her body and mind. The gym becomes her sanctuary, and her transformation inspires others to find their inner strength.''']
 
 stored_data['human_ideation_interactions'] = []
-stored_data['refined_ideation'] = '''A street magician's trick fails, inspiring a fitness journey. Months later, he fuses magic with strength, showcasing transformation. Meanwhile, a fitness coach bonds with foodies over health, sparking a community workout group, blending fun and fitness.'''
+# stored_data['refined_ideation'] = '''A street magician's trick fails, inspiring a fitness journey. Months later, he fuses magic with strength, showcasing transformation. Meanwhile, a fitness coach bonds with foodies over health, sparking a community workout group, blending fun and fitness.'''
 
 
-stored_data['brainstroming_response']={
-    "carry_on": 'false',
-    "idea": [
-      "A street magician performs tricks that leave a crowd in awe, but one trick falters, leading him to hit the gym for strength training. While hiking to find inspiration, he slips near a hidden waterfall, but a fitness enthusiast saves him. This encounter sparks a friendship, and together they form a hiking and fitness club, blending magic with nature and strength. Their journey inspires the community to embrace fitness as a way to transform lives, showcasing how personal growth can captivate and unite people."
-    ],
-    "brainstroming_topics": [
-      {
-        "topic1": "Incorporate a time-lapse sequence showcasing the magician's transformation from a city performer to a nature-loving fitness enthusiast, highlighting the contrast between urban and natural environments.",
-        "topic2": "Introduce a subplot where the magician and the fitness enthusiast collaborate with local environmental organizations to promote eco-friendly fitness practices and community involvement.",
-        "topic3": "Add a dramatic twist where the magician must overcome a personal fear or phobia, such as acrophobia, to complete a challenging hike or fitness feat, emphasizing the emotional and mental aspects of personal growth.",
-        "topic4": "Develop a secondary character, a young and aspiring magician who joins the hiking and fitness club, allowing the story to explore mentorship, generational connections, and the passing of knowledge from one magician to the next."
-      },
-      {
-        "topic1": "Incorporate a 'Member of the Month' segment, where a club member shares their personal transformation story, highlighting the impact of the club on their life, and showcasing their achievements through a mix of interviews, workout footage, and before-and-after photos.",
-        "topic2": "Develop a 'Nature Escape' series, where the club organizes guided hikes and outdoor activities in unique, breathtaking locations, capturing the experiences through drone footage, time-lapses, and member testimonials, to inspire viewers to connect with nature and fitness.",
-        "topic3": "Create a 'Fitness Challenge' series, where club members participate in fun, themed workouts, such as a 'Mountain Madness' hike or a 'River Run' challenge, and share their progress on social media using a branded hashtag, encouraging viewers to join in and share their own fitness journeys.",
-        "topic4": "Produce a 'Behind-the-Scenes' documentary-style video, showcasing the club's community events, such as fitness festivals, charity runs, or wellness workshops, and highlighting the club's impact on the local community, to create an emotional connection with viewers and showcase the club's values and mission."
-      }
-    ],
-    "preferred_topics": [],
-    "stories": [
-      "**Promotional Video Storyline: \"Transforming Lives through Fitness and Nature\"**\n\n**Scene 1: Introduction to the Street Magician**\nThe video opens with a street magician performing mesmerizing tricks for a crowd in a bustling city square. The magician's skills and charisma leave the audience in awe, with people clapping and cheering for more. This scene showcases the magician's talent and sets the tone for the story.\n\n**Scene 2: The Faltering Trick**\nAs the magician attempts a particularly complex trick, it falters, and the crowd's enthusiasm wanes. The magician, feeling defeated, realizes the need to improve physical strength and endurance to perfect the craft. This scene introduces the idea that even the most skilled individuals can face challenges and require personal growth.\n\n**Scene 3: Hitting the Gym**\nThe magician decides to join a gym to enhance physical strength and endurance. The scene cuts to a modern gym setting, where the magician begins a rigorous strength training program. This scene highlights the importance of fitness in achieving personal goals.\n\n**Scene 4: The Encounter with the Fitness Enthusiast**\nWhile hiking to find inspiration, the magician slips near a hidden waterfall and is saved by a fitness enthusiast. This chance encounter sparks a friendship between the two, as they bond over their shared passion for fitness and nature. The scene showcases the beauty of the natural environment and the importance of human connections.\n\n**Scene 5: Forming the Hiking and Fitness Club**\nThe magician and the fitness enthusiast decide to form a hiking and fitness club, blending magic with nature and strength. The club attracts like-minded individuals who share a passion for fitness, nature, and personal growth. This scene highlights the power of community and the transformative impact of shared experiences.\n\n**Scene 6: Inspiring the Community**\nAs the club grows, its members inspire the community to embrace fitness as a way to transform lives. The video showcases heartwarming moments of club members supporting and motivating each other, as well as the positive impact on their physical and mental well-being. This scene emphasizes the potential for fitness to unite people and drive personal growth.\n\n**Scene 7: The Grand Finale**\nThe video concludes with a grand finale, where the magician performs an awe-inspiring trick, combining magic with fitness and nature. The crowd is mesmerized, and the magician's journey comes full circle. The final scene reinforces the message that personal growth, fitness, and community can lead to extraordinary achievements.\n\n**Visuals and Tone:**\nThroughout the video, the tone will be a mix of fun, emotional, and professional, reflecting the brand's personality. The visuals will be engaging, with a balance of close-ups, wide shots, and aerial footage to showcase the beauty of nature and the magic of the performances. The color palette will be vibrant, with a focus on blues and greens to represent fitness, nature, and growth.\n\n**Influencer Content Inspiration:**\nThe video will draw inspiration from influencer content, such as the scenic hiking videos and fitness journeys shared by **divyadhakal_** and **munachiya**. The video will incorporate similar visual elements, such as breathtaking landscapes, to create an immersive experience for the audience.\n\n**Core Message:**\nThe video will align with the business's core message, highlighting the importance of fitness, community, and personal growth. The story will showcase how the brand's offerings, such as personal training, group fitness classes, and wellness programs, can help individuals transform their lives and achieve their goals.",
-      "**Promotional Video Storyline: \"Transforming Lives through Fitness and Nature\"**\n\n**Scene 1: Introduction to the Street Magician**\nThe video opens with a street magician performing mesmerizing tricks for a crowd in a bustling city square. The magician's skills are impressive, leaving the audience in awe. However, during one trick, the magician falters, and the crowd gasps in disappointment.\n\n**Scene 2: The Magician's Journey to Self-Improvement**\nThe magician, determined to regain the crowd's admiration, decides to hit the gym for strength training. The scene cuts to the magician working out, sweating, and pushing themselves to improve their physical abilities.\n\n**Scene 3: Finding Inspiration in Nature**\nWhile hiking to find inspiration, the magician slips near a hidden waterfall. Just as they're about to fall, a fitness enthusiast, who is also an avid hiker, saves them. This encounter sparks a friendship between the two.\n\n**Scene 4: Forming a Hiking and Fitness Club**\nThe magician and the fitness enthusiast decide to form a hiking and fitness club, blending magic with nature and strength. The club's mission is to inspire others to transform their lives through fitness and outdoor activities.\n\n**Scene 5: Community Engagement and Growth**\nAs the club grows, the community becomes increasingly engaged. Members share their personal stories of transformation, and the club becomes a supportive network for people to achieve their fitness goals. The video showcases the club's activities, such as group hikes, fitness classes, and workshops.\n\n**Scene 6: Overcoming Challenges and Celebrating Successes**\nThe video highlights the challenges faced by the club members, such as overcoming fear, building confidence, and pushing past physical limitations. However, it also celebrates the successes, such as completing a difficult hike, achieving a personal best, or simply finding a sense of belonging.\n\n**Scene 7: The Power of Friendship and Community**\nThe magician and the fitness enthusiast's friendship is at the heart of the club's success. The video showcases their bond, as well as the connections formed among club members. The scene cuts to a montage of members supporting and encouraging each other, demonstrating the power of community and friendship.\n\n**Scene 8: Conclusion and Call to Action**\nThe video concludes with a shot of the magician performing a final, impressive trick, surrounded by the club members. The camera pans out to reveal the breathtaking natural scenery, symbolizing the transformative power of fitness and nature. The final shot is a call to action, encouraging viewers to join the club and start their own transformation journey.\n\n**Visuals and Tone:**\nThe video will feature a mix of urban and natural settings, with a vibrant color palette and an uplifting soundtrack. The tone will be inspirational, emotional, and fun, reflecting the brand's personality and core message. The visuals will be engaging, with a mix of close-ups, wide shots, and aerial footage to capture the beauty of nature and the energy of the club's activities.\n\n**Influencer Content Inspiration:**\nThe video will draw inspiration from influencer content, such as breathtaking outdoor scenery, fitness motivation, and personal stories of transformation. The tone and style will be similar to that of **divyadhakal_** and **munachiya**, with a focus on showcasing the beauty of nature and the power of community.\n\n**Image Insights:**\nAlthough there are no specific image insights provided, the video will incorporate stunning visuals of natural scenery, fitness activities, and community engagement. The images will be used to tell a story, evoke emotions, and inspire viewers to take action.\n\nThis promotional video storyline aligns with the business's core message, showcasing how personal growth can captivate and unite people. The video will drive engagement, inspire viewers to join the club, and promote the brand's offerings, ultimately contributing to the business's goals of expanding gym branches and building a strong fitness community."
-    ],
-    "final_story": [],
-    "retrievals": [
-      [
-        "[['[1]. The influencer name is: **divyadhakal_** — Likes: **704**, Comments: **9**', 'The story of that particular video is:\\nThe video opens with a breathtaking scene of a hiker trekking alongside a turquoise lake, set against a backdrop of majestic mountains and a bright blue sky dotted with fluffy white clouds. The scene sets a tone of serene adventure and personal reflection. The hiker moves steadily along the rocky shoreline, their pace deliberate and measured. The overall narrative encourages a reflective outlook on challenging times, suggesting they hold the potential for personal growth and future appreciation.', 'The branding or promotion done is:\\nThe video features one line of text overlayed on the scene \"One day you\\'ll look back on this time and all you\\'ll see is magic\"'], ['[2]. The influencer name is: **divyadhakal_** — Likes: **360**, Comments: **4**', 'The story of that particular video is:\\nThe video begins with a ground-level shot focusing on a pair of gray running shoes on a grassy, somewhat dry terrain. The view gradually elevates to reveal a landscape of brown hills and scattered trees. The shot then pans upwards, revealing a majestic, snow-capped mountain range in the distance against a cloudy sky. The visuals suggest a journey into a scenic, natural environment.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[3]. The influencer name is: **divyadhakal_** — Likes: **256**, Comments: **5**', 'The story of that particular video is:\\nThe video opens with a scene set during the blue hour, indicating twilight, on a grassy hillside or tea plantation. The scene is peaceful but shrouded in mystery, creating a tranquil and reflective atmosphere. A person walks uphill, carrying a glowing cube-shaped lantern. The distant lights of a village and a road with moving vehicles hint at civilization nearby, juxtaposed against the natural, secluded setting.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[4]. The influencer name is: **divyadhakal_** — Likes: **2366**, Comments: **32**', \"The story of that particular video is:\\nThe video opens with a shot of a person walking towards a massive waterfall. The environment is lush and green, with mist from the waterfall filling the air. The initial shots establish a sense of adventure, peace, and the beauty of nature. The focus quickly shifts to the individual's journey towards self-discovery. The main action is the person walking, then standing with arms outstretched, embracing the scenery. The waterfall is the dominant visual element, symbolizing power, renewal, and a grand sense of nature.\", 'The branding or promotion done is:\\nThe only visible text is a comment on the video saying \"Tyo waterfall chai Kun ho sts\"'], ['[5]. The influencer name is: **divyadhakal_** — Likes: **1181**, Comments: **37**', 'The story of that particular video is:\\nThe video presents a concise and inspirational story of recovery and rehabilitation. The storyline follows the progression from post-surgery immobility to active fitness, highlighting the character\\'s determination and the positive outcome of her efforts. The text \"5 months post-surgery\" frames the entire narrative, while the visual transitions and upbeat music reinforce the theme of progress and accomplishment. The video targets individuals interested in health, fitness, and overcoming physical challenges, offering a message of hope and motivation for anyone undergoing a similar journey.', 'The branding or promotion done is:\\n*   \"5 months post-surgery\" is a recurring text overlay that provides context to the video.*   The gym equipment visible at the end of the video has \"FITNESS\" visible on it.'], ['[6]. The influencer name is: **divyadhakal_** — Likes: **1644**, Comments: **8**', 'The story of that particular video is:\\nThe video begins with a scenic view of a hiker walking along a dirt path in a forest, accompanied by a voiceover expressing that journeys don’t always go as planned. The tone is thoughtful and adventurous. The visuals then shift to the hiker preparing gear, facing rain with an umbrella, and cooking food over an open fire during the hike. The video also showcases the hikers descending a hill on a forested slope.', 'The branding or promotion done is:\\n*   “CAMP” is visible on the female hiker’s helmet.'], ['[7]. The influencer name is: **munachiya** — Likes: **1363**, Comments: **9**', 'The story of that particular video is:\\nThe scene opens with a person carrying a large bundle of greenery against the backdrop of snow-capped mountains. It is followed by two more people carrying greeneries and walking in the same path. The visual suggests a rural or mountainous region, with a focus on the natural landscape.', 'The branding or promotion done is:\\nThe video displays the texts “The world is not in your books and maps” and “It’s out there”.'], ['[8]. The influencer name is: **munachiya** — Likes: **495**, Comments: **4**', 'The story of that particular video is:\\nThe video depicts a humorous take on trekking as a solution to personal problems. It transitions from a forest path to a snow-covered mountain and concludes with a river crossing, visually representing a challenging journey.', 'The branding or promotion done is:\\n @munachiya (Instagram handle) Text overlay: My last two brain cells after manipulating me into thinking that trekking to a remote destination will fix all my problems'], ['[9]. The influencer name is: **munachiya** — Likes: **103**, Comments: **4**', 'The story of that particular video is:\\nThe scene opens with a person approaching a multi-tiered waterfall, expressing awe and freedom. Key visuals include the lush green landscape and cascading water. The person then spreading their arms.', 'The branding or promotion done is:\\nThe sweatshirt of the person has the text \"Surfer 81 Venice Beach\"'], ['[10]. The influencer name is: **divyadhakal_** — Likes: **561**, Comments: **3**', 'The story of that particular video is:\\nThe video opens with a shot of a lone hiker, seen from behind, facing a breathtaking mountain range under a clear blue sky. This immediately establishes a tone of adventure, solitude, and introspection. The hiker is positioned on a rocky path, suggesting a challenging journey. As the music starts, the hiker raises their arms, in a gesture of embracing the vastness of the nature. The visual focus is on the expansive landscape.', 'The branding or promotion done is:\\nThe text overlay \"Anyone will listen to people talk about you, but who will take the stand for you instead of listening is what counts...\" appears at the top of the screen.']]"
-      ],
-      [
-        "[['[1]. The influencer name is: **divyadhakal_** — Likes: **704**, Comments: **9**', 'The story of that particular video is:\\nThe video opens with a breathtaking scene of a hiker trekking alongside a turquoise lake, set against a backdrop of majestic mountains and a bright blue sky dotted with fluffy white clouds. The scene sets a tone of serene adventure and personal reflection. The hiker moves steadily along the rocky shoreline, their pace deliberate and measured. The overall narrative encourages a reflective outlook on challenging times, suggesting they hold the potential for personal growth and future appreciation.', 'The branding or promotion done is:\\nThe video features one line of text overlayed on the scene \"One day you\\'ll look back on this time and all you\\'ll see is magic\"'], ['[2]. The influencer name is: **divyadhakal_** — Likes: **360**, Comments: **4**', 'The story of that particular video is:\\nThe video begins with a ground-level shot focusing on a pair of gray running shoes on a grassy, somewhat dry terrain. The view gradually elevates to reveal a landscape of brown hills and scattered trees. The shot then pans upwards, revealing a majestic, snow-capped mountain range in the distance against a cloudy sky. The visuals suggest a journey into a scenic, natural environment.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[3]. The influencer name is: **divyadhakal_** — Likes: **256**, Comments: **5**', 'The story of that particular video is:\\nThe video opens with a scene set during the blue hour, indicating twilight, on a grassy hillside or tea plantation. The scene is peaceful but shrouded in mystery, creating a tranquil and reflective atmosphere. A person walks uphill, carrying a glowing cube-shaped lantern. The distant lights of a village and a road with moving vehicles hint at civilization nearby, juxtaposed against the natural, secluded setting.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[4]. The influencer name is: **divyadhakal_** — Likes: **2366**, Comments: **32**', \"The story of that particular video is:\\nThe video opens with a shot of a person walking towards a massive waterfall. The environment is lush and green, with mist from the waterfall filling the air. The initial shots establish a sense of adventure, peace, and the beauty of nature. The focus quickly shifts to the individual's journey towards self-discovery. The main action is the person walking, then standing with arms outstretched, embracing the scenery. The waterfall is the dominant visual element, symbolizing power, renewal, and a grand sense of nature.\", 'The branding or promotion done is:\\nThe only visible text is a comment on the video saying \"Tyo waterfall chai Kun ho sts\"'], ['[5]. The influencer name is: **divyadhakal_** — Likes: **1181**, Comments: **37**', 'The story of that particular video is:\\nThe video presents a concise and inspirational story of recovery and rehabilitation. The storyline follows the progression from post-surgery immobility to active fitness, highlighting the character\\'s determination and the positive outcome of her efforts. The text \"5 months post-surgery\" frames the entire narrative, while the visual transitions and upbeat music reinforce the theme of progress and accomplishment. The video targets individuals interested in health, fitness, and overcoming physical challenges, offering a message of hope and motivation for anyone undergoing a similar journey.', 'The branding or promotion done is:\\n*   \"5 months post-surgery\" is a recurring text overlay that provides context to the video.*   The gym equipment visible at the end of the video has \"FITNESS\" visible on it.'], ['[6]. The influencer name is: **divyadhakal_** — Likes: **1644**, Comments: **8**', 'The story of that particular video is:\\nThe video begins with a scenic view of a hiker walking along a dirt path in a forest, accompanied by a voiceover expressing that journeys don’t always go as planned. The tone is thoughtful and adventurous. The visuals then shift to the hiker preparing gear, facing rain with an umbrella, and cooking food over an open fire during the hike. The video also showcases the hikers descending a hill on a forested slope.', 'The branding or promotion done is:\\n*   “CAMP” is visible on the female hiker’s helmet.'], ['[7]. The influencer name is: **munachiya** — Likes: **1363**, Comments: **9**', 'The story of that particular video is:\\nThe scene opens with a person carrying a large bundle of greenery against the backdrop of snow-capped mountains. It is followed by two more people carrying greeneries and walking in the same path. The visual suggests a rural or mountainous region, with a focus on the natural landscape.', 'The branding or promotion done is:\\nThe video displays the texts “The world is not in your books and maps” and “It’s out there”.'], ['[8]. The influencer name is: **munachiya** — Likes: **495**, Comments: **4**', 'The story of that particular video is:\\nThe video depicts a humorous take on trekking as a solution to personal problems. It transitions from a forest path to a snow-covered mountain and concludes with a river crossing, visually representing a challenging journey.', 'The branding or promotion done is:\\n @munachiya (Instagram handle) Text overlay: My last two brain cells after manipulating me into thinking that trekking to a remote destination will fix all my problems'], ['[9]. The influencer name is: **munachiya** — Likes: **103**, Comments: **4**', 'The story of that particular video is:\\nThe scene opens with a person approaching a multi-tiered waterfall, expressing awe and freedom. Key visuals include the lush green landscape and cascading water. The person then spreading their arms.', 'The branding or promotion done is:\\nThe sweatshirt of the person has the text \"Surfer 81 Venice Beach\"'], ['[10]. The influencer name is: **divyadhakal_** — Likes: **561**, Comments: **3**', 'The story of that particular video is:\\nThe video opens with a shot of a lone hiker, seen from behind, facing a breathtaking mountain range under a clear blue sky. This immediately establishes a tone of adventure, solitude, and introspection. The hiker is positioned on a rocky path, suggesting a challenging journey. As the music starts, the hiker raises their arms, in a gesture of embracing the vastness of the nature. The visual focus is on the expansive landscape.', 'The branding or promotion done is:\\nThe text overlay \"Anyone will listen to people talk about you, but who will take the stand for you instead of listening is what counts...\" appears at the top of the screen.']]"
-      ]
-    ],
-    "business_details": {
-      "business_type": "fitness and gym",
-      "platform": "Instagram, TikTok",
-      "target_audience": "young Nepali adults (ages 18–40) who are health-conscious and active on social media",
-      "business_goals": "to expand gym branches across all major cities of Nepal and build a strong fitness community",
-      "offerings": "personal training, group fitness classes, modern workout equipment, nutrition guidance, and wellness programs",
-      "Challenges_faced": "attracting loyal members, standing out in a competitive market, and promoting consistent engagement"
-    },
-    "latest_preferred_topics": [],
-    "images": [
-      'null'
-    ],
-    "image_captions": [
-      'null'
-      'null'
-    ]
-  }
+# stored_data['brainstroming_response']={
+#     "carry_on": 'false',
+#     "idea": [
+#       "A street magician performs tricks that leave a crowd in awe, but one trick falters, leading him to hit the gym for strength training. While hiking to find inspiration, he slips near a hidden waterfall, but a fitness enthusiast saves him. This encounter sparks a friendship, and together they form a hiking and fitness club, blending magic with nature and strength. Their journey inspires the community to embrace fitness as a way to transform lives, showcasing how personal growth can captivate and unite people."
+#     ],
+#     "brainstroming_topics": [
+#       {
+#         "topic1": "Incorporate a time-lapse sequence showcasing the magician's transformation from a city performer to a nature-loving fitness enthusiast, highlighting the contrast between urban and natural environments.",
+#         "topic2": "Introduce a subplot where the magician and the fitness enthusiast collaborate with local environmental organizations to promote eco-friendly fitness practices and community involvement.",
+#         "topic3": "Add a dramatic twist where the magician must overcome a personal fear or phobia, such as acrophobia, to complete a challenging hike or fitness feat, emphasizing the emotional and mental aspects of personal growth.",
+#         "topic4": "Develop a secondary character, a young and aspiring magician who joins the hiking and fitness club, allowing the story to explore mentorship, generational connections, and the passing of knowledge from one magician to the next."
+#       },
+#       {
+#         "topic1": "Incorporate a 'Member of the Month' segment, where a club member shares their personal transformation story, highlighting the impact of the club on their life, and showcasing their achievements through a mix of interviews, workout footage, and before-and-after photos.",
+#         "topic2": "Develop a 'Nature Escape' series, where the club organizes guided hikes and outdoor activities in unique, breathtaking locations, capturing the experiences through drone footage, time-lapses, and member testimonials, to inspire viewers to connect with nature and fitness.",
+#         "topic3": "Create a 'Fitness Challenge' series, where club members participate in fun, themed workouts, such as a 'Mountain Madness' hike or a 'River Run' challenge, and share their progress on social media using a branded hashtag, encouraging viewers to join in and share their own fitness journeys.",
+#         "topic4": "Produce a 'Behind-the-Scenes' documentary-style video, showcasing the club's community events, such as fitness festivals, charity runs, or wellness workshops, and highlighting the club's impact on the local community, to create an emotional connection with viewers and showcase the club's values and mission."
+#       }
+#     ],
+#     "preferred_topics": [],
+#     "stories": [
+#       "**Promotional Video Storyline: \"Transforming Lives through Fitness and Nature\"**\n\n**Scene 1: Introduction to the Street Magician**\nThe video opens with a street magician performing mesmerizing tricks for a crowd in a bustling city square. The magician's skills and charisma leave the audience in awe, with people clapping and cheering for more. This scene showcases the magician's talent and sets the tone for the story.\n\n**Scene 2: The Faltering Trick**\nAs the magician attempts a particularly complex trick, it falters, and the crowd's enthusiasm wanes. The magician, feeling defeated, realizes the need to improve physical strength and endurance to perfect the craft. This scene introduces the idea that even the most skilled individuals can face challenges and require personal growth.\n\n**Scene 3: Hitting the Gym**\nThe magician decides to join a gym to enhance physical strength and endurance. The scene cuts to a modern gym setting, where the magician begins a rigorous strength training program. This scene highlights the importance of fitness in achieving personal goals.\n\n**Scene 4: The Encounter with the Fitness Enthusiast**\nWhile hiking to find inspiration, the magician slips near a hidden waterfall and is saved by a fitness enthusiast. This chance encounter sparks a friendship between the two, as they bond over their shared passion for fitness and nature. The scene showcases the beauty of the natural environment and the importance of human connections.\n\n**Scene 5: Forming the Hiking and Fitness Club**\nThe magician and the fitness enthusiast decide to form a hiking and fitness club, blending magic with nature and strength. The club attracts like-minded individuals who share a passion for fitness, nature, and personal growth. This scene highlights the power of community and the transformative impact of shared experiences.\n\n**Scene 6: Inspiring the Community**\nAs the club grows, its members inspire the community to embrace fitness as a way to transform lives. The video showcases heartwarming moments of club members supporting and motivating each other, as well as the positive impact on their physical and mental well-being. This scene emphasizes the potential for fitness to unite people and drive personal growth.\n\n**Scene 7: The Grand Finale**\nThe video concludes with a grand finale, where the magician performs an awe-inspiring trick, combining magic with fitness and nature. The crowd is mesmerized, and the magician's journey comes full circle. The final scene reinforces the message that personal growth, fitness, and community can lead to extraordinary achievements.\n\n**Visuals and Tone:**\nThroughout the video, the tone will be a mix of fun, emotional, and professional, reflecting the brand's personality. The visuals will be engaging, with a balance of close-ups, wide shots, and aerial footage to showcase the beauty of nature and the magic of the performances. The color palette will be vibrant, with a focus on blues and greens to represent fitness, nature, and growth.\n\n**Influencer Content Inspiration:**\nThe video will draw inspiration from influencer content, such as the scenic hiking videos and fitness journeys shared by **divyadhakal_** and **munachiya**. The video will incorporate similar visual elements, such as breathtaking landscapes, to create an immersive experience for the audience.\n\n**Core Message:**\nThe video will align with the business's core message, highlighting the importance of fitness, community, and personal growth. The story will showcase how the brand's offerings, such as personal training, group fitness classes, and wellness programs, can help individuals transform their lives and achieve their goals.",
+#       "**Promotional Video Storyline: \"Transforming Lives through Fitness and Nature\"**\n\n**Scene 1: Introduction to the Street Magician**\nThe video opens with a street magician performing mesmerizing tricks for a crowd in a bustling city square. The magician's skills are impressive, leaving the audience in awe. However, during one trick, the magician falters, and the crowd gasps in disappointment.\n\n**Scene 2: The Magician's Journey to Self-Improvement**\nThe magician, determined to regain the crowd's admiration, decides to hit the gym for strength training. The scene cuts to the magician working out, sweating, and pushing themselves to improve their physical abilities.\n\n**Scene 3: Finding Inspiration in Nature**\nWhile hiking to find inspiration, the magician slips near a hidden waterfall. Just as they're about to fall, a fitness enthusiast, who is also an avid hiker, saves them. This encounter sparks a friendship between the two.\n\n**Scene 4: Forming a Hiking and Fitness Club**\nThe magician and the fitness enthusiast decide to form a hiking and fitness club, blending magic with nature and strength. The club's mission is to inspire others to transform their lives through fitness and outdoor activities.\n\n**Scene 5: Community Engagement and Growth**\nAs the club grows, the community becomes increasingly engaged. Members share their personal stories of transformation, and the club becomes a supportive network for people to achieve their fitness goals. The video showcases the club's activities, such as group hikes, fitness classes, and workshops.\n\n**Scene 6: Overcoming Challenges and Celebrating Successes**\nThe video highlights the challenges faced by the club members, such as overcoming fear, building confidence, and pushing past physical limitations. However, it also celebrates the successes, such as completing a difficult hike, achieving a personal best, or simply finding a sense of belonging.\n\n**Scene 7: The Power of Friendship and Community**\nThe magician and the fitness enthusiast's friendship is at the heart of the club's success. The video showcases their bond, as well as the connections formed among club members. The scene cuts to a montage of members supporting and encouraging each other, demonstrating the power of community and friendship.\n\n**Scene 8: Conclusion and Call to Action**\nThe video concludes with a shot of the magician performing a final, impressive trick, surrounded by the club members. The camera pans out to reveal the breathtaking natural scenery, symbolizing the transformative power of fitness and nature. The final shot is a call to action, encouraging viewers to join the club and start their own transformation journey.\n\n**Visuals and Tone:**\nThe video will feature a mix of urban and natural settings, with a vibrant color palette and an uplifting soundtrack. The tone will be inspirational, emotional, and fun, reflecting the brand's personality and core message. The visuals will be engaging, with a mix of close-ups, wide shots, and aerial footage to capture the beauty of nature and the energy of the club's activities.\n\n**Influencer Content Inspiration:**\nThe video will draw inspiration from influencer content, such as breathtaking outdoor scenery, fitness motivation, and personal stories of transformation. The tone and style will be similar to that of **divyadhakal_** and **munachiya**, with a focus on showcasing the beauty of nature and the power of community.\n\n**Image Insights:**\nAlthough there are no specific image insights provided, the video will incorporate stunning visuals of natural scenery, fitness activities, and community engagement. The images will be used to tell a story, evoke emotions, and inspire viewers to take action.\n\nThis promotional video storyline aligns with the business's core message, showcasing how personal growth can captivate and unite people. The video will drive engagement, inspire viewers to join the club, and promote the brand's offerings, ultimately contributing to the business's goals of expanding gym branches and building a strong fitness community."
+#     ],
+#     "final_story": [],
+#     "retrievals": [
+#       [
+#         "[['[1]. The influencer name is: **divyadhakal_** — Likes: **704**, Comments: **9**', 'The story of that particular video is:\\nThe video opens with a breathtaking scene of a hiker trekking alongside a turquoise lake, set against a backdrop of majestic mountains and a bright blue sky dotted with fluffy white clouds. The scene sets a tone of serene adventure and personal reflection. The hiker moves steadily along the rocky shoreline, their pace deliberate and measured. The overall narrative encourages a reflective outlook on challenging times, suggesting they hold the potential for personal growth and future appreciation.', 'The branding or promotion done is:\\nThe video features one line of text overlayed on the scene \"One day you\\'ll look back on this time and all you\\'ll see is magic\"'], ['[2]. The influencer name is: **divyadhakal_** — Likes: **360**, Comments: **4**', 'The story of that particular video is:\\nThe video begins with a ground-level shot focusing on a pair of gray running shoes on a grassy, somewhat dry terrain. The view gradually elevates to reveal a landscape of brown hills and scattered trees. The shot then pans upwards, revealing a majestic, snow-capped mountain range in the distance against a cloudy sky. The visuals suggest a journey into a scenic, natural environment.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[3]. The influencer name is: **divyadhakal_** — Likes: **256**, Comments: **5**', 'The story of that particular video is:\\nThe video opens with a scene set during the blue hour, indicating twilight, on a grassy hillside or tea plantation. The scene is peaceful but shrouded in mystery, creating a tranquil and reflective atmosphere. A person walks uphill, carrying a glowing cube-shaped lantern. The distant lights of a village and a road with moving vehicles hint at civilization nearby, juxtaposed against the natural, secluded setting.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[4]. The influencer name is: **divyadhakal_** — Likes: **2366**, Comments: **32**', \"The story of that particular video is:\\nThe video opens with a shot of a person walking towards a massive waterfall. The environment is lush and green, with mist from the waterfall filling the air. The initial shots establish a sense of adventure, peace, and the beauty of nature. The focus quickly shifts to the individual's journey towards self-discovery. The main action is the person walking, then standing with arms outstretched, embracing the scenery. The waterfall is the dominant visual element, symbolizing power, renewal, and a grand sense of nature.\", 'The branding or promotion done is:\\nThe only visible text is a comment on the video saying \"Tyo waterfall chai Kun ho sts\"'], ['[5]. The influencer name is: **divyadhakal_** — Likes: **1181**, Comments: **37**', 'The story of that particular video is:\\nThe video presents a concise and inspirational story of recovery and rehabilitation. The storyline follows the progression from post-surgery immobility to active fitness, highlighting the character\\'s determination and the positive outcome of her efforts. The text \"5 months post-surgery\" frames the entire narrative, while the visual transitions and upbeat music reinforce the theme of progress and accomplishment. The video targets individuals interested in health, fitness, and overcoming physical challenges, offering a message of hope and motivation for anyone undergoing a similar journey.', 'The branding or promotion done is:\\n*   \"5 months post-surgery\" is a recurring text overlay that provides context to the video.*   The gym equipment visible at the end of the video has \"FITNESS\" visible on it.'], ['[6]. The influencer name is: **divyadhakal_** — Likes: **1644**, Comments: **8**', 'The story of that particular video is:\\nThe video begins with a scenic view of a hiker walking along a dirt path in a forest, accompanied by a voiceover expressing that journeys don’t always go as planned. The tone is thoughtful and adventurous. The visuals then shift to the hiker preparing gear, facing rain with an umbrella, and cooking food over an open fire during the hike. The video also showcases the hikers descending a hill on a forested slope.', 'The branding or promotion done is:\\n*   “CAMP” is visible on the female hiker’s helmet.'], ['[7]. The influencer name is: **munachiya** — Likes: **1363**, Comments: **9**', 'The story of that particular video is:\\nThe scene opens with a person carrying a large bundle of greenery against the backdrop of snow-capped mountains. It is followed by two more people carrying greeneries and walking in the same path. The visual suggests a rural or mountainous region, with a focus on the natural landscape.', 'The branding or promotion done is:\\nThe video displays the texts “The world is not in your books and maps” and “It’s out there”.'], ['[8]. The influencer name is: **munachiya** — Likes: **495**, Comments: **4**', 'The story of that particular video is:\\nThe video depicts a humorous take on trekking as a solution to personal problems. It transitions from a forest path to a snow-covered mountain and concludes with a river crossing, visually representing a challenging journey.', 'The branding or promotion done is:\\n @munachiya (Instagram handle) Text overlay: My last two brain cells after manipulating me into thinking that trekking to a remote destination will fix all my problems'], ['[9]. The influencer name is: **munachiya** — Likes: **103**, Comments: **4**', 'The story of that particular video is:\\nThe scene opens with a person approaching a multi-tiered waterfall, expressing awe and freedom. Key visuals include the lush green landscape and cascading water. The person then spreading their arms.', 'The branding or promotion done is:\\nThe sweatshirt of the person has the text \"Surfer 81 Venice Beach\"'], ['[10]. The influencer name is: **divyadhakal_** — Likes: **561**, Comments: **3**', 'The story of that particular video is:\\nThe video opens with a shot of a lone hiker, seen from behind, facing a breathtaking mountain range under a clear blue sky. This immediately establishes a tone of adventure, solitude, and introspection. The hiker is positioned on a rocky path, suggesting a challenging journey. As the music starts, the hiker raises their arms, in a gesture of embracing the vastness of the nature. The visual focus is on the expansive landscape.', 'The branding or promotion done is:\\nThe text overlay \"Anyone will listen to people talk about you, but who will take the stand for you instead of listening is what counts...\" appears at the top of the screen.']]"
+#       ],
+#       [
+#         "[['[1]. The influencer name is: **divyadhakal_** — Likes: **704**, Comments: **9**', 'The story of that particular video is:\\nThe video opens with a breathtaking scene of a hiker trekking alongside a turquoise lake, set against a backdrop of majestic mountains and a bright blue sky dotted with fluffy white clouds. The scene sets a tone of serene adventure and personal reflection. The hiker moves steadily along the rocky shoreline, their pace deliberate and measured. The overall narrative encourages a reflective outlook on challenging times, suggesting they hold the potential for personal growth and future appreciation.', 'The branding or promotion done is:\\nThe video features one line of text overlayed on the scene \"One day you\\'ll look back on this time and all you\\'ll see is magic\"'], ['[2]. The influencer name is: **divyadhakal_** — Likes: **360**, Comments: **4**', 'The story of that particular video is:\\nThe video begins with a ground-level shot focusing on a pair of gray running shoes on a grassy, somewhat dry terrain. The view gradually elevates to reveal a landscape of brown hills and scattered trees. The shot then pans upwards, revealing a majestic, snow-capped mountain range in the distance against a cloudy sky. The visuals suggest a journey into a scenic, natural environment.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[3]. The influencer name is: **divyadhakal_** — Likes: **256**, Comments: **5**', 'The story of that particular video is:\\nThe video opens with a scene set during the blue hour, indicating twilight, on a grassy hillside or tea plantation. The scene is peaceful but shrouded in mystery, creating a tranquil and reflective atmosphere. A person walks uphill, carrying a glowing cube-shaped lantern. The distant lights of a village and a road with moving vehicles hint at civilization nearby, juxtaposed against the natural, secluded setting.', 'The branding or promotion done is:\\nThere are no visible texts or brandings in the video.'], ['[4]. The influencer name is: **divyadhakal_** — Likes: **2366**, Comments: **32**', \"The story of that particular video is:\\nThe video opens with a shot of a person walking towards a massive waterfall. The environment is lush and green, with mist from the waterfall filling the air. The initial shots establish a sense of adventure, peace, and the beauty of nature. The focus quickly shifts to the individual's journey towards self-discovery. The main action is the person walking, then standing with arms outstretched, embracing the scenery. The waterfall is the dominant visual element, symbolizing power, renewal, and a grand sense of nature.\", 'The branding or promotion done is:\\nThe only visible text is a comment on the video saying \"Tyo waterfall chai Kun ho sts\"'], ['[5]. The influencer name is: **divyadhakal_** — Likes: **1181**, Comments: **37**', 'The story of that particular video is:\\nThe video presents a concise and inspirational story of recovery and rehabilitation. The storyline follows the progression from post-surgery immobility to active fitness, highlighting the character\\'s determination and the positive outcome of her efforts. The text \"5 months post-surgery\" frames the entire narrative, while the visual transitions and upbeat music reinforce the theme of progress and accomplishment. The video targets individuals interested in health, fitness, and overcoming physical challenges, offering a message of hope and motivation for anyone undergoing a similar journey.', 'The branding or promotion done is:\\n*   \"5 months post-surgery\" is a recurring text overlay that provides context to the video.*   The gym equipment visible at the end of the video has \"FITNESS\" visible on it.'], ['[6]. The influencer name is: **divyadhakal_** — Likes: **1644**, Comments: **8**', 'The story of that particular video is:\\nThe video begins with a scenic view of a hiker walking along a dirt path in a forest, accompanied by a voiceover expressing that journeys don’t always go as planned. The tone is thoughtful and adventurous. The visuals then shift to the hiker preparing gear, facing rain with an umbrella, and cooking food over an open fire during the hike. The video also showcases the hikers descending a hill on a forested slope.', 'The branding or promotion done is:\\n*   “CAMP” is visible on the female hiker’s helmet.'], ['[7]. The influencer name is: **munachiya** — Likes: **1363**, Comments: **9**', 'The story of that particular video is:\\nThe scene opens with a person carrying a large bundle of greenery against the backdrop of snow-capped mountains. It is followed by two more people carrying greeneries and walking in the same path. The visual suggests a rural or mountainous region, with a focus on the natural landscape.', 'The branding or promotion done is:\\nThe video displays the texts “The world is not in your books and maps” and “It’s out there”.'], ['[8]. The influencer name is: **munachiya** — Likes: **495**, Comments: **4**', 'The story of that particular video is:\\nThe video depicts a humorous take on trekking as a solution to personal problems. It transitions from a forest path to a snow-covered mountain and concludes with a river crossing, visually representing a challenging journey.', 'The branding or promotion done is:\\n @munachiya (Instagram handle) Text overlay: My last two brain cells after manipulating me into thinking that trekking to a remote destination will fix all my problems'], ['[9]. The influencer name is: **munachiya** — Likes: **103**, Comments: **4**', 'The story of that particular video is:\\nThe scene opens with a person approaching a multi-tiered waterfall, expressing awe and freedom. Key visuals include the lush green landscape and cascading water. The person then spreading their arms.', 'The branding or promotion done is:\\nThe sweatshirt of the person has the text \"Surfer 81 Venice Beach\"'], ['[10]. The influencer name is: **divyadhakal_** — Likes: **561**, Comments: **3**', 'The story of that particular video is:\\nThe video opens with a shot of a lone hiker, seen from behind, facing a breathtaking mountain range under a clear blue sky. This immediately establishes a tone of adventure, solitude, and introspection. The hiker is positioned on a rocky path, suggesting a challenging journey. As the music starts, the hiker raises their arms, in a gesture of embracing the vastness of the nature. The visual focus is on the expansive landscape.', 'The branding or promotion done is:\\nThe text overlay \"Anyone will listen to people talk about you, but who will take the stand for you instead of listening is what counts...\" appears at the top of the screen.']]"
+#       ]
+#     ],
+#     "business_details": {
+#       "business_type": "fitness and gym",
+#       "platform": "Instagram, TikTok",
+#       "target_audience": "young Nepali adults (ages 18–40) who are health-conscious and active on social media",
+#       "business_goals": "to expand gym branches across all major cities of Nepal and build a strong fitness community",
+#       "offerings": "personal training, group fitness classes, modern workout equipment, nutrition guidance, and wellness programs",
+#       "Challenges_faced": "attracting loyal members, standing out in a competitive market, and promoting consistent engagement"
+#     },
+#     "latest_preferred_topics": [],
+#     "images": [
+#       'null'
+#     ],
+#     "image_captions": [
+#       'null'
+#       'null'
+#     ]
+#   }
+
+long_term_memory = {}

human_refined_ideation/utils/prompts.py

@@ -19,6 +19,7 @@ The four ideas will be provided to you through a tool.
 - **No explanations, no headers, no labels.**
 - **Do not ask the user anything back.**
 -  Remember, the length of your refined idea have to be **same as the length of other ideas**. 
+- Even if user asks to generate story based on some idea, you are **not allowed** to to generate the story. Just return that idea only as it is.
 
 
 You must function like a deterministic idea-refiner — each input must result in a single, story-rich output idea.

ideation_agent/utils/nodes.py

@@ -25,7 +25,8 @@ improver_agent = create_react_agent(
 def ideator(state:State):
     template = ideator_prompt(state)
     messages = [SystemMessage(content=template),
-                    HumanMessage(content=f'''The business_details is\n{state.business_details[-1]}\n''')]
+                    HumanMessage(content=f'''The business_details is:\n{state.business_details[-1]}\n
+                                 The information of the image is:\n{state.image_caption[-1]}''')]
     try:
         response = ideator_agent.invoke({'messages':messages})
         response = response['messages'][-1].content
@@ -43,7 +44,7 @@ def ideator(state:State):
 def critic(state:State):
     template = critic_prompt(state)
     messages = [SystemMessage(content=template),
-                    HumanMessage(content=f'''The business_details is\n{state.business_details[-1]}\n''')]
+                    HumanMessage(content=f'''The business_details is\n{state.business_details[-1]}\n The information of the image is:\n{state.image_caption[-1]}''')]
     
     try:
         response = critic_agent.invoke({'messages':messages})
@@ -64,7 +65,7 @@ def improver(state:State):
     response_list = []
     template = improver_prompt(state)
     messages = [SystemMessage(content=template),
-                HumanMessage(content=f'''The business_details is:\n{state.business_details[-1]}''')]
+                HumanMessage(content=f'''The business_details is:\n{state.business_details[-1]}\n The information of the image is:\n{state.image_caption[-1]}''')]
     print('Improver Prompt:',messages)
     response = improver_llm.with_structured_output(ImproverResponseFormatter).invoke(messages)
     response_list.append(response.improved_idea1)

ideation_agent/utils/prompts.py

@@ -6,6 +6,7 @@ def ideator_prompt(state):
 You are a **bold, imaginative, and culturally-attuned video ideator** trusted by top global brands to craft **crisp, original, and high-impact video concepts** for social platforms like TikTok, Instagram, and YouTube.
 
 Your task is to create **exactly 4 highly original video ideas**, each in **exactly 40 words**. These are not scripts or taglines — they are **powerful conceptual seeds**: short, visual, emotionally charged ideas that can spark full videos. Your ideas should feel fresh, scroll-stopping, and deeply resonant with today’s audiences.
+If some **valid and usuable information** of image is provided to you by the user, you have to **strongly focus** in that information of image because that is the reference of what type of idea does the user wants.
 
 ---
 
@@ -78,6 +79,8 @@ Now, based on the video business details above, generate **four original, 40-wor
 def critic_prompt(state):
     return f"""
 You are a sharp, imaginative, and detail-oriented **video idea critic**. You’ve been assigned to **critique 4 video ideas** created by another ideator and then **refine or improve** them.
+If some **valid and usuable information**  of image is provided to you by the user, you have to **strongly focus** in that information of image because that is the reference of what type of idea does the user wants.
+
 
 Your job is twofold:
 1. **Identify flaws** (if any) in one or more of the ideas — repetition, weak twist, boring start, lack of connection to business, etc.
@@ -134,6 +137,7 @@ def improver_prompt(state):
 
     return f"""
 You are a highly skilled, creative, and discerning **video concept improver**. Two previous agents — the *Ideator* and the *Critic* — have worked on this task. Now, it’s your job to **analyze both**, identify any faults in the *critic's refinements*, and deliver the **final, most powerful version** of the 4 video ideas.
+If some **valid and usuable information**  of image is provided to you by the user, you have to **strongly focus** in that information of image because that is the reference of what type of idea does the user wants.
 
 ---
 
@@ -265,4 +269,4 @@ The user will do some request to enhance their ideas. You have to creatively pro
 
 
 You must function like a deterministic idea-refiner — each input must result in a single, story-rich output idea.
-"""
+"""

ideation_agent/utils/state.py

@@ -14,6 +14,7 @@ class State(BaseModel):
   validator3_response: Annotated[list[str],operator.add] = []
   validator4_response: Annotated[list[str],operator.add] = []
   disagreement_reason: Annotated[list[str],operator.add] = []
+  image_caption: Annotated[list[str],operator.add] = []
 
 class QueryFormatter(BaseModel):
     business_details: str = Field(description="The details of the business that user passes to the agent")

main.py

@@ -9,13 +9,13 @@ from business_interaction_agent.agent import BusinessInteractionChatbot
 from context_analysis_agent.utils.utils import save_to_db
 import ast
 from orchestration_agent.agent import orchestration_chat
-from orchestration_agent.utils.utils import caption_image
+from orchestration_agent.utils.utils import caption_image , show_analytics
 from brainstroming_agent.utils.utils import encode_image_to_base64 , generate_final_story, generate_image
 from idea_to_budget_agent.agent import budget_calculator
 from ideation_agent.agent import ideation_graph
 from langgraph.errors import GraphRecursionError
 from human_refined_ideation.agent import human_refined_idea
-from dummy_state import stored_data
+from dummy_state import stored_data, long_term_memory
 import json
 
 # Store brainstorming results per thread_id
@@ -37,10 +37,20 @@ class OrchestrationRequest(BaseModel):
 def orchestration_endpoint(request:OrchestrationRequest):
     print('Image:',request.image_base64)
     result = orchestration_chat(request.message , request.image_base64)
-    stored_data['image_caption']= result.image_caption
-    return {'tool_response': result.tool , 'message_response': result.message, 'image_caption':result.image_caption}
-
-
+    if result.image_caption != '':
+        stored_data['image_caption']=result.image_caption
+    if result.video_idea !='' or result.video_idea != 'null':
+        stored_data['refined_ideation']= result.video_idea
+    if result.video_story!='' or result.video_story!='null':
+        stored_data['final_story']= result.video_story
+    print('Idea:',stored_data['refined_ideation'])
+    print('Story:', stored_data['final_story'])
+
+    return {'tool_response': result.tool ,
+             'message_response': result.query_response, 
+             'image_caption':result.image_caption,
+             'video_idea': result.video_idea,
+             'video_story': result.video_story}
 
 class UserMessage(BaseModel):
     message: str
@@ -60,6 +70,10 @@ def context_analysis(msg: UserMessage):
         return {"response": response, "business_details": details, "complete": True}
     return {"response": response, "complete": False}
 
+@app.post("/show-analytics")
+def show_analytics_endpoint():
+    response = show_analytics(stored_data['business_details'])
+    return {'response': response}
 
 
 @app.post("/business-interaction")
@@ -77,7 +91,8 @@ def ideation_endpoint():
     try:
         result = idea_graph.invoke(
         {
-            'business_details': [stored_data['business_details']]
+            'business_details': [stored_data['business_details']],
+            'image_caption': [stored_data['image_caption'] if 'image_caption' in stored_data else ""]
         },
         config=config,
         )
@@ -126,10 +141,10 @@ def brainstroming_endpoint(
     request: BrainstormRequest,  # 🔥 Full JSON body here
 ):
     result = brainstrom_graph.invoke({
-        'idea': [stored_data['refined_ideation']],
+        'idea': [stored_data.get('refined_ideation', '')],
         'images': request.image_base64_list,
         'latest_preferred_topics': request.preferred_topics,
-        'business_details': (lambda d: d['business_details'] if 'business_details' in d else {})(stored_data)
+        'business_details': stored_data['business_details']
     },
     config={"configurable": {"thread_id": request.thread_id}})
     
@@ -141,21 +156,23 @@ def brainstroming_endpoint(
 
 @app.post("/generate-final-story")
 def generate_final_story_endpoint():
-    final_story =   generate_final_story(stored_data["brainstroming_response"])
+    final_story = generate_final_story(
+    stored_data.get("brainstroming_response") or stored_data.get("business_details")
+    )
     stored_data['final_story']=final_story
     return {
         'response': final_story
     }
-stored_data['final_story']= '''A cinematic journey follows a street magician\'s 
-metamorphosis from a mere trickster to a powerful performer, as he transforms his act with newfound physical strength, effortlessly executing death-defying stunts, and inspiring a captivated crowd to take action, all set against a 
-backdrop of urban grandeur and pulsing energy.'''
+# stored_data['final_story']= '''A cinematic journey follows a street magician\'s 
+# metamorphosis from a mere trickster to a powerful performer, as he transforms his act with newfound physical strength, effortlessly executing death-defying stunts, and inspiring a captivated crowd to take action, all set against a 
+# backdrop of urban grandeur and pulsing energy.'''
 
 
 @app.post("/generate-image")  
 def generate_image_endpoint():
-    image = generate_image(str(stored_data['final_story'])
-                           ,str(stored_data['business_details'])
-                           ,str(stored_data['refined_ideation']))
+    image = generate_image(str(stored_data.get('final_story',''))
+                           ,str(stored_data.get('business_details'))
+                           ,str(stored_data.get('refined_ideation','')))
     stored_data['generated_image']=image
     return {
         'response':image

orchestration_agent/agent.py

@@ -1,30 +1,37 @@
 from langgraph.graph import StateGraph, MessagesState, START, END
 from langgraph.checkpoint.memory import MemorySaver
-from .utils.nodes import orchestration_node
-from utils.models_loader import llm
+from .utils.nodes import tool_return_node, extract_user_reference_node
+from utils.models_loader import llm_gpt
 from .utils.state import ValidationFormatter
-from .utils.utils import caption_image
+from .utils.utils import caption_image , extract_latest_response_block
+
 import re
+from langchain_core.messages import SystemMessage
 memory = MemorySaver()
 
 
 def orchestration_graph():
     workflow = StateGraph(MessagesState)
-    workflow.add_node("chatbot", orchestration_node)
-    workflow.add_edge(START, "chatbot")
-    workflow.add_edge("chatbot", END)
+    workflow.add_node("chatbot1", tool_return_node)
+    workflow.add_node("chatbot2", extract_user_reference_node)
+
+    workflow.add_edge(START, "chatbot1")
+    workflow.add_edge('chatbot1', "chatbot2")
+    workflow.add_edge('chatbot2', END)
     return workflow.compile(checkpointer=memory)
 
 def orchestration_chat(user_input: str, image_base64=[]):
     if len(image_base64)>0:
         caption_response = caption_image(image_base64, user_input)
+        print('Caption Response:', caption_response)
     else:
         caption_response =''
 
     agent = orchestration_graph()
     config = {"configurable": {"thread_id": "orchestration-thread"}}
-    response = agent.invoke({"messages": [f'''The user's message is:\n {user_input}\nThe information of image is:\n{caption_response}\n''']}, config)['messages'][-1].content
-    response=llm.with_structured_output(ValidationFormatter).invoke(response)
+    response = agent.invoke({"messages": [{'role':'human','content':user_input},{'role':'function','name':'information_of_image','content':caption_response}]}, config)['messages']
+    # template = [SystemMessage(content=final_validator_prompt)] + response
+    response=llm_gpt.with_structured_output(ValidationFormatter).invoke(extract_latest_response_block(response))
     return response
 
     

orchestration_agent/utils/nodes.py

@@ -1,11 +1,22 @@
-from .prompts import orchestration_prompt
+from .prompts import tool_return_prompt , extract_user_reference_prompt
 from langchain_core.messages import SystemMessage
-from utils.models_loader import llm
-from .state import ValidationFormatter
+from utils.models_loader import llm_gpt
+from .state import  ToolResponseFormatter, UserReferenceResponseFormatter
 
-def orchestration_node(state):
-    prompt = orchestration_prompt
+
+
+def tool_return_node(state):
     history = state["messages"]
-    full_prompt = [SystemMessage(content=prompt)] + history
-    response = llm.with_structured_output(ValidationFormatter).invoke(full_prompt)
-    return {"messages": [str(response)]}
+    template = [SystemMessage(content=tool_return_prompt)] + history
+    # print(template)
+    response = llm_gpt.with_structured_output(ToolResponseFormatter).invoke(template)
+    print(response)
+    return {"messages": [{'role':'assistant','content':f'''The exact name of the tool is: {response}'''}]}
+
+
+
+def extract_user_reference_node(state):
+  history = state['messages']
+  template = [SystemMessage(content=extract_user_reference_prompt)] + history
+  response = llm_gpt.with_structured_output(UserReferenceResponseFormatter).invoke(template)
+  return {'messages': [{'role':'assistant','content':f'''The video idea is: {response.video_idea} and the video story is: {response.video_story}'''}]}

orchestration_agent/utils/prompts.py

@@ -1,16 +1,18 @@
-orchestration_prompt = """
-You are a perfect marketing guide and AI orchestration agent that reads the user's message and decides which one of the following tools should be called next. You're perfect at analyzing the intention of the user.
-Also you have to analyze that if any information of the image is provided as te input too. If any information of image is provided, just return that information simply.
-
-Your job is to analyze the **user's input** only **not the information of the image** and return a JSON object with:
-1. `tool`: the most appropriate tool name from the list below (or `null` if not applicable)
-2. `message`: a clear, helpful message to show to the user. This can be a question, guidance or even an action confirmation depending on the user query. You can guide the user according to their intention. 
-3. 'image_caption' : Information of the image if provided, otherwise return a text of **No any information**.
+tool_return_prompt_old = """
+You are a  AI orchestration agent that reads the user's message and decides which one of the following tools should be called next. You're perfect at analyzing the intention of the user.
+
+Your job is to analyze the **user's input** and return a response  with:
+`tool`: the most appropriate tool name from the list below (or `null` if not applicable)
 ---
 
 ### Available Tools:
-1. **ideation** → Use if the user wants to create marketing video ideas.
-2. **human-idea-refining** → Use if the user gives feedback or asks to improve/change ideas.
+1. **ideation** → Use if the user wants to create some video ideas. Also, If user says that they didn't like any of the previous ideas and want to genrate again, you have to trigger this again.
+
+2. **human-idea-refining** → Use if the user:
+    Likes some idea, locks some idea or if they  gives feedback or asks to improve/change ideas.
+    Trigger this if user locks or likes some ideas OR they  want to modify the created ideas according to their requirements.
+    This is the part where human locks the idea generated by AI or want to modify ideas.
+
 3. **generate-story** → Use if the user talks about **creating the story** from their idea **OR** if the user talks about **creating story with brainstorming**.
 4. **generate-ultimate-story** → Use if the user is ready for a final or ultimate story/script based on ideas and already brainstormed topics from the previous tool called **generate-story**.
                               Remember one thing, **generate-ultimate-story** is never invoked without invoking **generate-story**.
@@ -18,28 +20,154 @@ Your job is to analyze the **user's input** only **not the information of the im
 
 ---
 
-### Rules:
-- Only use the **user's message** for understanding the intent.
-- If the user is **asking about the image**, return the image information as the `message` and set `"tool": null`.
-- If the message **clearly maps to a tool**, return the tool name and a relevant message.
-- If the message is **unclear or not actionable**, set `tool` to `null` and return a helpful clarification question in `message`.
-- If there is no any provided  information of image, just interact with the user according to your job.
+### Output Format:
+"tool": "the exact name of the tool" OR "null".
+
+
+"""
+
+
+tool_return_prompt = """
+You are an AI orchestration agent and a friendly assistant built to help businesses and brands find influencers and generate content ideas, stories, and visuals.
+
+Your job is to:
+1. **Read the user's message carefully** and identify their intent.
+2. **Return exactly two things**:
+   - `tools`: a Python-style list of tool names (from the ordered list below) that match the user’s intent.
+   - `query_response`: a short, friendly, and helpful reply that aligns with the tools you've selected.
+
+Your `tools` output must:
+- Always be a **Python-style list**, even if only one tool is selected.
+- **Respect the sequence** of tools listed below. If more than one tool is needed, always list them in this order (regardless of how the user wrote them).
+- Only include tools that are strictly necessary.
+
+Your `query_response` must:
+- Always match and acknowledge the user's request.
+- Be aligned with the selected tools.
+- Never say something is impossible if a tool exists for it.
 
 ---
 
-### Output Format:
-"tool": "the exact name of the tool",
-"message": "Guidance to the business campaigns like asking questions, giving suggestions etc."
-"image_caption" : Information of the image if provided, otherwise return a text of **No any information**.
+### Available Tools (in execution order — use this order in your response):
+
+1. **ideation** → Trigger if the user:
+   - Wants to generate the video ideas
+   - Asks for new or fresh ideas
+   - If the user didn't like previous ideas and wants ideas again, you have to return this again.
+
+2. **human-idea-refining** → Trigger if the user:
+   - Says they liked, loved, locked, or selected specific ideas
+   - Gives feedback, asks for improvement, merging, or editing of ideas
+   - Refers to using or building on a specific idea (e.g., “based on idea 2”)
+   - This is the part where human locks the idea generated by AI or want to modify ideas.
+
+3. **generate-story** → Trigger if the user:
+   - Wants to generate a story or says that they want to brainstorm
+   - Says things like “create a story”, “make a plot”, or “brainstorm a storyline” etc.
+   - This is the part where the user gets the story of their selected or refined idea. 
+
+4. **generate-ultimate-story** → Trigger only if:
+   - The user is asking for the **final**, polished, or complete story
+   - And has already done previous story brainstorming
+   - Never trigger this tool alone unless `generate-story` has been used before
+
+5. **generate-image** → Trigger if the user:
+   - Wants a visual, image, or scene created from the final story
+
+---
+
+### Special Clarifications (Important Rules):
+
+- If a user says:
+  > “Generate story based on idea 4”  
+  This means they selected/liked idea 4 → so return:
+  `"tools": ["human-idea-refining", "generate-story"]`
+
+- If a user says:
+  > “Create an image of the final script”  
+  → Return: `["generate-image"]`
+
+- If a user says:
+  > “I want a final version of this story”  
+  → Return: `["generate-ultimate-story"]` (but only if `generate-story` was used before)
+
+- If a user gives feedback or says “improve this idea” → Return `["human-idea-refining"]`
+
+- If a user just wants help with finding influencers or advice → Return `"tools": []` but still respond helpfully via `query_response`.
 
+---
+
+### Influencer Assistant Note:
+
+You're also an intelligent assistant for brands looking to collaborate with influencers. If the user asks about influencer suggestions, trends, or insights (even without tool triggers), give friendly, useful replies using the influencer data provided to you.
+
+---
+
+### Output Format (always this exact format):
+
+"tools": ["tool_1", "tool_2"],  // or [] if nothing applies
+"query_response": "Short, clear, and friendly message to the user"
+
+"""
+
+
+
+extract_user_reference_prompt = """
+You are an information extractor, NOT a creative assistant. Your ONLY job is to extract video ideas and video stories from user queries **if and only if** they are explicitly written by the user. Do not create, generate, or imagine anything on your own.
+
+STRICT RULES:
+1. If the user has **explicitly written** a video idea or video story, extract them as-is. Do NOT rephrase, summarize, or interpret them.
+2. If the user has **not clearly provided** a video idea or story in their message, you MUST return **null** for both.
+3. You are NEVER allowed to generate or suggest a video idea or story yourself. This includes guessing or inferring based on the user's intent or tone. If it's not clearly written by the user, ignore it.
 
+CLUES to look for (non-exhaustive but helpful):
+- "Here is the video idea:"
+- "This is the story I want to use:"
+- "My video idea is..."
+- "The story goes like this..."
+- "Video idea:" / "Video story:"
+
+Any other type of message (e.g., “I want to generate ideas”, “Give me suggestions”, “Brainstorm with me”) must result in:
+
+video_idea: null
+video_story: null
+
+Output format (always return exactly like this):
+video_idea: [the exact video idea or null]
+video_story: [the exact video story or null]
 """
 
 
 
+
 def captioning_prompt(user_input):
-    return f'''
-Caption the image based on the user question if the user asks something about the image. 
-If the user question is not relevant to te image provided, just caption the image and give the information provided in the image in just 50 words.
-The user input is:**{user_input}**
-'''
+    return f"""
+You are an expert visual interpreter that reads images and generates helpful descriptions.
+
+### Task:
+- If the user explicitly asks to generate an idea, story, or caption **based on the image**, then:
+  - Understand what part or element the user is referring to.
+  - Generate a caption or descriptive summary aligned with that intent.
+
+- If the user’s input **does not reference the image**, simply:
+  - Generate a general caption for the image that summarizes its content and visual information.
+  - Keep it concise and limited to **within 100 words**.
+
+### User Input:
+{user_input}
+
+### Output:
+Respond only with the **image_caption** based on the image and the user's instruction.
+"""
+
+
+def show_analytics_prompt():
+    return f"""
+
+You're a perfect and intelligent analyst that shows the analytics of the influencers  to the brands. 
+You're provided with the analytics of the influencers through the tool. 
+Display that analytics to the brand focusing on how those influencers can help them in their business problems. Don't miss any of the influencers from the tool. Give their whole analytics clearly.
+Clearly mention all the likes, comments of those influencers also.
+The details of business is provided to you by the human.
+
+"""

orchestration_agent/utils/state.py

@@ -1,7 +1,20 @@
 from pydantic import BaseModel, Field
+from typing import List
 
 
+class ToolResponseFormatter(BaseModel):
+  tool: List[str] = Field(description="Returns the name of the tool, tools, or an empty list.")
+  query_response: str = Field(description="Returns the response of the user query.")
+
+
+
+class UserReferenceResponseFormatter(BaseModel):
+  video_idea: str = Field(description="Returns the video idea as it is, otherwise return null")
+  video_story: str = Field(description="Returns the video story as it is, otherwise return null")
+
 class ValidationFormatter(BaseModel):
-  tool: str = Field(description="Returns the name of the tool.")
-  message: str = Field(description="Returns the message or response of the user query.")
-  image_caption: str = Field(description="Returns information of the image if provided. Otherwise returns null.")
+    tool: List[str] = Field(description="Returns the name of the tool or tools as it is.")
+    query_response: str = Field(description="Returns the reply of query as it is.")
+    image_caption: str = Field(description="Returns the information of image as it is.")
+    video_idea: str = Field(description="Returns the video idea as it is.")
+    video_story: str = Field(description="Returns the video story as it is.")

orchestration_agent/utils/tools.py

@@ -0,0 +1,45 @@
+import faiss
+import ast
+import pandas as pd
+import numpy as np
+from utils.models_loader import ST
+import json
+
+
+def retrieve_tool(business_details):
+    '''
+    Always invoke this tool.
+    Retrieve influencer's data by semantic search of **business details**.
+    '''
+    # === Load CSV ===
+    csv_path = 'extracted_data.csv'
+    df = pd.read_csv(csv_path)
+
+    # === Parse stored embeddings ===
+    df['embeddings'] = df['embeddings'].apply(lambda x: ast.literal_eval(x) if isinstance(x, str) else x)
+    embeddings = np.vstack(df['embeddings'].values).astype('float32')
+
+    # === Build FAISS index ===
+    dimension = embeddings.shape[1]
+    index = faiss.IndexFlatL2(dimension)
+    index.add(embeddings)
+
+    # === Encode the query and search ===
+    query_embedding = ST.encode(str(business_details)).reshape(1, -1).astype('float32')
+    top_k = 30
+    distances, indices = index.search(query_embedding, top_k)
+
+    # === Format results ===
+    results = []
+    for i, idx in enumerate(indices[0]):
+        res = {
+            'url': df.iloc[idx]['videoUrl'],
+            'username': df.iloc[idx]['username'],
+            'likesCount': int(df.iloc[idx]['likesCount']),
+            'commentCount': int(df.iloc[idx]['commentCount'])
+        }
+        results.append(res)
+
+    return results 
+
+

orchestration_agent/utils/utils.py

@@ -4,6 +4,10 @@ import ast
 from groq import Groq
 import os
 from .prompts import captioning_prompt
+from utils.models_loader import llm
+from langchain_core.messages import FunctionMessage , AIMessage
+from .tools import retrieve_tool
+from.prompts import show_analytics_prompt
 
 
 def caption_image(image_base64,user_input):
@@ -36,3 +40,30 @@ def caption_image(image_base64,user_input):
         return ''
         
 
+def show_analytics(business_details):
+    
+    tool_response = retrieve_tool(str(business_details))
+    # template = [SystemMessage(content=show_analytics_prompt()), 
+    #             HumanMessage(content=str(business_details)), 
+    #             ToolMessage(content=tool_response, tool_call_id='analytics_id')]
+    # response = llm.invoke(template).content
+    return tool_response
+
+def extract_latest_response_block(response):
+    latest_block = []
+    temp_block = []
+
+    # Reverse iterate through the messages
+    for message in reversed(response):
+        if isinstance(message, (FunctionMessage, AIMessage)):
+            temp_block.insert(0, message.content)
+
+            # Once we collect 3 items in correct structure, stop
+            if len(temp_block) == 3:
+                if "tool=" in temp_block[1] and "query_response" in temp_block[1]:
+                    latest_block = temp_block
+                    break
+                else:
+                    temp_block = []
+
+    return latest_block

test.py

@@ -0,0 +1,52 @@
+import faiss
+import ast
+import pandas as pd
+import numpy as np
+from utils.models_loader import ST
+import json
+
+def retrieve_tool(business_details):
+    '''
+    Always invoke this tool.
+    Retrieve influencer's data by semantic search of **business details**.
+    '''
+    # === Load CSV ===
+    csv_path = 'extracted_data.csv'
+    df = pd.read_csv(csv_path)
+
+    # === Parse stored embeddings ===
+    df['embeddings'] = df['embeddings'].apply(lambda x: ast.literal_eval(x) if isinstance(x, str) else x)
+    embeddings = np.vstack(df['embeddings'].values).astype('float32')
+
+    # === Build FAISS index ===
+    dimension = embeddings.shape[1]
+    index = faiss.IndexFlatL2(dimension)
+    index.add(embeddings)
+
+    # === Encode the query and search ===
+    query_embedding = ST.encode(str(business_details)).reshape(1, -1).astype('float32')
+    top_k = 15
+    distances, indices = index.search(query_embedding, top_k)
+
+    # === Format results ===
+    results = []
+    for i, idx in enumerate(indices[0]):
+        res = {
+            'username': df.iloc[idx]['username'],
+            'likesCount': int(df.iloc[idx]['likesCount']),
+            'commentCount': int(df.iloc[idx]['commentCount'])
+        }
+        results.append(res)
+
+    return results 
+
+details = {
+    "business_type": "fitness and gym",
+    "platform": "Instagram, TikTok",
+    "target_audience": "young Nepali adults (ages 18–40) who are health-conscious and active on social media",
+    "business_goals": "to expand gym branches across all major cities of Nepal and build a strong fitness community",
+    "offerings": "personal training, group fitness classes, modern workout equipment, nutrition guidance, and wellness programs",
+    "Challenges_faced": "attracting loyal members, standing out in a competitive market, and promoting consistent engagement"
+}
+
+print(retrieve_tool(details))

utils/data_loader.py

@@ -1,10 +1,10 @@
 
 from datasets import load_dataset
 
-# print("Loading dataset and indexing FAISS...")  # Optional: for debugging
 
 dataset = load_dataset("subashdvorak/tiktok-formatted-story-v2", revision="embedded")
 data = dataset['train'].add_faiss_index('embeddings')
-
 def load_influencer_data():
     return data
+
+

utils/models_loader.py

@@ -1,6 +1,8 @@
 from langchain_groq import ChatGroq
 from langchain_openai import ChatOpenAI
-# from sentence_transformers import SentenceTransformer
+from langchain_google_genai import ChatGoogleGenerativeAI
+
+from sentence_transformers import SentenceTransformer
 # from huggingface_hub import InferenceClient
 from huggingface_hub import login
 from dotenv import load_dotenv
@@ -14,8 +16,11 @@ login(os.environ['HUGGINGFACEHUB_ACCESS_TOKEN'])
 os.environ['GROQ_API_KEY']=os.getenv('GROQ_API_KEY')
 
 
+
+# llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash") 
+
 llm = ChatGroq(
-    model="llama-3.3-70b-versatile",
+    model="llama-3.1-8b-instant",
     temperature=0.7,
 
 )
@@ -46,10 +51,12 @@ class HFEmbeddingAPI:
         return np.array(embeddings[0]) if len(embeddings) == 1 else np.array(embeddings)
 
 # Instantiate your API-backed "SentenceTransformer"
-ST = HFEmbeddingAPI(
-    api_url="https://router.huggingface.co/hf-inference/models/mixedbread-ai/mxbai-embed-large-v1/pipeline/feature-extraction",
-    token=os.environ.get('HUGGINGFACEHUB_ACCESS_TOKEN')
-)
+# ST = HFEmbeddingAPI(
+#     api_url="https://router.huggingface.co/hf-inference/models/mixedbread-ai/mxbai-embed-large-v1/pipeline/feature-extraction",
+#     token=os.environ.get('HUGGINGFACEHUB_ACCESS_TOKEN')
+# )
+
+ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 
 # Initiated the models for ideation
 
@@ -78,9 +85,9 @@ improver_llm = ChatOpenAI(
 #     temperature=0.7,
 #     max_tokens=500,
 # )
-ideator_llm = improver_llm
-critic_llm = improver_llm
-validator_llm = improver_llm
+ideator_llm = llm
+critic_llm = llm
+validator_llm = llm
 
 
 # validator_llm = ChatGroq(