import streamlit as st
from rag_pipeline import load_rag_pipeline, ask_question

st.set_page_config(page_title="Medical QA Assistant 💊", layout="centered")
st.title("🧠 Medical QA Assistant")
st.markdown("Ask medical questions and get answers based on PubMed research.")

query = st.text_input("Enter your medical question:")
if query:
    with st.spinner("Searching..."):
        pipe = load_rag_pipeline()
        result = ask_question(pipe, query)
        st.markdown("### 💡 Answer")
        st.write(result)