llm.py

from groq import Groq
import os

client = Groq(
    api_key=os.getenv("GROQ_API_KEY")
)

def generate_instagram_captions(description):

    prompt = f"""
You are a social media caption generator.

Based on this image description:
{description}

Generate 7 short, cool Instagram captions.

Rules:
- Max 8 words
- Stylish and aesthetic
- No numbering
- Each caption on a new line
"""

    chat_completion = client.chat.completions.create(
        messages=[
            {"role": "user", "content": prompt}
        ],
        model="llama3-70b-8192",
    )

    captions = chat_completion.choices[0].message.content

    return captions.split("\n")