app.py

import asyncio
import os
import streamlit as st
from dotenv import load_dotenv
from pipeline import CDSSPipeline

# Load environment variables
load_dotenv()
api_key = os.getenv("GROQ_API_KEY")

# Initialize pipeline only once per session
@st.cache_resource
def get_pipeline():
    if not api_key:
        st.error("GROQ_API_KEY not found in .env file. Please add it.")
        st.stop()
    return CDSSPipeline(groq_api_key=api_key)

st.set_page_config(page_title="PharmaRAG CDSS", page_icon="💊", layout="wide")

st.title("💊 Clinical Decision Support System (PharmaRAG)")
st.markdown("""
Enter a clinical scenario in natural language below. The system will extract the proposed medication, 
current medications, and patient conditions, then check OpenFDA and RxNorm for interactions and side effects.
""")

pipeline = get_pipeline()

query = st.text_area("Doctor's Clinical Query:", height=150, 
                     placeholder="e.g., I have a 68-year-old patient with atrial fibrillation currently on warfarin. I'm considering prescribing ibuprofen...")

if st.button("Generate Alert", type="primary"):
    if not query.strip():
        st.warning("Please enter a query.")
    else:
        with st.spinner("Analyzing query and retrieving medical facts..."):
            try:
                # Run the async pipeline
                alert = asyncio.run(pipeline.run(query))
                
                st.subheader("Results")
                
                # Show summary and severity
                col1, col2 = st.columns(2)
                with col1:
                    st.info(f"**Query Summary:** {alert.query_summary}")
                with col2:
                    severity_color = {
                        "HIGH": "red",
                        "MODERATE": "orange",
                        "LOW": "green",
                        "UNKNOWN": "gray"
                    }.get(alert.severity_flag.name, "gray")
                    st.markdown(f"**Severity Flag:** <span style='color:{severity_color}; font-weight:bold; font-size:1.2em;'>{alert.severity_flag.value}</span>", unsafe_allow_html=True)
                
                st.markdown("---")
                
                # Show the raw alert text
                st.markdown("### Clinical Alert Synthesis")
                st.markdown(alert.raw_alert_text)
                
                st.markdown("---")
                st.caption(alert.disclaimer)
                st.caption(f"Sources Cited: {', '.join(alert.sources_cited)}")
                
            except Exception as e:
                st.error(f"An error occurred: {e}")