Update src/chat_langraph.py

src/chat_langraph.py

@@ -8,6 +8,8 @@ from langgraph.prebuilt.tool_node import ToolNode
 import sqlite3
 import subprocess
 import requests
+import matplotlib.pyplot as plt 
+import uuid
 from datetime import datetime
 import os
 
@@ -155,7 +157,53 @@ def search_tool(query: str) -> dict:
         return response.json()
     except Exception as e:
         return {"error": str(e)}
+@tool
+def plot_graph(expression: str, variable: str = "x", range_start: float = 0, range_end: float = 10, step: float = 1, title: str = "Graph") -> str:
+    """
+    Plot a graph from a dynamic expression.
 
+    Args:
+        expression (str): Python expression as a function of variable (e.g., "2*x + 3").
+        variable (str): Variable name to use in expression (default: "x").
+        range_start (float): Start of variable range.
+        range_end (float): End of variable range.
+        step (float): Step size for variable.
+        title (str): Graph title.
+
+    Returns:
+        str: Path to saved image file.
+    """
+    try:
+        x_values = []
+        y_values = []
+        safe_locals = {"math": math}  # Allow math functions
+
+        val = range_start
+        while val <= range_end:
+            safe_locals[variable] = val
+            try:
+                y = eval(expression, {"__builtins__": None}, safe_locals)
+            except Exception as e:
+                return f"Error evaluating expression: {e}"
+            x_values.append(val)
+            y_values.append(y)
+            val += step
+
+        plt.figure()
+        plt.plot(x_values, y_values, marker="o")
+        plt.title(title)
+        plt.xlabel(variable)
+        plt.ylabel("Value")
+        plt.grid(True)
+
+        filename = f"/tmp/graph_{uuid.uuid4().hex}.png"
+        plt.savefig(filename)
+        plt.close()
+
+        return f"Graph saved at {filename}"
+
+    except Exception as e:
+        return f"Error plotting graph: {e}"
 
 # ======================== STATE GRAPH ======================== #