Update app.py

app.py

@@ -32,11 +32,11 @@ chat3 = ChatMistralAI(
     max_retries=6,
     mistral_api_key = os.getenv("Mistral")
 )
-chat2 = ChatMistralAI(
-    model="mistral-ocr-2505",
-    temperature=0.01,
-    max_retries=6,
-    mistral_api_key = os.getenv("Mistral")
+# chat2 = ChatMistralAI(
+#     model="mistral-ocr-2505",
+#     temperature=0.01,
+#     max_retries=6,
+#     mistral_api_key = os.getenv("Mistral")
 )
 print(f"Model {chat2.model} downloaded!")
 chat1 = ChatMistralAI(
@@ -64,7 +64,7 @@ def get_file_path(question: str) -> str:
 
 def get_ref_content(path: str) -> str | object:
     """Retrieves content from the reference path provided."""
-    if path.endswith('mp3') or path.startswith('https://www.youtube.com/'):
+    if path.endswith('.mp3') or path.startswith('https://www.youtube.com/'):
         file = librosa.load(path)
     elif path.endswith(".jpg") or path.endswith(".jpeg"):
         file = cv2.imread(path)
@@ -157,7 +157,7 @@ class AgentState(TypedDict):
 
 
 def build_agent(chat):    
-    tools = [get_file_path_tool, get_content_tool, search_web_tool, extract_text_tool, play_chess_tool, run_code_tool]
+    tools = [get_content_tool, search_web_tool, extract_text_tool, play_chess_tool, run_code_tool]
     chat_with_tools = chat.bind_tools(tools, parallel_tool_calls=False)
     # The graph
     builder = StateGraph(AgentState)