Create app.py

app.py

@@ -0,0 +1,126 @@
+"""
+Rizzy v0.2 – Stream-friendly, streak-building dating-chat ghost-writer
+──────────────────────────────────────────────────────────────────────
+▪ Streams Dobby-Unhinged replies live
+▪ Lets the user pick a vibe (playful / sweet / roast)
+▪ Keeps only the last few turns to save tokens
+▪ Uses env var or Streamlit Secrets for the Fireworks key
+▪ Click-to-copy output (JS hack)
+"""
+
+from __future__ import annotations
+import os, json, requests, datetime
+from typing import List, Dict
+
+import streamlit as st
+
+# ── 🔑  API CONFIG ──────────────────────────────────────────────────
+API_KEY = os.getenv("FIREWORKS_API_KEY") or st.secrets["fireworks_key"]
+MODEL_ID = "accounts/sentientfoundation/models/dobby-unhinged-llama-3-3-70b-new"
+FW_URL   = "https://api.fireworks.ai/inference/v1/chat/completions"
+
+# ── 📝  SYSTEM PROMPT TEMPLATE  ────────────────────────────────────
+BASE_PROMPT = """
+You are **Rizzy**, an outrageously charismatic ghost-writer for dating apps.
+
+Output **exactly**:
+  [Say:"<max-20-token reply>"]  <≤7-word reason>
+
+Rules
+1. Tone: {tone}. Abbreviate (ur, tbh, lol).
+2. Never exceed 2 sentences / 20 tokens.
+3. No hashtags, no emoji spam (≤2).
+4. If you break a rule, forcibly truncate.
+"""
+
+# ── 🏷️  VIBE PICKER ────────────────────────────────────────────────
+TONE2TXT = {
+    "Playful":   "friendly, jokey, flirty",
+    "Sweet":     "warm, encouraging, wholesome",
+    "Roast":     "teasing, sarcastic, cocky-fun"
+}
+
+# ── 🔌  FIREWORKS STREAM CALL  ─────────────────────────────────────
+def stream_dobby(history: List[Dict], temperature: float = 0.8):
+    payload = {
+        "model": MODEL_ID,
+        "stream": True,
+        "temperature": temperature,
+        "top_p": 1,
+        "top_k": 40,
+        "messages": history,
+    }
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Accept": "application/json",
+        "Content-Type": "application/json",
+    }
+
+    with requests.post(FW_URL, headers=headers, data=json.dumps(payload), stream=True, timeout=60) as r:
+        r.raise_for_status()
+        for line in r.iter_lines():
+            if not line or line.startswith(b":"):  # skip keep-alive
+                continue
+            data = json.loads(line.decode().removeprefix("data: "))
+            delta = data["choices"][0]["delta"].get("content")
+            if delta:
+                yield delta
+
+# ── 🖼️  UI  SETUP ──────────────────────────────────────────────────
+st.set_page_config("Rizzy", "💬")
+st.title("💬 Rizzy – your AI wing-person")
+
+# streak badge (simple local)
+today = datetime.date.today()
+if st.session_state.get("last_seen") == today:
+    streak = st.session_state.get("streak", 1) + 1
+else:
+    streak = 1
+st.session_state["last_seen"] = today
+st.session_state["streak"] = streak
+st.markdown(f"**🔥 Streak: {streak} day(s)**")
+
+tone = st.radio("Pick the vibe:", list(TONE2TXT.keys()), horizontal=True)
+
+# Chat history container
+if "chat" not in st.session_state:
+    st.session_state.chat: List[Dict] = [
+        {"role":"assistant","content":"Yo, I'm Rizzy. Drop a dating-app situation and I’ll craft the perfect reply."}
+    ]
+
+# Print past messages
+for msg in st.session_state.chat:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
+
+# ── 💬  USER INPUT ────────────────────────────────────────────────
+prompt = st.chat_input("What's going on?")
+if prompt:
+    st.session_state.chat.append({"role":"user","content":prompt})
+    with st.chat_message("user"): st.markdown(prompt)
+
+    # Build trimmed history (last 6 turns to save tokens)
+    history = [{"role":"system","content": BASE_PROMPT.format(tone=TONE2TXT[tone])}]
+    history += st.session_state.chat[-12:]
+
+    # Stream assistant reply
+    reply_box = st.chat_message("assistant")
+    placeholder  = reply_box.empty()
+    full_reply   = ""
+
+    try:
+        for chunk in stream_dobby(history):
+            full_reply += chunk
+            placeholder.markdown(full_reply)
+    except Exception as e:
+        full_reply = f"*Error talking to Dobby →* `{e}`"
+        placeholder.markdown(full_reply)
+
+    st.session_state.chat.append({"role":"assistant","content":full_reply})
+
+    # Click-to-copy (JS hack)
+    copy_code = f"""
+    <textarea id="toCopy" style="opacity:0">{full_reply}</textarea>
+    <button onclick="navigator.clipboard.writeText(document.getElementById('toCopy').value);this.innerText='Copied!'">📋 Copy reply</button>
+    """
+    reply_box.components.v1.html(copy_code, height=35, scrolling=False)