Update app.py

app.py

@@ -1,13 +1,30 @@
 import streamlit as st
-from freeGPT import Client
+import logging
+from bing_chat import UnlimBotChat
 
 st.title("ChatGPT4 UI")
 cont = st.container(height=600)
 prompt = st.chat_input(placeholder="Eg. How are you?")
 
-if prompt:
-    output = Client.create_completion("gpt3", prompt)
-    user_pr = cont.chat_message("user")
-    user_pr.write(prompt)
-    output_msg = cont.chat_message("assistant")
-    output_msg.write(output)
+logging.basicConfig(level=logging.INFO)
+
+chat = UnlimBotChat("")
+
+def main():
+    st.title("ChatGPT4 UI")
+    cont = st.container(height=600)
+    prompt = st.chat_input(placeholder="Eg. How are you?")
+    
+    if st.button("Send"):
+        messages = chat.run(prompt)
+
+        for message in messages:
+            if message['source'] == 'user':
+                user_msg = cont.chat_message("user")
+                user_msg.write(message['message'])
+            elif message['source'] == 'bing':
+                out_msg = cont.chat_message("assistant")
+                out_msg.write(message['message'])
+
+if __name__ == "__main__":
+    main()