Spaces:
Sleeping
Sleeping
import streamlit as st | |
import os | |
import pandas as pd | |
from streamlit_option_menu import option_menu | |
from bardapi import Bard | |
from getvalues import getValues | |
from pymongo import MongoClient | |
from transformers import pipeline, Conversation | |
classifyr = pipeline("zero-shot-classification") | |
convo = pipeline("conversational") | |
# classifi = pipeline(model="facebook/bart-large-mnli") | |
uri = os.environ["MONGO_CONNECTION_STRING"] | |
client = MongoClient(uri, tlsCertificateKeyFile="database/cert.pem") | |
db = client["myapp"] | |
col = db["reminders"] | |
bardkey = os.environ.get("BARD_API_KEY") | |
bard = Bard(token=bardkey) | |
def view_rem(): | |
allrem = list(col.find()) | |
remdata = pd.DataFrame(allrem) | |
st.dataframe(remdata) | |
def Chatbot(): | |
st.title("Chatbot") | |
if user_input := st.chat_input("Describe your goal. e.g: I want to achieve this goal in this time. Be as specific and explanatory as you can."): | |
bardans = bard.get_answer(user_input)['content'] | |
anslist = bard.get_answer(f"Make a list of this answer: \n{bardans} \nfor this goal: \n{user_input}\n\nThe list should be in two section, section 1 for all the reminders to track called Daily Routine and section 2 for all information that should be consumed to achieve the goal and stay very focused and motivated with excitement and this section is called Notes")['content'] | |
# result = classifyr(user_input,candidate_labels=["reminders", "notes"]) | |
with st.chat_message("assistant"): | |
st.write(anslist) | |
# with st.chat_message("user"): | |
# st.write(result["labels"][0]) | |
# if ans["labels"][0] == "reminders": | |
# values = getValues(query.lower()) | |
# with st.chat_message("assistant"): | |
# st.write(values) | |
# col.insert_one(values) | |
# elif ans["labels"][0] == "general conversation": | |
# umsg = bard.get_answer(query)["content"] | |
# with st.chat_message("assistant"): | |
# st.write(umsg) | |
# elif ans["labels"][0] == "notes": | |
# Notes = query.lower().replace( " create a new note", "",).replace(" no new note", "") | |
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")) | |