# -*- coding: utf-8 -*-
import requests
import json

OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
MODEL_NAME = "gemma3"  # 설치한 모델명과 맞춰주세요.

prompt = """
너는 부동산 숏츠 전문 대본 작가다.

아래 정보를 바탕으로 유튜브 쇼츠 30초 대본을 작성해줘.

조건:
1. 첫 문장은 강한 후킹 문장
2. 부동산 초보도 이해하기 쉽게
3. 과장 표현 금지
4. 상담 유도 문장 포함
5. 한국어 자연스럽게
6. 300자 이내

블로그 분석 정보:
- 지역: 인천 부평구
- 매물 유형: 신축 빌라
- 핵심 키워드: 역세권, 주차 가능, 생활권 우수, 실거주 추천
- 목적: 블로그 유입 및 상담 문의 유도
"""

payload = {
    "model": MODEL_NAME,
    "prompt": prompt,
    "stream": False
}

try:
    response = requests.post(OLLAMA_URL, json=payload, timeout=120)
    response.raise_for_status()

    data = response.json()
    result = data.get("response", "")

    print("\n===== AI 숏츠 대본 결과 =====\n")
    print(result)

except requests.exceptions.ConnectionError:
    print("Ollama 서버에 연결할 수 없습니다.")
    print("먼저 PowerShell에서 ollama run gemma3 를 실행해보세요.")

except requests.exceptions.Timeout:
    print("요청 시간이 초과되었습니다. 모델이 너무 무겁거나 PC 성능이 부족할 수 있습니다.")

except Exception as e:
    print("오류 발생:", e)