Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import getpass
 import regex as re
 import gradio as gr
 import requests
@@ -13,14 +14,23 @@ 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_ollama.chat_models import ChatOllama
+from langchain.chat_models import init_chat_model
+from langchain_mistralai import ChatMistralAI
+from mistralai import Mistral
+
+
+import getpass
+import os
+
+if "Mistral" not in os.environ:
+    os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
 
 
 print("Loading LLM...")
-ollama.pull('gemma3:4b')
-chat = ChatOllama(model="gemma3:4b", verbose=True)
 print(f"Model {chat.model} downloaded!")
-
+chat = ChatMistralAI(
+    model="mistral-large-latest", mistral_api_key=os.getenv["Mistral"]
+)
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"