회사 이름만 입력하면 끝! Python으로 B2B 고객사 조사 자동화하기

아래 내용의 자동화 프로그램을 만들고 싶어.
회사 이름만 입력하면 다음 정보를 자동으로 수집하고 리포트를 생성:
1. 회사 개요 (사업 분야, 규모, 위치 등)
2. 최근 뉴스 및 동향 (최근 3개월)
3. 주요 경쟁사 분석
4. 산업 트렌드
5. 디지털 마케팅 인사이트
6. 영업 접근 포인트 제안

Antrophic API는 과금이 되는 것 같아. 다른 방법으로 구현할 수 없을까?
→ Claude Code가 Google Gemini 무료 API로 자동 전환

#!/usr/bin/env python3
"""
고객사 조사 자동화 도구
B2B 영업을 위한 고객사 조사를 자동화하는 Python 스크립트
Google Gemini API (무료) 사용
"""

import sys
import os
import time
from datetime import datetime

from google import genai
from google.genai import types
from google.genai.errors import ClientError


def check_api_key():
    """API 키 설정 여부 확인"""
    api_key = os.environ.get("GEMINI_API_KEY")
    if not api_key:
        print("❌ GEMINI_API_KEY 환경변수가 설정되지 않았습니다.")
        print()
        print("해결 방법:")
        print('  export GEMINI_API_KEY="your-api-key-here"')
        sys.exit(1)
    return api_key


def build_research_prompt(company_name: str) -> str:
    """고객사 조사용 프롬프트 생성"""
    today = datetime.now().strftime("%Y년 %m월 %d일")

    return f"""당신은 B2B 영업 전문 리서처입니다. "{company_name}"에 대한 고객사 조사 리포트를 작성해주세요.

오늘 날짜: {today}

다음 항목들을 웹 검색을 통해 조사하고, 한국어로 상세한 리포트를 작성해주세요:

## 조사 항목

1. **회사 개요**: 사업 분야, 설립연도, 직원 수, 매출 규모, 본사 위치, 대표이사, 주요 서비스/제품
2. **최근 뉴스 및 동향** (최근 3개월): 신규 서비스 출시, 인수합병, 투자 유치, 인사 변동, 파트너십 등
3. **주요 경쟁사 분석**: 직접 경쟁사 3-5개, 각 경쟁사의 강점/약점, 시장 포지셔닝 비교
4. **산업 트렌드**: 해당 산업의 최신 트렌드, 규제 변화, 기술 동향
5. **디지털 마케팅 인사이트**: 온라인 존재감, 마케팅 채널, 콘텐츠 전략, SNS 활동
6. **영업 접근 포인트**: 잠재적 페인 포인트, 협업 기회, 제안 가능한 솔루션, 접근 전략

## 출력 형식

마크다운 형식으로 작성하되, 다음 구조를 따라주세요:

# {{회사명}} 고객사 조사 리포트

> 작성일: {today}
> 작성 도구: 고객사 조사 자동화 도구 v1.0

---

## 1. 회사 개요
## 2. 최근 뉴스 및 동향
## 3. 주요 경쟁사 분석
## 4. 산업 트렌드
## 5. 디지털 마케팅 인사이트
## 6. 영업 접근 포인트
## 7. 다음 단계 체크리스트

중요: 각 섹션은 구체적인 데이터와 수치를 포함해야 합니다. 추측보다는 검증된 정보를 우선으로 작성하고, 확인이 필요한 정보는 명시적으로 표시해주세요."""


def research_company(company_name: str) -> str:
    """Google Gemini API를 사용하여 고객사 조사 수행"""
    api_key = os.environ.get("GEMINI_API_KEY")
    client = genai.Client(api_key=api_key)

    prompt = build_research_prompt(company_name)

    print(f"🔍 '{company_name}' 조사를 시작합니다...")
    print("   Google 검색 및 AI 분석 중...")
    print()

    # Gemini API 호출 (Google Search 그라운딩 사용, 재시도 포함)
    max_retries = 3
    response = None

    for attempt in range(max_retries):
        try:
            response = client.models.generate_content(
                model="gemini-2.0-flash",
                contents=prompt,
                config=types.GenerateContentConfig(
                    tools=[types.Tool(google_search=types.GoogleSearch())],
                    temperature=0.7,
                    max_output_tokens=8192,
                ),
            )
            break
        except ClientError as e:
            if "429" in str(e) and attempt < max_retries - 1:
                wait = 60 * (attempt + 1)
                print(f"   ⏳ API 할당량 초과. {wait}초 후 재시도합니다... ({attempt + 1}/{max_retries})")
                time.sleep(wait)
            else:
                raise

    report_text = response.text

    # 사용량 정보 출력
    if response.usage_metadata:
        meta = response.usage_metadata
        print(f"   📊 토큰 사용량: 입력 {meta.prompt_token_count:,} / 출력 {meta.candidates_token_count:,}")

    print("   ✅ 조사 완료")
    print()

    return report_text


def save_report(company_name: str, report: str) -> str:
    """리포트를 파일로 저장"""
    output_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "outputs")
    os.makedirs(output_dir, exist_ok=True)

    timestamp = datetime.now().strftime("%Y%m%d_%H%M")
    filename = f"{company_name}_고객조사_{timestamp}.md"
    filepath = os.path.join(output_dir, filename)

    with open(filepath, "w", encoding="utf-8") as f:
        f.write(report)

    return filepath


def main():
    """메인 실행 함수"""
    if len(sys.argv) < 2:
        print("사용법: python research_client_simple.py <회사명>")
        print()
        print("예시:")
        print('  python research_client_simple.py "무신사"')
        print('  python research_client_simple.py "쿠팡"')
        print('  python research_client_simple.py "토스"')
        sys.exit(1)

    company_name = sys.argv[1]
    check_api_key()

    print("=" * 50)
    print("  고객사 조사 자동화 도구 v1.0")
    print(f"  대상: {company_name}")
    print("  모델: Google Gemini 2.0 Flash (무료)")
    print("=" * 50)
    print()

    report = research_company(company_name)
    filepath = save_report(company_name, report)

    print()
    print("=" * 50)
    print("  📄 리포트가 생성되었습니다!")
    print(f"  📁 저장 위치: {filepath}")
    print("=" * 50)


if __name__ == "__main__":
    main()

pip install google-genai

echo 'export GEMINI_API_KEY="발급받은-키"' >> ~/.zshrc
source ~/.zshrc

python3 research_client_simple.py "무신사"

# ~/.zshrc에 추가
연구() {
  python3 ~/Claude_automation/research_client_simple.py "$1"
}