Update app.py

app.py

@@ -1,22 +1,14 @@
 import streamlit as st
-from streamlit_option_menu import option_menu
-from pymongo import MongoClient
 import os 
-import pandas as pd
-
-
-# Connecting to MongoDB
-uri = os.environ["MONGO_CONNECTION_STRING"]
-
-client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem")
-
-db = client["myapp"]
-col = db["chatbot"]
+from transformers import pipeline, Conversation
 
+convo = pipeline("facebook/blenderbot-400M-distill")
 
 def chatbot():
     st.title("ChatBot")
-    message = st.text_input("Enter your query here")
-    if st.button ("Submit"):
-        col.insert_one({"message": message})
+    if message := st.chat_input("Enter your query"):
+        umsg = Conversation(message)
+        answer = convo(umsg)
+        with st.chat_message("Assistant"):
+            st.write(answer)