Update app.py

app.py

@@ -62,14 +62,12 @@ with generative:
 
 with chat:
     st.title("AI Chat")
+    messages = st.container(height="600")
     prompt_gpt = st.chat_input("Ask something")
     if prompt_gpt:
-        user_msg = st.chat_message("user")
-        user_msg.write(prompt_gpt)
+        user_msg = messages.chat_message("user").write(prompt_gpt)
         try:
-            resp = Client.create_completion("gpt", prompt)
-            message_out = st.chat_message("assistant")
-            message_out.write(resp)
+            resp = Client.create_completion("gpt3", prompt)
+            message_out = messages.chat_message("assistant").write(resp)
         except Exception as e:
-            message_out = st.chat_message("assistant")
-            message_out.write(e)
+            message_out = messages.chat_message("assistant").write(e)