Update app.py

app.py

@@ -132,8 +132,10 @@ async def echo(websocket):
         )["answer"]
         if response.startswith("? "):
             response.replace("? ", "")
-        if response.lower().replace("\n", "").startswith("assistant: ") or response.lower().replace("\n", "").startswith("ai: "):
-            response.replace("Assistant: ", "").replace("AI: ", "")
+        if "assistant: " in response.lower() or "ai: " in response.lower():
+            response.replace("Assistant:", "").replace("AI:", "")
+        response.replace("\n", "")
+        response.strip()
         await websocket.send(json.dumps({"response": response}))
 
 async def main():