roshithindia commited on
Commit
f136beb
1 Parent(s): 8a81974

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ import pandas as pd
3
+
4
+ qa_pipeline = pipeline("question-answering", model='deepset/roberta-base-squad2')
5
+
6
+ def chatbot(question):
7
+ with open(r"ayurdata.txt", "r", encoding="utf-8") as file:
8
+ context = file.read()
9
+ answer = qa_pipeline(question=question, context=context)
10
+ return answer
11
+
12
+ def prints(questions):
13
+ response = chatbot(questions)
14
+ return response['answer']
15
+
16
+ import streamlit as st
17
+ st.title("AyurEasy AI Bot")
18
+ if "messages" not in st.session_state:
19
+ st.session_state.messages = []
20
+ st.session_state.messages.append({
21
+ 'role':'assistant',
22
+ 'content':"Hi! I'm your AI Bot"
23
+ })
24
+ for message in st.session_state.messages:
25
+ with st.chat_message(message["role"]):
26
+ st.markdown(message["content"])
27
+ prompt = st.chat_input("What is up?")
28
+ if prompt:
29
+ with st.chat_message("user"):
30
+ st.markdown(prompt)
31
+ st.session_state.messages.append({"role":"user","content":prompt})
32
+ response = f"ChatBot: {prints(prompt)}"
33
+ with st.chat_message("assistant"):
34
+ st.markdown(response)
35
+ st.session_state.messages.append({"role":"assistant","content":response})