Spaces:
Sleeping
Sleeping
File size: 2,909 Bytes
fd88516 b64f151 fd88516 b64f151 fd88516 b64f151 fd88516 | 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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | import os, requests, json
import streamlit as st
from dotenv import load_dotenv
load_dotenv()
# Get API key from Streamlit secrets or environment
def get_api_key():
try:
if hasattr(st, 'secrets') and "DEEPSEEK_API_KEY" in st.secrets:
return st.secrets["DEEPSEEK_API_KEY"]
except:
pass
return os.getenv("DEEPSEEK_API_KEY")
# Don't call get_api_key() at import time - call it when needed
DEEPSEEK_API_KEY = None
BASE = os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com")
def chat(messages, model="deepseek-chat", temperature=0.9):
api_key = get_api_key()
if not api_key:
raise ValueError("DEEPSEEK_API_KEY not found. Please set it in Hugging Face Space secrets or environment variables.")
headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
payload = {"model": model, "messages": messages, "temperature": temperature}
r = requests.post(f"{BASE}/chat/completions", headers=headers, data=json.dumps(payload), timeout=60)
r.raise_for_status()
return r.json()["choices"][0]["message"]["content"]
def generate_scripts(persona, boundaries, content_type, tone, refs, n=6):
system = (
"You write Instagram-compliant, suggestive-but-not-explicit Reels briefs. "
"Use tight hooks, concrete visual beats, clear CTAs. Avoid explicit sexual terms. "
"Return ONLY JSON: an array of length N, each with {title,hook,beats,voiceover,caption,hashtags,cta}."
)
user = f"""
Persona: {persona}
Boundaries: {boundaries}
Content type: {content_type} | Tone: {tone} | Duration: 15–25s
Reference snippets (inspire, don't copy):
{chr(10).join(f"- {r}" for r in refs)}
N = {n}
JSON array ONLY.
"""
out = chat([{"role":"system","content":system},{"role":"user","content":user}])
# Be lenient if model wraps JSON with text
start = out.find("[")
end = out.rfind("]")
return json.loads(out[start:end+1])
def revise_for(prompt_label, draft: dict, guidance: str):
system = f"You revise scripts to {prompt_label}. Keep intent; return ONLY JSON with the same schema."
user = json.dumps({"draft": draft, "guidance": guidance})
out = chat([{"role":"system","content":system},{"role":"user","content":user}], temperature=0.6)
start = out.find("{")
end = out.rfind("}")
return json.loads(out[start:end+1])
def selective_rewrite(draft: dict, field: str, snippet: str, prompt: str):
system = "You rewrite only the targeted snippet inside the specified field. Keep style. Return ONLY JSON."
user = json.dumps({"field": field, "snippet": snippet, "prompt": prompt, "draft": draft})
out = chat([{"role":"system","content":system},{"role":"user","content":user}], temperature=0.7)
start = out.find("{")
end = out.rfind("}")
return json.loads(out[start:end+1])
|