Update phase1_agents.py

phase1_agents.py

@@ -5,6 +5,9 @@ from phi.tools.duckduckgo import DuckDuckGo
 from phi.model.openai import OpenAIChat
 from pydantic import BaseModel, Field
 from fastapi import UploadFile
+from rich.live import Live
+from rich.console import Console
+from rich.errors import LiveError
 
 # Load environment variables (API keys, etc.)
 from dotenv import load_dotenv
@@ -31,7 +34,7 @@ def search_company(company_name: str) -> dict:
     """ Searches for detailed company information using web search. """
     query = f"Find details for {company_name}, including its official website, mission, services, and AI-related initiatives. Include sources."
     response = company_search_agent.print_response(query)
-    return {"company_name": company_name, "details": response}
+    return {"company_name": company_name, "details": response.content}
 
 
 ##############################
@@ -45,10 +48,17 @@ firecrawl_agent = Agent(
     markdown=True,
 )
 
-def scrape_website(url: str) -> dict:
-    """ Scrapes relevant company information from the given website. """
-    response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
-    return {"website": url, "scraped_data": response}
+def scrape_website(url):
+    try:
+        console = Console()
+        with Live(console=console, refresh_per_second=10) as live_log:
+            response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
+    except LiveError:
+        # If Live() is already active, print a simple log instead
+        response = firecrawl_agent.print_response(f"Extract business details from {url}, including mission, services, and AI-related information.")
+    
+    return response.content
+
 
 
 ##############################
@@ -66,7 +76,7 @@ text_processing_agent = Agent(
 def process_company_description(text: str) -> dict:
     """ Summarizes the user-provided company description. """
     response = text_processing_agent.print_response(f"Summarize the following description: {text}. Focus on mission, key services, industry, and AI potential.")
-    return {"user_description": text, "summary": response}
+    return {"user_description": text, "summary": response.content}
 
 
 ##############################