from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

# Load model and tokenizer
model = AutoModelForQuestionAnswering.from_pretrained("spyrosbriakos/greek_legal_bert_v2")
tokenizer = AutoTokenizer.from_pretrained("spyrosbriakos/greek_legal_bert_v2")

# Create QA pipeline
qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)

# Title of the app
st.title("Question Answering in Greek")

# Input fields for context and question
context = st.text_area("Context", "Provide the context here...")
question = st.text_input("Question", "Ask your question here...")

# Generate answer when the user presses the button
if st.button("Get Answer"):
    if context and question:
        result = qa_pipeline(question=question, context=context)
        st.write(f"Answer: {result['answer']}")
    else:
        st.write("Please provide both context and a question.")