import streamlit as st
from whale_core import parser, agents

st.set_page_config(page_title="HuggingWhale.AI", page_icon="🐋")

st.title("🐋 HuggingWhale.AI")
st.write("Drop your chaos here and we'll pretend to make sense of it...")

uploaded_file = st.file_uploader("📄 Upload your PDF or Text file", type=["pdf", "txt"])

if uploaded_file:
    try:
        doc = parser.parse_file(uploaded_file)
        chunks = parser.chunk_and_embed(doc)

        agent_cfgs = agents.load_agents("config.yaml")
        responses = agents.run_agents_on_text(agent_cfgs, doc)

        st.subheader("🧠 Agent Responses")
        for name, reply in responses.items():
            st.markdown(f"**🤖 {name} says:**\n\n{reply}")

    except Exception as e:
        st.error(f"💥 Oops! Something went wrong:\n{str(e)}")