Update app.py

app.py

@@ -1,126 +1,70 @@
-"""
-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
+import requests
+import json
+from typing import List, Dict
 
-# ── 🔑  API CONFIG ──────────────────────────────────────────────────
-API_KEY = os.getenv("FIREWORKS_API_KEY") or st.secrets["fireworks_key"]
+# --------------------- API CONFIG ---------------------
+API_KEY = "fw_3ZVyvgMy2KqPc5RhPJuXp8rJ"  # or set env FIREWORKS_API_KEY
 MODEL_ID = "accounts/sentientfoundation/models/dobby-unhinged-llama-3-3-70b-new"
-FW_URL   = "https://api.fireworks.ai/inference/v1/chat/completions"
+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.
+# --------------------- LLM CALL -----------------------
 
-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):
+def ask_dobby(history: List[Dict], max_tokens: int = 150, temperature: float = 0.9) -> str:
     payload = {
         "model": MODEL_ID,
-        "stream": True,
+        "max_tokens": max_tokens,
         "temperature": temperature,
         "top_p": 1,
         "top_k": 40,
         "messages": history,
     }
     headers = {
-        "Authorization": f"Bearer {API_KEY}",
         "Accept": "application/json",
         "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}",
     }
+    r = requests.post(FW_URL, headers=headers, data=json.dumps(payload), timeout=60)
+    r.raise_for_status()
+    return r.json()["choices"][0]["message"]["content"].strip()
 
-    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 ──────────────────────────────────────────────────
+# --------------------- 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."}
-    ]
+    st.session_state.chat: List[Dict] = []
 
-# Print past messages
+SYSTEM_PROMPT = (
+    "You are a dating friend, designed to be a tinder/hinge/bumble chat response generator, an outrageously charismatic friend who gives authentic, witty, flirty lines."
+    "It should be SHORT, easygoing, light, concise, simple, casual, carefree, cool. The key is not to try too hard."
+    "NEVER be generic. Keep suggestions SUPER SHORT and text acronym all the time (like 'ur' instead of 'your', m2, omw, etc.). Only give one answer. 2 sentences at max, ever."
+    'For your answers: give the text message the user needs to paste to the dating app FIRST, and then maybe your super short reasoning. Always start with: [Say:"the txt message"], then maybe your super short description.'
+)
+
+# print existing chat
 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:]
+if not st.session_state.chat:
+    welcome = "👋 Yo, I'm Rizzy. I'm here to help you chat for your dating life 🔥."
+    st.session_state.chat.append({"role": "assistant", "content": welcome})
+    st.chat_message("assistant").markdown(welcome)
 
-    # Stream assistant reply
-    reply_box = st.chat_message("assistant")
-    placeholder  = reply_box.empty()
-    full_reply   = ""
+user_input = st.chat_input("What's the situation or reply?")
+if user_input is not None:
+    # append user message
+    st.session_state.chat.append({"role": "user", "content": user_input})
+    with st.chat_message("user"):
+        st.markdown(user_input)
 
+    # build history (system + last 10 turns)
+    history = [{"role": "system", "content": SYSTEM_PROMPT}] + st.session_state.chat[-10:]
     try:
-        for chunk in stream_dobby(history):
-            full_reply += chunk
-            placeholder.markdown(full_reply)
+        answer = ask_dobby(history)
     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})
+        answer = f"[Error contacting Dobby: {e}]"
 
-    # 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)
+    st.session_state.chat.append({"role": "assistant", "content": answer})
+    with st.chat_message("assistant"):
+        st.markdown(answer)