Update app.py

app.py

@@ -26,14 +26,14 @@ if "Mistral" not in os.environ:
     os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
 
 print("Loading LLM...")
-chat1 = ChatMistralAI(
+chat2 = ChatMistralAI(
     model="pixtral-large-2411",
     temperature=0.01,
     max_retries=6,
     mistral_api_key = os.getenv("Mistral")
 )
 print(f"Model {chat1.model} downloaded!")
-chat2 = ChatMistralAI(
+chat1 = ChatMistralAI(
     model="mistral-medium-2505",
     temperature=0.01,
     max_retries=6,
@@ -189,7 +189,7 @@ class BasicAgent:
             YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. \
             If you are asked for a number, use only digits in your final answer. Don't use comma nor brackets to write your number neither use units such as $ or percent sign unless specified otherwise. \
             If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. \
-            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.\
+            If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. Wrap the list with [square brackets].\
             If there is a file attached, open the file and read it. \
             If you don't have enough references to answer, use your tools, search the web, run your code or convert data to a data frame, whatever helps. \
             If the question refers to an external content and there is no reference file attached, perform a web search and retrieve relevant information from the internet. \
@@ -214,7 +214,7 @@ class BasicAgent:
             answer = answer.replace('*', '')
             print(f"Agent returning fixed answer: {answer}")        
             
-        return answer
+            return answer
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):