Update app.py

app.py

@@ -4,18 +4,10 @@ from freeGPT import Client
 st.title("ChatGPT4 UI")
 cont = st.container(height=400)
 prompt = st.chat_input(placeholder="Eg. How are you?")
-num_user_msg = 0
-num_ai_msg = 0
-ai_msg = []
-user_msg = []
 
 if prompt:
-    num_user_msg += 1
-    num_ai_msg += 1
     output = Client.create_completion("gpt3", prompt)
     user_pr = cont.chat_message
-    user_msg.append(user_pr)
-    user_msg[num_user_msg-1].write(prompt)
+    user_pr.write(prompt)
     output_msg = cont.chat_message("assistant")
-    ai_msg.append(output_msg)
-    ai_msg[num_ai_msg-1].write(output)
+    output_msg.write(output)