Merge branch 'main' of https://huggingface.co/spaces/Nikhil0987/omm

auth.py → app.py

@@ -5,7 +5,7 @@ import json
 import uuid
 
 
-st.set_page_config(page_title="Authentication", page_icon=":guardsman:", layout="wide")
+# st.set_page_config(page_title="Authentication", page_icon=":guardsman:", layout="wide")
 
 # st.title("Authentication")
 

chat.py

@@ -16,13 +16,12 @@ def Chat():
     if ans["scores"][0] > 0.85:
         st.session_state["user"] = "visitor"
         with st.chat_message("assistant"):
-            "You are now sleeping in dream" 
+            "You are now living in dream" 
         st.experimental_rerun()
     else:
         with st.chat_message("assistant"):
             chitchat
         
 
-Chat()
 
 

convo.py

@@ -2,24 +2,26 @@ from transformers import pipeline, Conversation
 import streamlit as st
 from streamlit_option_menu import option_menu
 
+chatbot = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
 
-def convo():
-    convo = st.text_input("Enter your message")
-    chatbot = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
-    conversation = Conversation(convo)
+def Convo():
+    if convo := st.chat_input("Enter your message"):
+    
+        conversation = Conversation(convo)
 
 
 
     # candidate_labels = ["HELP", "PROBLEM SOLVE", "GENERAL TALK"]
   
    
-    ans = chatbot(conversation)
+        ans = chatbot(conversation)
     # add_user_input = st.button("Add User Input")
 
     # conversation.add_user_input("{}".format(convo))
     # conversation = chatbot(conversation)
-    # conversation.generated_responses[-1]
-    st.write(ans)
+    with st.chat_message("assistant"):
+        ans.generated_responses[-1]
+    # st.write(ans)
 
 
-convo()
+# convo()

home.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from streamlit_option_menu import option_menu
 from chat import Chat
-from convo import convo
+from convo import Convo
 
 def homepage():
     st.title("Home")   
@@ -22,7 +22,7 @@ def dashboard():
     elif selected == "Chat":
         Chat()
     elif selected == "Conversation": 
-        convo()
+        Convo()
     elif selected == "Logout":
         st.session_state["user"] = "visitor"
         st.experimental_rerun()

requirements.txt

@@ -0,0 +1,5 @@
+streamlit
+streamlit-option-menu
+transformers
+dnspython
+torch