File size: 3,163 Bytes
6608bce |
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 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
import streamlit as st
from transformers import pipeline
import time
# Page configuration
st.set_page_config(page_title="AI Q&A Assistant", layout="wide")
# Custom styling
st.markdown("""
<style>
.header {
background-color: #1565C0;
padding: 60px;
text-align: center;
border-radius: 15px;
color: white;
font-family: 'Arial', sans-serif;
}
.header h1 {
font-size: 50px;
font-weight: bold;
}
.qa-box {
background-color: #f5f5f5;
padding: 20px;
border-radius: 10px;
margin: 10px 0;
}
</style>
<div class="header">
<h1>🤔 AI Q&A Assistant</h1>
<p>Get answers to your questions from any text</p>
</div>
""", unsafe_allow_html=True)
# Load model
@st.cache_resource
def load_qa_model():
return pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
qa_model = load_qa_model()
# Main content
col1, col2 = st.columns([6, 4])
with col1:
st.subheader("Context")
context = st.text_area(
"Enter the text passage:",
height=300,
help="This is the text that the AI will use to answer questions"
)
with col2:
st.subheader("Question")
question = st.text_input(
"Ask a question about the text:",
help="Make sure your question can be answered using the provided text"
)
# History tracking
if 'qa_history' not in st.session_state:
st.session_state.qa_history = []
if qa_model and st.button("Get Answer"):
if context and question:
with st.spinner("Finding answer..."):
start_time = time.time()
result = qa_model(
question=question,
context=context
)
end_time = time.time()
# Add to history
st.session_state.qa_history.append({
'question': question,
'answer': result['answer'],
'confidence': result['score']
})
# Display result
st.markdown(f"""
<div class="qa-box">
<h3>Answer:</h3>
<p>{result['answer']}</p>
<small>Confidence: {result['score']:.2%}</small><br>
<small>Response time: {(end_time - start_time):.2f} seconds</small>
</div>
""", unsafe_allow_html=True)
else:
st.warning("Please provide both context and a question")
# Display history
if st.session_state.qa_history:
st.subheader("Question History")
for i, qa in enumerate(reversed(st.session_state.qa_history[-5:])):
st.markdown(f"""
<div class="qa-box">
<strong>Q: {qa['question']}</strong><br>
A: {qa['answer']}<br>
<small>Confidence: {qa['confidence']:.2%}</small>
</div>
""", unsafe_allow_html=True)
# Clear history button
if st.session_state.qa_history:
if st.button("Clear History"):
st.session_state.qa_history = []
st.experimental_rerun()
|