Update app.py

app.py

@@ -60,6 +60,13 @@ def generate_ai_stream(message: str):
 
     return partial.strip()
 
+ final_text = partial.strip() + " full generated"
+
+    if message in db:
+        db[message]["reply"] = final_text
+        db[message]["status"] = "done"
+
+    return final_text
 
 # 🔥 WORKER
 def worker():