publisher/metadata_engine.py

from publisher.ai.gemini_client import get_model


def generate_metadata(video_path: str):

    model = get_model()

    prompt = f"""
    Generate viral short-form video metadata.
    Source: {video_path}

    Return JSON:
    title
    description
    hook
    audience
    """

    response = model.generate_content(prompt)

    text = response.text.strip()

    return {
        "metadata": text
    }