You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 

33 lines
1.2 KiB

from summarizer.fetcher import fetch_web_content, format_links
from summarizer.summarizer import get_relevant_links
from summarizer.brochure import generate_brochure
import logging
logger = logging.getLogger(__name__)
def main():
company_name = input("Enter company name: ") or "HuggingFace"
url = input("Enter company website: ") or "https://huggingface.co"
model_choice = input("Enter LLM model (default:deepseek-r1:1.5B, gpt-4): ") or "deepseek-r1:1.5B"
provider_choice = input("Enter provider (openai/ollama(ollama_lib/ollama_api), default: ollama_lib): ") or "ollama_api"
logger.info(f"Fetching links from {url}...")
links = fetch_web_content(url)
if not links:
logger.error("No links found. Exiting...")
return
formatted_links = format_links(url, links)
logger.info(f"Extracted and formatted {len(formatted_links)} links.")
relevant_links = get_relevant_links(company_name, formatted_links, model=model_choice, provider=provider_choice)
logger.info("Filtered relevant links.")
brochure = generate_brochure(company_name, relevant_links, model=model_choice, provider=provider_choice)
print("\nGenerated Brochure:\n")
print(brochure)
if __name__ == "__main__":
main()