Update app.py

app.py

@@ -13,6 +13,7 @@ import os
 import re
 import uuid
 from gtts import gTTS
+from bson import ObjectId
 
 # app = FastAPI()
 app = Flask(__name__)
@@ -481,6 +482,56 @@ def save_donation():
         return {"error": f"An error occurred: {str(e)}"}, 500
 
 
+@app.get('/chats/<username>')
+def list_chats(username):
+    items = list(chat_collection.find({'username': username}, {'_id':1,'user_data':1,'bot_data':1,'timestamp':1,'category':1}).sort('timestamp', -1))
+    out = []
+    for it in items:
+        out.append({
+            'id': str(it['_id']),
+            'title': (it.get('user_data') or 'Chat')[:40],
+            'timestamp': it.get('timestamp'),
+            'category': it.get('category','General')
+        })
+    return jsonify(out), 200
+
+@app.patch('/chat/<chat_id>/rename')
+def rename_chat(chat_id):
+    title = (request.get_json() or {}).get('title')
+    if not title: return jsonify({'error':'title required'}), 400
+    chat_collection.update_one({'_id': ObjectId(chat_id)}, {'$set': {'title': title}})
+    return jsonify({'ok': True}), 200
+
+@app.delete('/chat/<chat_id>')
+def delete_chat(chat_id):
+    chat_collection.delete_one({'_id': ObjectId(chat_id)})
+    return jsonify({'ok': True}), 200
+
+# Simple memory K/V
+mem = client['user_memory_db']
+mem_col = mem['memory']
+
+@app.post('/memory/<username>')
+def save_memory(username):
+    data = request.get_json() or {}
+    key, value = data.get('key'), data.get('value')
+    if not key: return jsonify({'error':'key required'}), 400
+    mem_col.update_one({'username': username, 'key': key},
+                       {'$set': {'value': value, 'updated': datetime.datetime.now()}},
+                       upsert=True)
+    return jsonify({'ok': True}), 200
+
+@app.get('/memory/<username>')
+def get_memory(username):
+    items = list(mem_col.find({'username': username}, {'_id':0}))
+    return jsonify(items), 200
+
+@app.delete('/memory/<username>/<key>')
+def delete_memory(username, key):
+    mem_col.delete_one({'username': username, 'key': key})
+    return jsonify({'ok': True}), 200
+
+    
         
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)