Ankush05 commited on
Commit
cf012cb
·
1 Parent(s): 8d6936c
Files changed (2) hide show
  1. app.py +10 -5
  2. requirements.txt +0 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
- from streamlit_option_menu import option_menu
3
  from pymongo import MongoClient
4
  import os
5
- import pandas as pd
 
6
 
7
 
8
  # Connecting to MongoDB
@@ -16,7 +16,12 @@ col = db["chatbot"]
16
 
17
  def chatbot():
18
  st.title("ChatBot")
19
- message = st.text_input("Enter your query here")
20
- if st.button ("Submit"):
21
- col.insert_one({"message": message})
 
 
 
 
 
22
 
 
1
  import streamlit as st
 
2
  from pymongo import MongoClient
3
  import os
4
+ from transformers import pipeline, Conversation
5
+
6
 
7
 
8
  # Connecting to MongoDB
 
16
 
17
  def chatbot():
18
  st.title("ChatBot")
19
+ uinput = st.chat_input("Enter your message")
20
+ if uinput:
21
+ st.chat(uinput)
22
+ pipe = Conversation()
23
+ response = pipe(uinput)
24
+ st.chat(response)
25
+ print(response)
26
+
27
 
requirements.txt CHANGED
@@ -1,7 +1,5 @@
1
  streamlit
2
  streamlit-option-menu
3
- pandas
4
  pymongo
5
  transformers[torch]
6
- sentence-transformers
7
  dnspython
 
1
  streamlit
2
  streamlit-option-menu
 
3
  pymongo
4
  transformers[torch]
 
5
  dnspython