Update app.py

app.py

@@ -23,6 +23,7 @@ if not tavily_api_key:
     raise ValueError("Missing required environment variable: TAVILY_API_KEY")
 
 # ------------------- Tool Definitions -------------------
+# Tavily Search Tool
 tavily_tool = TavilySearchResults(max_results=5)
 
 def multiply(a: int, b: int) -> int:
@@ -61,24 +62,22 @@ app_graph.add_conditional_edges("assistant", tools_condition)
 app_graph.add_edge("tools", "assistant")
 react_graph = app_graph.compile()
 
-# ------------------- Save Graph Visualization -------------------
+# Save graph visualization as an image
 with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmpfile:
-    graph_image_path = tmpfile.name
     graph = react_graph.get_graph(xray=True)
-    graph.draw_mermaid_png(tmpfile.name)
+    tmpfile.write(graph.draw_mermaid_png())  # Write binary image data to file
+    graph_image_path = tmpfile.name
 
 # ------------------- Streamlit Interface -------------------
-st.title("ReAct Agent")
+st.title("ReAct Agent with Arithmetic and Search")
 
 # Display the workflow graph
 st.header("LangGraph Workflow Visualization")
 st.image(graph_image_path, caption="LangGraph Workflow Visualization")
 
 # Prompt user for inputs
-user_question = st.text_area(
-    "Enter your question:",
-    placeholder="Example: 'Add 3 and 4. Multiply the result by 2. Divide it by 5.'"
-)
+user_question = st.text_area("Enter your question:",
+                             placeholder="Example: 'Add 3 and 4. Multiply the result by 2. Divide it by 5.'")
 
 if st.button("Submit"):
     if not user_question.strip():
@@ -96,7 +95,7 @@ if st.button("Submit"):
 
     st.success("Processing complete!")
 
-# Example Questions
+# Example Placeholder Suggestions
 st.sidebar.subheader("Example Questions")
 st.sidebar.write("- Add 3 and 4. Multiply the result by 2. Divide it by 5.")
 st.sidebar.write("- Tell me how many centuries Virat Kohli scored.")