File size: 1,933 Bytes
58ac83f 2a95781 58ac83f a274967 58ac83f c103376 58ac83f c103376 a274967 58ac83f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import os
import requests
GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
GROQ_API_KEY = os.getenv("GROQ_API_KEY") # Secure key from Hugging Face Secrets
if not GROQ_API_KEY:
raise ValueError("GROQ_API_KEY is missing! Set it in Hugging Face Secrets.")
# Helper function to load system prompt from txt files
def load_prompt(filename):
with open(f'config/system_prompts/{filename}', 'r') as file:
return file.read()
# Function to call Groq API with system prompt and user input
def call_groq(system_prompt, user_input):
headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
payload = {
"model": "mixtral-8x7b-32768",
"messages": [
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_input}
]
}
print("Sending payload to Groq API...")
print(payload)
response = requests.post(GROQ_API_URL, json=payload, headers=headers)
print("Groq API response status:", response.status_code)
print("Groq API response body:", response.text)
if response.status_code != 200:
raise Exception(f"Groq API error: {response.status_code} - {response.text}")
data = response.json()
if 'choices' not in data or not data['choices']:
raise Exception("Groq API returned no choices.")
return data['choices'][0]['message']['content']
def ad_copy_agent(product, description, audience, tone):
system_prompt = load_prompt("ad_copy_prompt.txt")
user_input = f"Product: {product}\nDescription: {description}\nAudience: {audience}\nTone: {tone}"
return call_groq(system_prompt, user_input)
def sentiment_agent(social_data):
system_prompt = load_prompt("sentiment_prompt.txt")
return call_groq(system_prompt, social_data)
def timing_agent(platform):
system_prompt = load_prompt("timing_prompt.txt")
return call_groq(system_prompt, f"Platform: {platform}") |