Update app.py

app.py

@@ -38,6 +38,7 @@ def search(query: str):
 
 tools = [search, multiply]
 tool_map = {tool.name: tool for tool in tools}
+
 model_with_tools = model.bind_tools(tools)
 
 # Define Agent State class
@@ -104,43 +105,38 @@ def render_graph_nx():
     st.pyplot(plt)
 
 # Streamlit UI
-st.title("Blah Blah Demo")
+st.title("LLM Tool Workflow Demo")
 st.write("Compare results **with and without human intervention** in the workflow.")
 
 # Sidebar for configuration
 st.sidebar.header("Configuration")
 st.session_state['human_loop'] = st.sidebar.checkbox("Enable Human-in-the-Loop", value=False)
 
-# Input and comparison mode
+# Input prompt
 prompt = st.text_input("Enter your question:", "What is 24 * 365?")
 if st.button("Run Workflow"):
     st.subheader("Execution Results")
 
-    # Without human-in-the-loop
-    st.markdown("### Without Human-in-the-Loop")
-    st.session_state['human_loop'] = False
-    start_time = time.time()
-    try:
-        intermediate_outputs = []
-        for s in compiled_app.stream({"messages": [prompt]}):
-            intermediate_outputs.append(s)
-        st.write("Response:", intermediate_outputs[-1]['messages'][0])
-    except Exception as e:
-        st.error(f"Error: {e}")
-    st.write(f"Execution Time: {time.time() - start_time:.2f} seconds")
-
-    # With human-in-the-loop
-    st.markdown("### With Human-in-the-Loop")
-    st.session_state['human_loop'] = True
-    start_time = time.time()
-    try:
+    # Run workflow without and with human-in-the-loop
+    for human_loop in [False, True]:
+        st.markdown(f"### {'With' if human_loop else 'Without'} Human-in-the-Loop")
+        st.session_state['human_loop'] = human_loop
         intermediate_outputs = []
-        for s in compiled_app.stream({"messages": [prompt]}):
-            intermediate_outputs.append(s)
-        st.write("Response:", intermediate_outputs[-1]['messages'][0])
-    except Exception as e:
-        st.error(f"Error: {e}")
-    st.write(f"Execution Time: {time.time() - start_time:.2f} seconds")
+        start_time = time.time()
+
+        try:
+            # Collect intermediate outputs
+            for s in compiled_app.stream({"messages": [prompt]}):
+                intermediate_outputs.append(s)
+            
+            # Extract final response
+            response = intermediate_outputs[-1]['messages'][0].content
+            st.write("Response:", response)
+        except Exception as e:
+            st.error(f"Error: {e}")
+        
+        st.write(f"Execution Time: {time.time() - start_time:.2f} seconds")
+        st.write("---")
 
     # Display Workflow Graph
     st.subheader("Workflow Graph")