| import streamlit as st | |
| from rag_pipeline import generate_answer | |
| from rag_pipeline import index, corpus_embeddings, corpus, embed_model | |
| st.set_page_config(page_title="Medical QA Assistant 🧬", page_icon="🧠") | |
| st.title("🧠 Medical QA Assistant") | |
| st.write("Ask any medical question and get a science-based answer from PubMed!") | |
| query = st.text_input("Enter your medical question:") | |
| if st.button("Get Answer") and query: | |
| answer = generate_answer(query, index, corpus_embeddings, corpus, embed_model) | |
| st.success(answer) | |