Update app.py

app.py

@@ -235,41 +235,15 @@ st.markdown("""
 st.title("🧠MitraVerse")
 
 # st.set_page_config(page_title="Tool Button Executor", layout="wide")
-st.title("🧠 Agent Tools – Run by Button")
+# st.title("🧠 Agent Tools – Run by Button")
 
 # User input for all tools
-user_prompt = st.text_area("📝 Enter your prompt / code / query:")
+# 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:
@@ -294,13 +268,54 @@ if "latest_code" in st.session_state:
 
 
 # Custom Chat Input Box UI
+# with st.container():
+#     with st.form("chat_form", clear_on_submit=True):
+#         st.markdown('<div id="floating-container">', unsafe_allow_html=True)        
+#         st.markdown('</div>', unsafe_allow_html=True) 
+#         user_input = st.text_input("Ask me", label_visibility="collapsed", placeholder="Ask me Anything")
+
+#         submitted = st.form_submit_button(label = "Send")
+        
+#         if submitted and user_input:
+#             st.session_state.chat_history.append(HumanMessage(content=user_input))
+        
+#             config = {"configurable": {"thread_id": st.session_state.thread_id}}
+#             state_input = {
+#                 "messages": st.session_state.chat_history,
+#                 "input": user_input,
+#             }
+        
+#             result = graph.invoke(state_input, config=config)
+            
+            # if result.get("code"):
+            #     st.session_state.latest_code = result["code"]
+            #     st.session_state.chat_history.append(
+            #         AIMessage(content="**💻 Generated Code:**\n\n```python\n" + result["code"] + "\n```")
+            #     )
+            
+            # if result.get("execution_result"):
+            #     st.session_state.chat_history.append(
+            #         AIMessage(content="**🧪 Code Execution Result:**\n\n" + result["execution_result"])
+            #     )
+            
+            # if result.get("explanation"):
+            #     st.session_state.latest_explanation = result["explanation"]
+            #     st.session_state.chat_history.append(
+            #         # AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"])
+            #         AIMessage(content="**🔍 Code Explanation:**\n\n```\n" + result["explanation"] + "\n```")
+            #     )
+            
+            # elif result.get("messages"):
+            #     st.session_state.chat_history.append(result["messages"][-1])
+            # st.rerun()
+            # st.markdown("<script>window.scrollTo(0, document.body.scrollHeight);</script>", unsafe_allow_html=True)
 with st.container():
     with st.form("chat_form", clear_on_submit=True):
         st.markdown('<div id="floating-container">', unsafe_allow_html=True)        
         st.markdown('</div>', unsafe_allow_html=True) 
-        user_input = st.text_input("Ask me", label_visibility="collapsed", placeholder="Ask me Anything")
 
-        submitted = st.form_submit_button(label = "Send")
+        user_input = st.text_input("Ask me", label_visibility="collapsed", placeholder="Ask me Anything")
+        submitted = st.form_submit_button(label="Send")
         
         if submitted and user_input:
             st.session_state.chat_history.append(HumanMessage(content=user_input))
@@ -323,16 +338,51 @@ with st.container():
                 st.session_state.chat_history.append(
                     AIMessage(content="**🧪 Code Execution Result:**\n\n" + result["execution_result"])
                 )
-            
+
             if result.get("explanation"):
                 st.session_state.latest_explanation = result["explanation"]
                 st.session_state.chat_history.append(
-                    # AIMessage(content="**🔍 Code Explanation:**\n\n" + result["explanation"])
                     AIMessage(content="**🔍 Code Explanation:**\n\n```\n" + result["explanation"] + "\n```")
                 )
             
             elif result.get("messages"):
                 st.session_state.chat_history.append(result["messages"][-1])
+
             st.rerun()
             st.markdown("<script>window.scrollTo(0, document.body.scrollHeight);</script>", unsafe_allow_html=True)
 
+# ================================
+# TOOL BUTTONS SECTION
+# ================================
+col1, col2, col3 = st.columns(3)
+
+user_prompt = st.session_state.get("latest_code", "") or user_input  # fallback to user_input if needed
+
+with st.container():
+    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.")
+