Update app.py

app.py

@@ -193,16 +193,17 @@ def run_agent(client, model_name, system_prompt_template, user_input_state, conf
     prompt = system_prompt + "\n\n" + user_message_content
 
     try:
-        # *** FIX: Define model_to_use by using the passed model_name parameter ***
+        # Define model_to_use by using the passed model_name parameter
         model_to_use = model_name
 
         # Pass a list containing the single combined prompt string
         response = client.models.generate_content(
-            model=model_to_use, # <-- Now defined
-            contents=[prompt],
+            model=model_to_use,
+            contents=[prompt], # Pass a list of strings
             config=config
         )
-        response.raise_for_status() # Raise an exception for bad status codes
+        # *** FIX: REMOVED response.raise_for_status() as it doesn't exist on GenerateContentResponse ***
+        # API errors are handled by the SDK raising exceptions caught below.
 
         # Some models return parts, concatenate them
         response_text = "".join([part.text for part in response.candidates[0].content.parts])