Update app.py

app.py

@@ -4,6 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from jobspy import scrape_jobs
+import pandas as pd
 
 from Gradio_UI import GradioUI
 
@@ -33,6 +35,39 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def fetch_jobs(search_term: str, location: str, results: int) -> str:
+    """A tool that fetches recent job postings from job boards using JobSpy.
+    
+    Args:
+        search_term: The job title or keywords to search for (e.g., 'data scientist').
+        location: The geographic location to filter jobs by (e.g., 'New York, NY' or 'Remote').
+        results: The number of job postings to retrieve (e.g., 10 or 20).
+    
+    Returns:
+        A string containing a formatted list of job titles, companies, locations, and links.
+        Each job is shown on its own line.
+    """
+    jobs = scrape_jobs(
+        site_name=["indeed", "linkedin", "zip_recruiter", "glassdoor", "google", "bayt", "naukri"],
+        search_term=search_term,
+        location=location,
+        results_wanted=results,
+        hours_old=72,
+        country_indeed='USA',
+        # linkedin_fetch_description=True # gets more info such as description, direct job url (slower)
+        # proxies=["208.195.175.46:65095", "208.195.175.45:65095", "localhost"],
+    )
+    
+    # Prepare a human-readable response
+    jobs_subset = jobs[["title", "company", "location", "job_url"]]
+    lines = [
+        f"{row.title} at {row.company} ({row.location})\n{row.job_url}"
+        for _, row in jobs_subset.iterrows()
+    ]
+    return "\n\n".join(lines) if lines else "No jobs found for the given criteria."
+
+duckduck_tool = DuckDuckGoSearchTool()
 
 final_answer = FinalAnswerTool()
 
@@ -46,8 +81,6 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
 custom_role_conversions=None,
 )
 
-duckduck_tool = DuckDuckGoSearchTool()
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -56,7 +89,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[duckduck_tool, final_answer], ## add your tools here (don't remove final answer)
+    tools=[duckduck_tool, fetch_jobs, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,