File size: 2,510 Bytes
d827268 82e59d7 d827268 4cdb788 d827268 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
import streamlit as st
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity
import numpy as np
# Load Q&A data
def load_data():
return pd.read_csv('qna.csv', encoding='latin1', delimiter=';')
def get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
new_tfidf = vectorizer.transform([new_sentence])
similarities = cosine_similarity(new_tfidf, tfidf_matrix)
most_similar_index = np.argmax(similarities)
similarity_percentage = similarities[0, most_similar_index] * 100
return answers[most_similar_index], similarity_percentage
def answer_the_question(new_sentence, vectorizer, tfidf_matrix, questions, answers):
most_similar_answer, similarity_percentage = get_most_similar_question(new_sentence, vectorizer, tfidf_matrix, questions, answers)
if similarity_percentage > 70:
return most_similar_answer
else:
return 'Sorry, I am not aware of this information :('
def main():
st.markdown(
"<h1 style='text-align: center; color: #269A96; font-size: 70px;'>Unicorn</h1>",
unsafe_allow_html=True
)
# Load Q&A data
data = load_data()
questions = data['question'].tolist()
answers = data['answer'].tolist()
# Create TF-IDF vectorizer
vectorizer = TfidfVectorizer()
tfidf_matrix = vectorizer.fit_transform(questions)
# Get user input
user_question = st.text_input("Ask me a question:", key="user_question")
# Apply CSS to style the text input field
st.markdown(
"""
<style>
.stTextInput > div > div > div > input {
color: #000000 !important; /* Text color set to black */
background-color: #247370 !important;
}
</style>
""",
unsafe_allow_html=True
)
# Process the question when submitted
if st.button("Submit", key="submit_button", help="Submit your question"):
if user_question:
response = answer_the_question(user_question, vectorizer, tfidf_matrix, questions, answers)
st.write(response)
else:
st.warning("Please ask a question.")
# Apply CSS to style the button
st.markdown(
"""
<style>
.css-1bkz6v6 {
background-color: #a5eae9;
color: #FFFFFF;
}
</style>
""",
unsafe_allow_html=True
)
if __name__ == "__main__":
main()
|