Update app.py

app.py

@@ -11,7 +11,9 @@ 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"]
@@ -33,23 +35,27 @@ def view_rem():
  
 def Chatbot():
     st.title("Chatbot")
-    if query := st.chat_input("Enter your message"):
-        ans = classifyr(query,candidate_labels=["reminders", "general conversation", "notes"])
-        if ans["labels"][0] == "reminders":
-            values = getValues(query.lower())
-            with st.chat_message("assistant"):
-                st.write(values)
-                col.insert_one(values)
+    if user_input := st.chat_input("Enter your message"):
+        result = classifyr(user_input,candidate_labels=["reminders", "general conversation", "notes", "nature", "technology", "maths", "business", "war"])
+
+        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] == "general conversation":
+        #     umsg = bard.get_answer(query)["content"]
+        #     with st.chat_message("assistant"):
+        #         st.write(umsg)
         
-        elif ans["labels"][0] ==  "notes":
+        # elif ans["labels"][0] ==  "notes":
             
-             Notes = entry.lower().replace( " create a new note", "",).replace(" no new note", "")
+        #      Notes = query.lower().replace( " create a new note", "",).replace(" no new note", "")