Update agent.py

agent.py

@@ -1,44 +1,18 @@
 """LangGraph: agent graph w/ tools"""
 import os
 from dotenv import load_dotenv
-from typing import List, Dict, Any, Optional
-import tempfile
-import re
-import json
-import requests
-from urllib.parse import urlparse
-import pytesseract
-from PIL import Image, ImageDraw, ImageFont, ImageEnhance, ImageFilter
-import cmath
-import pandas as pd
-import uuid
-import numpy as np
-
-
-""" Langchain imports"""
 from langgraph.graph import START, StateGraph, MessagesState
-from langchain_core.messages import SystemMessage, HumanMessage
-from langgraph.prebuilt import ToolNode, tools_condition
-from langchain_core.tools import tool
+from langgraph.prebuilt import tools_condition
+from langgraph.prebuilt import ToolNode
+from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings
 from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.document_loaders import ArxivLoader
-#from langchain_community.vectorstores import SupabaseVectorStore
-from langchain_google_genai import ChatGoogleGenerativeAI
-#from langchain.tools.retriever import create_retriever_tool
-#from supabase.client import Client, create_client
-#from code_interpreter import CodeInterpreter
-#interpreter_instance = CodeInterpreter()
-#from image_processing import *
-
-"""
-import getpass
-import os
-
-if "GOOGLE_API_KEY" not in os.environ:
-    os.environ["GOOGLE_API_KEY"] = getpass.getpass("Enter your Google AI API key: ")
-"""
+from langchain_core.messages import SystemMessage, HumanMessage
+from langchain_core.tools import tool
+from langchain.tools.retriever import create_retriever_tool
+from supabase.client import Client, create_client
 
 load_dotenv()
 
@@ -146,8 +120,6 @@ def web_search(query: str) -> str:
     return {"web_results": formatted_search_docs}
 
 
-
-
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()