import streamlit as st
from document_parser import parse_pdf
from rag_qa import get_answer
from summarizer import summarize_text

st.set_page_config(page_title="IntelliDoc.AI", layout="wide")

st.title("📄 IntelliDoc.AI - Smart Document Analyzer")

uploaded_file = st.file_uploader("Upload a PDF document", type="pdf")

if uploaded_file:
    with st.spinner("Parsing document..."):
        extracted_text = parse_pdf(uploaded_file)

    st.subheader("📄 Extracted Text")
    st.text_area("Document Content", extracted_text, height=300)

    if st.button("Summarize"):
        with st.spinner("Summarizing..."):
            summary = summarize_text(extracted_text)
        st.subheader("📝 Summary")
        st.success(summary)

    question = st.text_input("Ask a question based on the document:")
    if st.button("Get Answer"):
        with st.spinner("Searching for answer..."):
            answer = get_answer(extracted_text, question)
        st.subheader("💬 Answer")
        st.info(answer)
else:
    st.info("👆 Please upload a PDF file to begin.")