Update app.py

app.py

@@ -76,7 +76,7 @@ def chat_gen(message, history, return_buffer=True):
         }
         ai_msg = conv_chain.invoke(state)
         print(ai_msg)
-        for token in ai_msg["answer"]:
+        for token in ai_msg:
             buffer += token
             yield buffer