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")) | |