Update app.py

app.py

@@ -7,7 +7,6 @@ import gradio as gr
 import pandas as pd
 import requests
 from smolagents import (
-    AmazonBedrockServerModel,
     DuckDuckGoSearchTool,
     FinalAnswerPromptTemplate,
     ManagedAgentPromptTemplate,
@@ -15,44 +14,10 @@ from smolagents import (
     OpenAIServerModel,
     PlanningPromptTemplate,
     PromptTemplates,
-    Tool,
     ToolCallingAgent,
     VisitWebpageTool,
 )
 
-
-class CustomAmazonBedrockServerModel(AmazonBedrockServerModel):
-    """
-    Fixed model class to handle Amazon Bedrock model.
-    """
-
-    def _prepare_completion_kwargs(
-        self,
-        messages: list[dict[str, str]],
-        stop_sequences: list[str] | None = None,
-        grammar: str | None = None,
-        tools_to_call_from: list[Tool] | None = None,
-        custom_role_conversions: dict[str, str] | None = None,
-        convert_images_to_image_urls: bool = False,
-        **kwargs,
-    ) -> dict:
-        """Remove invalid keys."""
-        completion_kwargs = super()._prepare_completion_kwargs(
-            messages=messages,
-            stop_sequences=None,  # Bedrock support stop_sequence using Inference Config
-            grammar=None,  # Bedrock doesn't support grammar
-            tools_to_call_from=tools_to_call_from,
-            custom_role_conversions=custom_role_conversions,
-            convert_images_to_image_urls=convert_images_to_image_urls,
-            **kwargs,
-        )
-
-        completion_kwargs.pop("tools")
-        completion_kwargs.pop("tool_choice")
-
-        return completion_kwargs
-
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -61,9 +26,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class GAIAgent:
-    SYSTEM_PROMPT = """You are a helpful assistant with web browsing capabilities.
-    You can search the web and visit webpages to find relevant information.
-    You should answer user questions based on the information you find."""
+    SYSTEM_PROMPT = (
+        "You are a helpful assistant with data and image analysis and web browsing capabilities. You can search the "
+        "web and visit webpages to find relevant information. You should answer user questions based on the "
+        "information you find.\n"
+        "Your final answer should only contain the answer to the question and nothing else."
+    )
 
     def __init__(self, model: Model):
         prompt_templates = PromptTemplates(