Spaces:
Sleeping
Sleeping
File size: 1,226 Bytes
2ddafac 6898f19 43b9079 2ddafac 7e0db7b 2ddafac 43b9079 2ddafac 43b9079 e9cea2c 43b9079 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import streamlit as st
import os
from getvalues import getValues
from pymongo import MongoClient
from transformers import pipeline, Conversation
classifyr = pipeline("zero-shot-classification")
convo = pipeline("conversational")
uri = os.environ["MONGO_CONNECTION_STRING"]
client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem")
db = client["myapp"]
col = db["reminders"]
def Chatbot():
st.title("Chatbot")
if message :=st.chat_input("Enter your message"):
ans = classifyr(message,candidate_labels=["reminders", "general conversation"])
if ans["labels"][0] == "reminders":
values = getValues(x.lower())
with st.chat_message("assistant"):
st.write(values)
col.insert_one(values)
elif ans["labels"][0] == "general conversation":
umsg = bard.get_answer(message)["content"]
with st.chat_message("assistant"):
st.write(umsg)
Chatbot()
def Create_Reminder():
st.title("Create Reminder")
message = st.text_input("Share your plan of today")
time = str(st.time_input("Time"))
date = str(st.date_input("Date")) |