| 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 | |
| st.set_page_config(layout="wide") | |
| # 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'] | |
| listrem = bard.get_answer(f"Act as a ToDo Reminder AI who sets reminders or daily routine based upon the daily routine provided below:\n{anslist} \n\nMake a list of reminders with exact message, time, repetation frequecy, day/s kind of neccessary detail that would be required to set a reminder notification. Make it a numeric list.")['content'] | |
| # result = classifyr(user_input,candidate_labels=["reminders", "notes"]) | |
| with st.chat_message("assistant"): | |
| st.write(f"What to do to achive the goal:\n{bardans}\n\nHow to do it:\n{anslist}\n\nList of Reminders you should make:\n{listrem}") | |
| # 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")) | |