Update app.py

app.py

@@ -82,71 +82,32 @@ class NewMessage(BaseModel):
     text: str = Field(..., min_length=1, max_length=5000)
 
 @app.get("/chat/{video_id}")
-async def get_messages(video_id: str, limit: int = 50, since: Optional[str] = None):
-    """
-    Fetch messages for a video.
-    - limit: max messages returned (default 50)
-    - since: ISO8601 timestamp; return only messages with created_at > since
-    """
-    limit = max(1, min(limit, 200))
-    path = _chat_file_for(video_id)
-    items = await _read_jsonl(path)
-
-    # Filter by 'since' if provided
-    if since:
-        try:
-            since_dt = datetime.fromisoformat(since.replace("Z", "+00:00"))
-            def _is_new(msg):
-                try:
-                    ts = datetime.fromisoformat(str(msg.get("created_at", "")).replace("Z", "+00:00"))
-                    return ts > since_dt
-                except Exception:
-                    return False
-            items = list(filter(_is_new, items))
-        except Exception:
-            # If bad 'since', just ignore it
-            pass
-
-    # Return the most recent 'limit' messages (sorted ascending by created_at)
-    items.sort(key=lambda m: m.get("created_at", ""))
-    if len(items) > limit:
-        items = items[-limit:]
-
-    return {"video_id": video_id, "count": len(items), "messages": items}
+async def get_messages(video_id: str):
+    messages = app.state.chat_storage.get(video_id, [])
+    return {
+        "video_id": video_id,
+        "count": len(messages),
+        "messages": messages
+    }
+
 
 @app.post("/chat/{video_id}")
-async def post_message(video_id: str, msg: NewMessage, request: Request):
-    """
-    Append a message to a video's chat.
-    Body: { "author": "Ross", "text": "hello" }
-    """
-    path = _chat_file_for(video_id)
-    lock = _lock_for(path)
-
-    author = _valid_author(msg.author)
-    text = _valid_text(msg.text)
-    if not text:
-        return {"ok": False, "error": "Empty message"}
-
-    created = _now_iso()
-    # Lightweight id; good enough for demo
-    mid = hashlib.sha1(f"{video_id}|{author}|{created}|{text}".encode("utf-8")).hexdigest()[:16]
-    ip = request.client.host if request and request.client else None
-
-    record = {
-        "id": mid,
+async def send_message(video_id: str, msg: ChatMessage, request: Request):
+    if video_id not in app.state.chat_storage:
+        app.state.chat_storage[video_id] = []
+
+    message = {
+        "id": uuid.uuid4().hex[:16],
         "video_id": video_id,
-        "author": author,
-        "text": text,
-        "created_at": created,
-        "ip": ip,
-        "ua": request.headers.get("user-agent", "")[:200]
+        "author": msg.user,  # ✅ FIXED: use provided username
+        "text": msg.text,
+        "created_at": datetime.utcnow().isoformat(),
+        "ip": request.client.host if request.client else "unknown",
+        "ua": request.headers.get("user-agent", "")
     }
 
-    async with lock:
-        await _append_jsonl(path, record)
-
-    return {"ok": True, "message": record}
+    app.state.chat_storage[video_id].append(message)
+    return {"status": "ok", "message": message}
 
 # ---------------------------
 # (Your original iCloud album endpoints — unchanged)