admin-valentin/awesome-llm-apps
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

final

Browse source
This commit is contained in:
Madhu 2025-01-16 19:26:53 +05:30
parent 07d62f26ee
commit c0a001656e
1 changed files with 15 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
ai_agent_tutorials/ai_lead_generation_agent/ai_lead_generation_agent.py
View file

@ -22,7 +22,7 @@ class QuoraPageSchema(BaseModel):
interactions: List[QuoraUserInteractionSchema] = Field(description="List of all user interactions (questions and answers) on the page") interactions: List[QuoraUserInteractionSchema] = Field(description="List of all user interactions (questions and answers) on the page")
# Step 1: Search for relevant URLs using Firecrawl # Step 1: Search for relevant URLs using Firecrawl
def search_for_urls(company_description, firecrawl_api_key): def search_for_urls(company_description, firecrawl_api_key, num_links):
print("Step 1: Searching for relevant URLs using Firecrawl...") print("Step 1: Searching for relevant URLs using Firecrawl...")
url = "https://api.firecrawl.dev/v1/search" url = "https://api.firecrawl.dev/v1/search"
headers = { headers = {
@ -32,7 +32,7 @@ def search_for_urls(company_description, firecrawl_api_key):
query1 = f"quora websites where people are looking for {company_description} services" query1 = f"quora websites where people are looking for {company_description} services"
payload = { payload = {
"query": query1, "query": query1,
"limit": 3, # Adjust the limit as needed "limit": num_links, # Use the num_links parameter here
"lang": "en", "lang": "en",
"location": "United States", "location": "United States",
"timeout": 60000, "timeout": 60000,
@ -151,12 +151,23 @@ def main():
st.title("AI Lead Generation Agent") st.title("AI Lead Generation Agent")
st.info("This firecrawl powered agent helps you generate leads from Quora by searching for relevant posts and extracting user information.") st.info("This firecrawl powered agent helps you generate leads from Quora by searching for relevant posts and extracting user information.")
# Sidebar for API keys # Sidebar for API keys and number of links
with st.sidebar: with st.sidebar:
st.header("API Keys") st.header("API Keys")
firecrawl_api_key = st.text_input("Firecrawl API Key", type="password") firecrawl_api_key = st.text_input("Firecrawl API Key", type="password")
st.caption(" Get your Firecrawl API key from [Firecrawl's website](https://www.firecrawl.dev/app/api-keys)")
openai_api_key = st.text_input("OpenAI API Key", type="password") openai_api_key = st.text_input("OpenAI API Key", type="password")
st.caption(" Get your Composio API key from [Composio's website](https://composio.ai)")
composio_api_key = st.text_input("Composio API Key", type="password") composio_api_key = st.text_input("Composio API Key", type="password")
st.caption(" Get your OpenAI API key from [OpenAI's website](https://platform.openai.com/api-keys)")
# Add a numeric input for the number of links
num_links = st.number_input("Number of links to search", min_value=1, max_value=10, value=3)
# Reset button
if st.button("Reset"):
st.session_state.clear()
st.experimental_rerun()
# Main input for company description # Main input for company description
company_description = st.text_input("Enter your company description or the niche you want to find leads in:", placeholder="e.g. AI voice cloning, Video Generation AI tools") company_description = st.text_input("Enter your company description or the niche you want to find leads in:", placeholder="e.g. AI voice cloning, Video Generation AI tools")
@ -166,7 +177,7 @@ def main():
st.error("Please fill in all the API keys and the company description.") st.error("Please fill in all the API keys and the company description.")
else: else:
with st.spinner("Searching for relevant URLs..."): with st.spinner("Searching for relevant URLs..."):
urls = search_for_urls(company_description, firecrawl_api_key) urls = search_for_urls(company_description, firecrawl_api_key, num_links) # Pass num_links to search_for_urls
if urls: if urls:
st.subheader("Quora Links Used:") st.subheader("Quora Links Used:")