# import streamlit as st
# from transformers import pipeline

# st.title('Question Answering')
# model_checkpoint = 'Diezu/Viedumodel'
# question_answerer = pipeline("question-answering", model=model_checkpoint)

# context = st.text_area('CONTEXT')
# question = st.text_input('QUESTION')

# if st.button('ANSWER'):
#     i=question_answerer(question,context)
#     #st.write(i['answer'])
#     st.markdown(i['answer']) 
#     #st.write('<h1 style="font-size: 50px;">i['answer']</h1>')


import streamlit as st
from transformers import pipeline

# Thiết lập mô hình
st.title('Question Answering')
model_checkpoint = 'Diezu/Viedumodel'
question_answerer = pipeline("question-answering", model=model_checkpoint)

# Thêm Sidebar cho tùy chọn
st.sidebar.title("Cài đặt")
theme_color = st.sidebar.color_picker("Chọn màu chủ đạo", "#4CAF50")
font_size = st.sidebar.slider("Cỡ chữ câu trả lời", 14, 50, 20)

# Di chuyển CONTEXT vào Sidebar
context = st.sidebar.text_area('Nhập Nội Dung (CONTEXT)', height=200)

# Tạo câu hỏi mẫu hoặc cho phép người dùng chọn câu hỏi từ danh sách

question = st.text_input("Nhập câu hỏi ")

# Thêm CSS cho giao diện
st.markdown(
    f"""
    <style>
    .stTitle {{
        font-size: 32px;
        font-weight: bold;
        color: {theme_color};
    }}
    textarea {{
        border: 2px solid {theme_color};
        border-radius: 10px;
        padding: 10px;
        font-size: 16px;
    }}
    input[type="text"] {{
        border: 2px solid {theme_color};
        border-radius: 10px;
        padding: 10px;
        font-size: 16px;
    }}
    .stButton > button {{
        background-color: {theme_color};
        color: white;
        border-radius: 10px;
        padding: 10px 20px;
        font-size: 16px;
        border: none;
        transition: 0.3s;
    }}
    .stButton > button:hover {{
        background-color: #45a049;
    }}
    .stMarkdown {{
        font-size: {font_size}px;
        font-weight: bold;
        color: #333333;
    }}
    </style>
    """,
    unsafe_allow_html=True
)

# Xử lý câu trả lời
if st.button('ANSWER'):
    if context and question:
        with st.spinner('Đang xử lý...'):
            result = question_answerer(question=question, context=context)
        st.success(f"Trả lời: {result['answer']}")
    else:
        st.warning("Vui lòng nhập CONTEXT và chọn câu hỏi.")