Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import gradio as gr
 import pandas as pd
 from sqlalchemy import create_engine, text
+from langchain.tools import tool
 from code_agent import CodeAgent
 from hf_api_model import HfApiModel