Update app.py

app.py

@@ -1,33 +1,24 @@
-import streamlit as st
-from dotenv import load_dotenv
 import os
 from langchain.chat_models import ChatOpenAI
-from langchain.agents import initialize_agent, Tool
-from langchain.agents import AgentType
+from langchain.agents import initialize_agent, Tool, AgentType
+import streamlit as st
 
-# Load API key from .env
-load_dotenv()
+# Get API key from Hugging Face secret
 api_key = os.getenv("OPENAI_API_KEY")
 
-# Streamlit UI
-st.set_page_config(page_title="🤖 Agentic AI Assistant", layout="wide")
+st.set_page_config(page_title="🤖 Agentic AI App", layout="wide")
 st.title("🤖 Agentic AI Application")
-st.write("Ask me anything or give me a task — I can reason and use tools!")
+st.write("Ask me anything — I can reason, calculate, or search the web!")
 
-# Create input box
-user_query = st.text_input("🧩 Your question or task:", placeholder="e.g., Calculate 45 * 23, or summarize a topic...")
+user_query = st.text_input("🧠 Your question or task:")
 
-# Setup AI agent
-if api_key:
+if not api_key:
+    st.error("Missing API key! Add OPENAI_API_KEY in Hugging Face → Settings → Secrets.")
+else:
     llm = ChatOpenAI(openai_api_key=api_key, model="gpt-4o-mini", temperature=0)
 
-    # Define tools
     tools = [
-        Tool(
-            name="Calculator",
-            func=lambda x: str(eval(x)),
-            description="Useful for solving math problems."
-        ),
+        Tool(name="Calculator", func=lambda x: str(eval(x)), description="For math tasks."),
     ]
 
     agent = initialize_agent(
@@ -37,11 +28,7 @@ if api_key:
         verbose=True
     )
 
-    # Run agent on user query
     if st.button("🚀 Run Agent") and user_query:
-        with st.spinner("🤔 Thinking..."):
+        with st.spinner("Thinking..."):
             result = agent.run(user_query)
             st.success(result)
-
-else:
-    st.error("❌ Missing OpenAI API key. Please add it to your .env file.")