Update app.py

app.py

@@ -1,20 +1,58 @@
+import base64
 import os
+import tempfile
+from io import BytesIO
 
 import gradio as gr
 import pandas as pd
 import requests
 from smolagents import (
+    AmazonBedrockServerModel,
     DuckDuckGoSearchTool,
     FinalAnswerPromptTemplate,
-    InferenceClientModel,
     ManagedAgentPromptTemplate,
     Model,
+    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"
@@ -45,12 +83,36 @@ class GAIAgent:
             prompt_templates=prompt_templates,
         )
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, task_id: str, question: str, file_name: str | None = None) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+        if file_name:
+            attachment_text = self._process_attachment(task_id, file_name)
+            question += f"\n{attachment_text}"
         answer = self.agent.run(question)
         print(f"Agent returning answer: {answer}")
         return answer
 
+    def _process_attachment(self, task_id: str, file_name: str) -> str:
+        api_url = DEFAULT_API_URL
+        get_associated_files_url = f"{api_url}/files/{task_id}"
+        response = requests.get(get_associated_files_url, timeout=15)
+        response.raise_for_status()
+
+        if file_name.endswith(".mp3"):
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as f:
+                f.write(response.content)
+                return f".mp3 file path: {f.name}\n"
+        elif file_name.endswith(".py"):
+            file_content = response.text
+            return "Python code:\n```python\n" + file_content + "\n```\n"
+        elif file_name.endswith(".xlsx"):
+            xlsx_io = BytesIO(response.content)
+            csv_data = pd.read_excel(xlsx_io).to_csv(index=False)
+            return "Excel file (as CSV):\n```csv\n" + csv_data + "\n```\n"
+        elif file_name.endswith(".png"):
+            base64_str = base64.b64encode(response.content).decode("utf-8")
+            return ".png image (in base64 format):\n\n```base64\n" + base64_str + "\n```\n"
+
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
@@ -73,8 +135,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        model = InferenceClientModel(
-            model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        model = OpenAIServerModel(
+            model_id="gpt-4o-mini",
+            api_base=os.environ["OPENAI_URL"],
+            api_key=os.environ["OPENAI_API_KEY"],
         )
         agent = GAIAgent(model=model)
     except Exception as e:
@@ -112,11 +176,12 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        file_name = item.get("file_name")
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(task_id, question_text, file_name)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: