Update app.py

app.py

@@ -50,13 +50,32 @@ wiki_tool = WikipediaTool()
 tools = [audio_tool, image_tool, wiki_tool]
 
 # Define the custom agent using Mixtral on Hugging Face
+#class MyAgent(CodeAgent):
+ #   def __init__(self):
+ #       model = HfApiModel(
+  #          api_url="https://api-inference.huggingface.com/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
+   #         api_key=os.getenv("HF_API_TOKEN")
+   #     )
+   #     super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
+
+
+
 class MyAgent(CodeAgent):
     def __init__(self):
         model = HfApiModel(
             api_url="https://api-inference.huggingface.com/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
             api_key=os.getenv("HF_API_TOKEN")
         )
-        super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
+        super().__init__(model=model, tools=tools)
+
+    def __call__(self, question_dict):
+        question = question_dict.get("question", "")
+        full_prompt = SYSTEM_PROMPT + "\n\n" + question
+        # Pass a new dict with the full prompt, adjust key if your agent expects differently
+        new_input = question_dict.copy()
+        new_input["question"] = full_prompt
+        return super().__call__(new_input)
+
 
 # Evaluation + Submission function
 def run_and_submit_all(profile: gr.OAuthProfile | None):