ChatBot / app.py
Ankush05's picture
This
cf012cb
raw
history blame
551 Bytes
import streamlit as st
from pymongo import MongoClient
import os
from transformers import pipeline, Conversation
# Connecting to MongoDB
uri = os.environ["MONGO_CONNECTION_STRING"]
client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem")
db = client["myapp"]
col = db["chatbot"]
def chatbot():
st.title("ChatBot")
uinput = st.chat_input("Enter your message")
if uinput:
st.chat(uinput)
pipe = Conversation()
response = pipe(uinput)
st.chat(response)
print(response)