Update app.py

app.py

@@ -48,7 +48,7 @@ if userinput := st.chat_input("Hi"):
 
     
     answer = model.generate_content(newuserinputstring)
-
     
-    # with st.chat_message("Assistant"):
-    #     st.write(completetion.text)
+    
+    with st.chat_message("Assistant"):
+        st.write(answer)