Newcode / app.py
Ankush05's picture
Update app.py (#4)
b63fd88
raw
history blame contribute delete
No virus
2.47 kB
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"))