Update Home.py

Home.py

@@ -1,22 +1,29 @@
 import streamlit as st 
 from streamlit_option_menu import option_menu
+from transformers import pipeline, Conversation
 
+pipe = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
 
-def homepage():
-    st.write("Timeline")
-    # allmessages =[]
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
 
-    if usrmsg := st.chat_input("Share a thought"):
-        st.session_state.messages.append(usrmsg)
-        with st.chat_message("user"):
-            st.session_state.messages
+# def homepage():
+#     st.write("Timeline")
+#     # allmessages =[]
+#     if "messages" not in st.session_state:
+#         st.session_state.messages = []
+
+#     if usrmsg := st.chat_input("Share a thought"):
+#         st.session_state.messages.append(usrmsg)
+#         with st.chat_message("user"):
+#             st.session_state.messages
     
     
 def chat():
-    st.write("Chat")
-    st.write("Welcome to the chat page")
+    if query := st.chat_input("Enter your message"):
+        uquery = Conversation(query)
+        response = pipe(uquery)
+        with st.chat_message("assistant"):
+            response.generated_responses[-1]
+        
     
 # def invoke_document():
 #     st.write("Invoke Document")
@@ -44,11 +51,11 @@ def chat():
 def dashboard():
     
     with st.sidebar:
-        selected = option_menu(None, ['Home', 'Chat', "Logout"],
-                               icons=['🏠', '💬', "📄"])
-    if selected == 'Home':
-        homepage()
-    elif selected == 'Chat':
+        selected = option_menu(None, [Chat', "Logout"],
+                               icons=[💬', "📄"])
+    # if selected == 'Home':
+    #     homepage()
+    if selected == 'Chat':
         chat()
     elif selected == 'Logout':
         st.session_state.user = None