# app.py
import streamlit as st
from langchain.llms import OpenAI
from langchain.agents import initialize_agent, AgentType, load_tools
import os

st.set_page_config(page_title="🧠 AI Task Agent")

st.title("🤖 AI Task Agent – Powered by LangChain")

# Get API key from Hugging Face Secrets
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")

llm = OpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)
tools = load_tools(["serpapi", "llm-math"], llm=llm)

agent = initialize_agent(
    tools=tools,
    llm=llm,
    agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
    verbose=False
)

# Streamlit input/output
user_query = st.text_input("Ask me anything...")

if user_query:
    with st.spinner("Thinking..."):
        response = agent.run(user_query)
    st.success(response)