Update app.py

app.py

@@ -2,8 +2,8 @@ import os
 import streamlit as st
 import requests
 
-# ── Config ────────────────────────────────────────────────────────────────
-HF_TOKEN      = os.getenv("HF_TOKEN") 
+# ── Configuration ──────────────────────────────────────────────────────────
+HF_TOKEN      = os.getenv("HF_TOKEN")  # read‐only token in Space secrets
 MODEL_ID      = "sourize/phi2-memory-lora"
 CONTEXT_TURNS = 6
 
@@ -32,22 +32,24 @@ def query_hf(prompt: str) -> str:
     }
     r = requests.post(API_URL, headers=HEADERS, json=payload, timeout=60)
     r.raise_for_status()
-    out = r.json()[0]["generated_text"].strip()
+    data = r.json()
+    text = data[0]["generated_text"].strip()
     for m in ("User:", "Assistant:"):
-        if m in out:
-            out = out.split(m)[0].strip()
-    return out or "I don't know."
+        if m in text:
+            text = text.split(m)[0].strip()
+    return text or "I don't know."
 
 # ── Streamlit UI ──────────────────────────────────────────────────────────
 st.set_page_config(layout="centered")
-st.title("🧠 DeepTalks (API)")
+st.title("🧠 DeepTalks (Inference API)")
 st.subheader("Your personal AI Companion")
 
 if "history" not in st.session_state:
     st.session_state.history = []
 
-for role, txt in st.session_state.history:
-    st.chat_message("user" if role=="You" else "assistant").write(txt)
+# Render chat history
+for role, msg in st.session_state.history:
+    st.chat_message("user" if role=="You" else "assistant").write(msg)
 
 user_input = st.chat_input("Type your message…")
 if user_input:
@@ -55,7 +57,7 @@ if user_input:
     st.session_state.history.append(("You", user_input))
 
     recent = st.session_state.history[-CONTEXT_TURNS*2:]
-    context = "\n".join(msg for _, msg in recent)
+    context = "\n".join(t for _, t in recent)
     prompt = (
         f"{SYSTEM_PROMPT}\n\n"
         f"Context:\n{context}\n\n"