Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,7 @@ from streamlit_extras.add_vertical_space import add_vertical_space
|
|
6 |
from streamlit_mic_recorder import speech_to_text
|
7 |
from model_pipeline import ModelPipeLine
|
8 |
from q_learning_chatbot import QLearningChatbot
|
|
|
9 |
|
10 |
from gtts import gTTS
|
11 |
from io import BytesIO
|
@@ -101,6 +102,17 @@ def get_text():
|
|
101 |
def generate_response(prompt):
|
102 |
response = mdl.call_conversational_rag(prompt,final_chain)
|
103 |
return response['answer']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
104 |
|
105 |
## Applying the user input box
|
106 |
with input_container:
|
@@ -131,7 +143,7 @@ with input_container:
|
|
131 |
# Retrieve question
|
132 |
if user_sentiment in ["Negative", "Moderately Negative", "Neutral"]:
|
133 |
question = retriever.get_response(
|
134 |
-
user_message
|
135 |
)
|
136 |
st.session_state.asked_questions.append(question)
|
137 |
show_question = True
|
@@ -195,7 +207,7 @@ with input_container:
|
|
195 |
# Retrieve question
|
196 |
if user_sentiment in ["Negative", "Moderately Negative", "Neutral"]:
|
197 |
question = retriever.get_response(
|
198 |
-
user_message
|
199 |
)
|
200 |
st.session_state.asked_questions.append(question)
|
201 |
show_question = True
|
|
|
6 |
from streamlit_mic_recorder import speech_to_text
|
7 |
from model_pipeline import ModelPipeLine
|
8 |
from q_learning_chatbot import QLearningChatbot
|
9 |
+
from retriever import create_vectorstore
|
10 |
|
11 |
from gtts import gTTS
|
12 |
from io import BytesIO
|
|
|
102 |
def generate_response(prompt):
|
103 |
response = mdl.call_conversational_rag(prompt,final_chain)
|
104 |
return response['answer']
|
105 |
+
|
106 |
+
|
107 |
+
def get_response(self, user_query):
|
108 |
+
db=self.create_vectorstore(documents)
|
109 |
+
docs = db.similarity_search(user_query)
|
110 |
+
most_similar_question = docs[0].page_content.split("\n")[0] # Extract the first question
|
111 |
+
if user_query==most_similar_question:
|
112 |
+
most_similar_question=docs[1].page_content.split("\n")[0]
|
113 |
+
|
114 |
+
print(most_similar_question)
|
115 |
+
return most_similar_question
|
116 |
|
117 |
## Applying the user input box
|
118 |
with input_container:
|
|
|
143 |
# Retrieve question
|
144 |
if user_sentiment in ["Negative", "Moderately Negative", "Neutral"]:
|
145 |
question = retriever.get_response(
|
146 |
+
user_message
|
147 |
)
|
148 |
st.session_state.asked_questions.append(question)
|
149 |
show_question = True
|
|
|
207 |
# Retrieve question
|
208 |
if user_sentiment in ["Negative", "Moderately Negative", "Neutral"]:
|
209 |
question = retriever.get_response(
|
210 |
+
user_message
|
211 |
)
|
212 |
st.session_state.asked_questions.append(question)
|
213 |
show_question = True
|