Update tools.py

tools.py

@@ -8,18 +8,6 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 import base64
 
 #LLMs
-def get_access_token():
-    credential = EnvironmentCredential()
-    access_token = credential.get_token("https://cognitiveservices.azure.com/.default")
-    return access_token.token
-
-llm = AzureChatOpenAI(
-    model_name="gpt-4o",
-    api_key=get_access_token(),
-    azure_endpoint="https://cog-sandbox-dev-eastus2-001.openai.azure.com/",
-    api_version="2024-08-01-preview"
-)
-
 google_llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-lite')
 
 #IMAGE_TOOLS
@@ -65,7 +53,7 @@ def extract_text(img_path: str) -> str:
         ]
 
         # Call the vision-capable model
-        response = llm.invoke(message)
+        response = google_llm.invoke(message)
 
         # Append extracted text
         all_text += response.content + "\n\n"
@@ -119,7 +107,7 @@ def describe_image(img_path: str) -> str:
         ]
 
         # Call the vision-capable model
-        response = llm.invoke(message)
+        response = google_llm.invoke(message)
 
         # Append extracted text
         all_text += response.content + "\n\n"