Update app.py

app.py

@@ -5,7 +5,6 @@ import gradio as gr
 import requests
 import pandas as pd
 import base64
-import ollama
 
 from typing import TypedDict, Annotated
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage
@@ -14,18 +13,19 @@ from langchain.tools import Tool
 from langgraph.graph import START, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
-from langchain_google_vertexai.gemma import GemmaChatLocalKaggle
+from langchain_mistralai import ChatMistralAI
 
 
 import getpass
 import os
 
-if "KAGGLE_API" not in os.environ:
-    os.environ["KAGGLE_API"] = getpass.getpass("Enter your Kaggle API key: ")
+if "Mistral" not in os.environ:
+    os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
 
 print("Loading LLM...")
-chat = GemmaChatLocalKaggle(
-    model="gemma3:4b", kwargs={"api_key": os.getenv("KAGGLE_API")}
+chat = ChatMistralAI(
+    model="mistral-large-latest",
+    mistral_api_key = os.getenv("Mistral")
 )
 print(f"Model {chat.model} downloaded!")