Update app.py

app.py

@@ -12,7 +12,7 @@ from langgraph.graph.message import add_messages
 from typing import Annotated
 from typing_extensions import TypedDict
 from langchain_together import Together
-from tools import execute_python_code
+from tools import execute_python_code, web_search, deep_think
 import io
 import contextlib
 import traceback
@@ -233,6 +233,43 @@ st.markdown("""
 """, unsafe_allow_html=True)
  
 st.title("🧠MitraVerse")
+
+st.set_page_config(page_title="Tool Button Executor", layout="wide")
+st.title("🧠 Agent Tools – Run by Button")
+
+# User input for all tools
+user_prompt = st.text_area("📝 Enter your prompt / code / query:")
+
+# Columns for button layout
+col1, col2, col3 = st.columns(3)
+
+# Execution section
+if col1.button("⚙️ Run Python Code"):
+    if user_prompt:
+        with st.spinner("Executing your Python code..."):
+            result = execute_python_code.invoke({"code": user_prompt})
+            st.success("✅ Output:")
+            st.code(result, language="python")
+    else:
+        st.warning("Please enter Python code in the input box.")
+
+if col2.button("🌐 Web Search"):
+    if user_prompt:
+        with st.spinner("Searching the web..."):
+            result = web_search.invoke({"query": user_prompt})
+            st.success("🔎 Search Result:")
+            st.write(result)
+    else:
+        st.warning("Please enter a search query.")
+
+if col3.button("🧠 Deep Think"):
+    if user_prompt:
+        with st.spinner("Thinking deeply..."):
+            result = deep_think.invoke({"prompt": user_prompt})
+            st.success("🧠 Reasoned Output:")
+            st.write(result)
+    else:
+        st.warning("Please enter a prompt.")
  
 # Initialize session
 if "thread_id" not in st.session_state: