Create aap.py
Browse files
aap.py
ADDED
|
@@ -0,0 +1,172 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# app.py
|
| 2 |
+
import os
|
| 3 |
+
from crewai import Agent, Task, Crew
|
| 4 |
+
from crewai_tools import DirectoryReadTool, FileReadTool, SerperDevTool, BaseTool
|
| 5 |
+
from langchain_groq import ChatGroq
|
| 6 |
+
import gradio as gr
|
| 7 |
+
|
| 8 |
+
# Set environment variables for API keys
|
| 9 |
+
os.environ["SERPER_API_KEY"] = "3c75331dffc120acfa03b3bc75a4fbb3202c4927"
|
| 10 |
+
os.environ['GROQ_API_KEY'] = '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'
|
| 11 |
+
|
| 12 |
+
# Initialize the ChatGroq model
|
| 13 |
+
llm = ChatGroq(
|
| 14 |
+
api_key=os.getenv('GROQ_API_KEY'),
|
| 15 |
+
model="mixtral-8x7b-32768"
|
| 16 |
+
)
|
| 17 |
+
|
| 18 |
+
# Define the agents
|
| 19 |
+
class SalesAgents:
|
| 20 |
+
def __init__(self):
|
| 21 |
+
self.llm = llm
|
| 22 |
+
|
| 23 |
+
def sales_rep_agent(self):
|
| 24 |
+
return Agent(
|
| 25 |
+
role="Sales Representative",
|
| 26 |
+
goal="Identify high-value leads that match our ideal customer profile",
|
| 27 |
+
backstory=(
|
| 28 |
+
"As a key member of the dynamic sales team at COMET Estimating LLC, your mission is to navigate the construction and "
|
| 29 |
+
"estimating sectors to uncover promising leads. Leveraging state-of-the-art tools and a strategic approach, "
|
| 30 |
+
"you delve into data, market trends, and industry interactions to identify opportunities that others might miss. "
|
| 31 |
+
"Your role is essential in fostering valuable connections and driving the growth of our estimating services, "
|
| 32 |
+
"ensuring that general contractors and subcontractors benefit from precise, actionable insights and expert support."
|
| 33 |
+
),
|
| 34 |
+
allow_delegation=False,
|
| 35 |
+
verbose=True,
|
| 36 |
+
llm=self.llm,
|
| 37 |
+
max_iter=5,
|
| 38 |
+
memory=True
|
| 39 |
+
)
|
| 40 |
+
|
| 41 |
+
def lead_sales_rep_agent(self):
|
| 42 |
+
return Agent(
|
| 43 |
+
role="Lead Sales Representative",
|
| 44 |
+
goal="Nurture leads with personalized, compelling communications",
|
| 45 |
+
backstory=(
|
| 46 |
+
"Within the dynamic landscape of crewAI's sales department, you excel as the crucial link "
|
| 47 |
+
"between prospective clients and the estimating solutions they require. By crafting tailored, engaging communications, "
|
| 48 |
+
"you not only showcase our expert services but also ensure that general contractors and subcontractors feel understood and valued. "
|
| 49 |
+
"Your role is key in transforming interest into tangible outcomes, guiding leads from initial curiosity through to securing our "
|
| 50 |
+
"estimating services."
|
| 51 |
+
),
|
| 52 |
+
allow_delegation=False,
|
| 53 |
+
verbose=True,
|
| 54 |
+
llm=self.llm,
|
| 55 |
+
max_iter=5,
|
| 56 |
+
memory=True
|
| 57 |
+
)
|
| 58 |
+
|
| 59 |
+
# Define the tools
|
| 60 |
+
directory_read_tool = DirectoryReadTool(directory='./content')
|
| 61 |
+
file_read_tool = FileReadTool()
|
| 62 |
+
search_tool = SerperDevTool()
|
| 63 |
+
|
| 64 |
+
class SentimentAnalysisTool(BaseTool):
|
| 65 |
+
name: str = "Sentiment Analysis Tool"
|
| 66 |
+
description: str = ("Analyzes the sentiment of text to ensure positive and engaging communication according to context.")
|
| 67 |
+
|
| 68 |
+
def _run(self, text: str) -> str:
|
| 69 |
+
# Custom sentiment analysis logic
|
| 70 |
+
return "positive"
|
| 71 |
+
|
| 72 |
+
sentiment_analysis_tool = SentimentAnalysisTool()
|
| 73 |
+
|
| 74 |
+
# Define the tasks
|
| 75 |
+
class SalesTasks:
|
| 76 |
+
def lead_profiling_task(self):
|
| 77 |
+
return Task(
|
| 78 |
+
description=(
|
| 79 |
+
"""
|
| 80 |
+
Conduct a thorough analysis of [Lead Name], a general contractor or subcontractor in the construction sector that
|
| 81 |
+
recently expressed interest in our estimating services. Utilize all available data sources, including industry databases,
|
| 82 |
+
company websites, LinkedIn, and recent news articles, to compile a comprehensive profile. This profile should include:
|
| 83 |
+
|
| 84 |
+
1. Identification of key decision-makers within the company, such as project managers or procurement officers.
|
| 85 |
+
2. A summary of recent project developments or contracts that the company is involved in.
|
| 86 |
+
3. An assessment of their specific needs related to estimating services, such as budgeting, cost forecasting, or bidding accuracy.
|
| 87 |
+
|
| 88 |
+
Ensure that all the information used is accurate and verified, as this analysis will be crucial for customizing our
|
| 89 |
+
engagement strategy to effectively meet the company’s needs.
|
| 90 |
+
"""
|
| 91 |
+
),
|
| 92 |
+
expected_output=(
|
| 93 |
+
"A detailed report on {lead_name}, covering company background, key personnel and specific needs related to estimating services. "
|
| 94 |
+
"Identify potential areas where our estimating solutions can add value and propose customized engagement strategies."
|
| 95 |
+
),
|
| 96 |
+
tools=[directory_read_tool, file_read_tool, search_tool],
|
| 97 |
+
agent=SalesAgents().sales_rep_agent()
|
| 98 |
+
)
|
| 99 |
+
|
| 100 |
+
def personalized_outreach_task(self):
|
| 101 |
+
return Task(
|
| 102 |
+
description=(
|
| 103 |
+
"""
|
| 104 |
+
Work at an estimating company where you are tasked with targeting general contractors or subcontractors in the construction sector.
|
| 105 |
+
Using insights from sales representative agents and the lead profiling report on {lead_name}, create a personalized outreach campaign for targeting {key_decision_maker},
|
| 106 |
+
the {position} of {lead_name}. The campaign should address their recent project developments and demonstrate how our estimating services can enhance their operations.
|
| 107 |
+
Your communication must align with {lead_name}'s company culture and demonstrate a clear understanding of their needs and objectives.
|
| 108 |
+
Ensure all information is accurate and verified.
|
| 109 |
+
|
| 110 |
+
In the dynamic world of sales, you act as the crucial link between potential clients and the services they need.
|
| 111 |
+
By crafting tailored, engaging communications, you not only showcase the strengths of the services offered but also ensure that clients feel understood and valued,
|
| 112 |
+
particularly when targeting general contractors or subcontractors in the construction sector.
|
| 113 |
+
Your role is key in transforming interest into actionable outcomes, guiding leads from initial curiosity to securing the service offered.
|
| 114 |
+
Adapt these strategies to effectively connect with any company looking to engage with similar clients.
|
| 115 |
+
"""
|
| 116 |
+
),
|
| 117 |
+
expected_output=(
|
| 118 |
+
"A series of customized email drafts for {lead_name}, specifically directed at {key_decision_maker}. "
|
| 119 |
+
"Each draft should clearly connect our estimating services with their project developments and goals. "
|
| 120 |
+
"The tone should be professional, engaging, and consistent with {lead_name}'s company culture."
|
| 121 |
+
),
|
| 122 |
+
tools=[sentiment_analysis_tool, search_tool],
|
| 123 |
+
agent=SalesAgents().lead_sales_rep_agent()
|
| 124 |
+
)
|
| 125 |
+
|
| 126 |
+
# Create the crew
|
| 127 |
+
crew = Crew(
|
| 128 |
+
agents=[
|
| 129 |
+
SalesAgents().sales_rep_agent(),
|
| 130 |
+
SalesAgents().lead_sales_rep_agent()
|
| 131 |
+
],
|
| 132 |
+
tasks=[
|
| 133 |
+
SalesTasks().lead_profiling_task(),
|
| 134 |
+
SalesTasks().personalized_outreach_task()
|
| 135 |
+
],
|
| 136 |
+
verbose=2,
|
| 137 |
+
memory=True
|
| 138 |
+
)
|
| 139 |
+
|
| 140 |
+
# Define the Gradio app
|
| 141 |
+
def run_tasks(lead_name, industry, key_decision_maker, position, milestone):
|
| 142 |
+
# Set up inputs for the crew
|
| 143 |
+
inputs = {
|
| 144 |
+
"lead_name": lead_name,
|
| 145 |
+
"industry": industry,
|
| 146 |
+
"key_decision_maker": key_decision_maker,
|
| 147 |
+
"position": position,
|
| 148 |
+
"milestone": milestone
|
| 149 |
+
}
|
| 150 |
+
|
| 151 |
+
# Run the crew tasks
|
| 152 |
+
result = crew.kickoff(inputs=inputs)
|
| 153 |
+
return result
|
| 154 |
+
|
| 155 |
+
# Create the Gradio interface
|
| 156 |
+
iface = gr.Interface(
|
| 157 |
+
fn=run_tasks,
|
| 158 |
+
inputs=[
|
| 159 |
+
gr.Textbox(label="Lead Name", placeholder="Enter the company name"),
|
| 160 |
+
gr.Textbox(label="Industry", placeholder="Enter the industry"),
|
| 161 |
+
gr.Textbox(label="Key Decision Maker", placeholder="Enter the name of the key decision maker"),
|
| 162 |
+
gr.Textbox(label="Position", placeholder="Enter the position of the key decision maker"),
|
| 163 |
+
gr.Textbox(label="Milestone", placeholder="Enter the milestone")
|
| 164 |
+
],
|
| 165 |
+
outputs="text",
|
| 166 |
+
title="Sales Task Runner",
|
| 167 |
+
description="Enter the company details to run the sales tasks and get the results."
|
| 168 |
+
)
|
| 169 |
+
|
| 170 |
+
# Launch the Gradio app
|
| 171 |
+
if __name__ == "__main__":
|
| 172 |
+
iface.launch()
|