Update agent.py

agent.py

@@ -35,44 +35,44 @@ import whisper
 #    except Exception as e:
 #        return f"Error performing Google search: {str(e)}"
 
-@tool
-def ImageAnalysisTool(question: str, model: LiteLLMModel) -> str:
-    """Tool for analyzing images mentioned in the question.
-    Args:
-        question (str): The question text which may contain an image URL.
-    Returns:
-        str: Image description or error message.
-    """
-    # Extract URL from question using regex
-    url_pattern = r'https?://\S+'
-    match = re.search(url_pattern, question)
-    if not match:
-        return "No image URL found in the question."
-    image_url = match.group(0)
-
-    headers = {
-        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"
-    }
-    try:
-        response = requests.get(image_url, headers=headers)
-        response.raise_for_status()
-        image = Image.open(BytesIO(response.content)).convert("RGB")
-    except Exception as e:
-        return f"Error fetching image: {e}"
-
-    agent = CodeAgent(
-        tools=[],
-        model=model,
-        max_steps=10,
-        verbosity_level=2
-    )
-
-    response = agent.run(
-        "Describe in details the chess position you see in the image.",
-        images=[image]
-    )
-    
-    return f"The image description: '{response}'"
+#@tool
+#def ImageAnalysisTool(question: str, model: LiteLLMModel) -> str:
+#    """Tool for analyzing images mentioned in the question.
+#    Args:
+#        question (str): The question text which may contain an image URL.
+#    Returns:
+#        str: Image description or error message.
+#    """
+#    # Extract URL from question using regex
+#    url_pattern = r'https?://\S+'
+#    match = re.search(url_pattern, question)
+#    if not match:
+#        return "No image URL found in the question."
+#    image_url = match.group(0)
+#
+#    headers = {
+#        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"
+#    }
+#    try:
+#        response = requests.get(image_url, headers=headers)
+#        response.raise_for_status()
+#        image = Image.open(BytesIO(response.content)).convert("RGB")
+#    except Exception as e:
+#        return f"Error fetching image: {e}"
+#
+#    agent = CodeAgent(
+#        tools=[],
+#        model=model,
+#        max_steps=10,
+#        verbosity_level=2
+#    )
+#
+#    response = agent.run(
+#        "Describe in details the chess position you see in the image.",
+#        images=[image]
+#    )
+#    
+#    return f"The image description: '{response}'"
 
 @tool
 def SpeechToTextTool(audio_path: str) -> str:
@@ -98,6 +98,7 @@ def youtube_transcript(url: str) -> str:
         url: YouTube video url in ""
     """    
     video_id = url.partition("https://www.youtube.com/watch?v=")[2]
+
     transcript = YouTubeTranscriptApi.get_transcript(video_id)
     transcript_text = " ".join([item["text"] for item in transcript])
     return {"youtube_transcript": transcript_text}
@@ -138,7 +139,7 @@ class MagAgent:
 #                GoogleSearchTool,
                 DuckDuckGoSearchTool(),
                 WikipediaSearchTool(),
-                ImageAnalysisTool,
+#                ImageAnalysisTool,
                 SpeechToTextTool
 #                LocalFileAudioTool()
             ]