Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 from transformers import pipeline
 from bardapi import Bard 
 import os 
+from getvalues import getValues
 
 bardkey = os.environ.get("BARD_API_KEY")
 
@@ -15,12 +16,16 @@ classifi = pipeline(model="facebook/bart-large-mnli")
 
 def chatbot():
     st.title("ChatBot")
-    if message := st.chat_input("Enter your message"):
+    if query := st.chat_input("Enter your message"):
         ans = classifi(message, candidate_labels=["Reminder", "General Conversation"])
-        umsg = bard.get_answer(message)["content"]
-        with st.chat_message("assistant"):
-            st.write(umsg)
-            st.write(ans)
+        if ans["labels"][0] == "Reminder":
+            values = getValues(query)
+            with st.chat_message("assistant"):
+                st.write(values)
+        elif ans["labels"][0] == "General Conversation":
+            umsg = bard.get_answer(message)["content"]
+            with st.chat_message("assistant"):
+                st.write(umsg)