Refactor check_format function to disallow nested lists and update agent model to 'Qwen/Qwen3-32B'

app.py

@@ -49,19 +49,20 @@ def load_file(path: str) -> list | dict:
     
 def check_format(answer: str | list, *args, **kwargs) -> list:
     """Check if the answer is a list and not a nested list."""
-    print("Checking format of the answer:", answer)
+    #print("Checking format of the answer:", answer)
     if isinstance(answer, list):
         for item in answer:
             if isinstance(item, list):
-                print("list detected")
+                print("Nested list detected")
                 raise TypeError("Nested lists are not allowed in the final answer.")
+        print("Final answer is a list:", answer)
+        return answer
     elif isinstance(answer, str):
         return [answer]
     elif isinstance(answer, dict):
-        raise TypeError(f"Final answer must be a list, not a dict. Please check the answer format. Error: {e}")
+        raise TypeError(f"Final answer must be a list, not a dict. Please check the answer format.")
         
         
-
 ## tools definition
 @tool
 def download_images(image_urls: str) -> list:
@@ -156,7 +157,7 @@ def generate_audio(prompt: str) -> object:
 ## agent definition
 class Agent:
     def __init__(self, ):
-        client = HfApiModel("google/gemma-3-27b-it", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
+        client = HfApiModel("Qwen/Qwen3-32B", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
         self.agent = CodeAgent(
             model=client,
             tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],
@@ -188,15 +189,21 @@ def respond(message: str, history : dict, web_search: bool = False):
     if not message.get("files") and not web_search: # no files uploaded
         print("No files received.")
         message = agent(text + "\nADDITIONAL CONTRAINT: Don't use web search", conversation_history=history) # conversation_history is a dict with the history of the conversation 
-    elif not message.get("files") and web_search==True: # no files uploaded
+    elif not message.get("files") and web_search: # no files uploaded
         print("No files received + web search enabled.")
         message = agent(text, conversation_history=history)
     else:
         files = message.get("files", [])
         print(f"files received: {files}")
-        if is_image_extension(files[0]):
+        if is_image_extension(files[0]) and not web_search:
             image = load_file(files[0]) # assuming only one file is uploaded at a time (gradio default behavior)
+            message = agent(text + "\nADDITIONAL CONTRAINT: Don't use web search", images=image, conversation_history=history)
+        elif is_image_extension(files[0]) and web_search:
+            image = load_file(files[0])
             message = agent(text, images=image, conversation_history=history)
+        elif not web_search:
+            file = load_file(files[0])
+            message = agent(text + "\nADDITIONAL CONTRAINT: Don't use web search", files=file, conversation_history=history)
         else:
             file = load_file(files[0])
             message = agent(text, files=file, conversation_history=history)

system_prompt.txt

@@ -312,19 +312,17 @@ Additional domain-specific behaviors:
    - This includes flagging problematic language, stereotyped depictions, or systemic imbalance in representation.
    - Where applicable, the assistant must provide annotated examples and suggest possible reframing or source guidance (e.g., UNESCO media diversity principles).
 ---- 
- 
 
 It is MANDATORY to use these rules for the 'final_answer' tool:
 	1.	Always return a Python list. Do not return a dictionary or any other type.
-	2.	The first element of the list must be a Gradio component (for example, gr.Image(…), gr.File(…), etc.).
-	3.	Any text or explanation must come after the component, as string elements in the same list.
-	4.	If there is no component to return, return a list whose only element is the text.
-	5.	Examples of valid returns:
-        • [gr.Image(value=“output.png”), “Here is the chart.”]
-        • [gr.File(value=“report.pdf”), “Download the report.”]
-	7.  Any deviation (returning a dict, tuple, raw PIL image, etc.) is invalid.
-    8.  Nested list are forbidden.
-    9.  Always put generated images in the list, do NOT create external link.
+	2.	Any text or explanation must come after the component, as string elements in the same list.
+	3.	If there is no component to return, return a list whose only element is the text.
+	4.	Examples of valid returns:
+        - [image, “Here is the first image”, image2, "Here is the second image"]
+        - [file, “Download the report.”]
+	5.  Any deviation (returning a dict, tuple, raw PIL image, etc.) is invalid.
+    6.  Nested list are forbidden.
+    7.  Always put generated images in the list, do NOT create external link.
 
 Now Begin!